From 7ee3facd2ea85429d76c52741f12cc27cb5d7243 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 20 Sep 2015 11:50:04 -0500
Subject: [PATCH] Make ProgressBar a context manager

---
 pytools/__init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pytools/__init__.py b/pytools/__init__.py
index 74ebba1..557278b 100644
--- a/pytools/__init__.py
+++ b/pytools/__init__.py
@@ -1648,6 +1648,12 @@ class ProgressBar:
         self.set_progress(self.total)
         sys.stderr.write("\n")
 
+    def __enter__(self):
+        self.draw()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.finished()
+
 # }}}
 
 
-- 
GitLab