diff --git a/progressbar/progressbar.py b/progressbar/progressbar.py index 3baf530..a3becc3 100644 --- a/progressbar/progressbar.py +++ b/progressbar/progressbar.py @@ -164,6 +164,12 @@ def __next__(self): # an iterator. next = __next__ + def __enter__(self): + self.start() + return self + + def __exit__(self, type, value, traceback): + self.finish() def _env_size(self): """Tries to find the term_width from the environment."""