Skip to content

Instantly share code, notes, and snippets.

@samuell
Created August 15, 2014 18:11
Show Gist options
  • Select an option

  • Save samuell/c28d50d3df4374adeb1e to your computer and use it in GitHub Desktop.

Select an option

Save samuell/c28d50d3df4374adeb1e to your computer and use it in GitHub Desktop.
class TimingMixin():
start_time = None
end_time = None
exec_time = None
@luigi.Task.event_handler(luigi.Event.DEPENDENCY_DISCOVERED)
def save_start_time(self):
self.start_time = time.time()
self.log("Dependency present. Starting task")
@luigi.Task.event_handler(luigi.Event.SUCCESS)
def save_end_time(self):
self.end_time = time.time()
import pdb; pdb.set_trace()
self.exec_time = time.strftime('%H:%M:%S', time.gmtime(self.end_time - self.start_time))
self.log("Task succeded. Time executed (H:M:S): " + self.exec_time)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment