catching GeneratorExit to make sure we always end clean to fix #212

wolph · Nov 5, 2019 · d038ce8 · d038ce8
1 parent 0c4298e
commit d038ce8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/progressbar/bar.py b/progressbar/bar.py
@@ -532,7 +532,8 @@ def __next__(self):
             return value
         except StopIteration:
             self.finish()
-            raise
+        except GeneratorExit:
+            self.finish(dirty=True)
 
     def __exit__(self, exc_type, exc_value, traceback):
         self.finish(dirty=bool(exc_type))