diff --git a/src/helpers.py b/src/helpers.py index 0fccf15b85662c61471e7c2bde1ac8116039ba95..5646eb94dbd43941b5673e64f6b70a7ed0e51c26 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()