From e63014b3796e5a15aa14fe4c53ff6a7d1ca1a06b Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Tue, 4 Feb 2020 11:48:09 +0100
Subject: [PATCH] TimeTracking can be used by the with statement to easily
measure time spent
---
src/helpers.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/src/helpers.py b/src/helpers.py
index 0fccf15b..5646eb94 100644
--- a/src/helpers.py
+++ b/src/helpers.py
@@ -88,6 +88,13 @@ class TimeTracking(object):
def duration(self):
return self._duration()
+ def __enter__(self):
+ return self
+
+ def __exit__(self, exc_type, exc_val, exc_tb):
+ self.stop()
+ logging.info(f"undefined job finished after {self}")
+
def prepare_host(create_new=True):
hostname = socket.gethostname()
--
GitLab