Add set_items_completed() to ProgressBarCounter

prompt-toolkit · Jun 14, 2020 · 8f31302 · 8f31302
1 parent 5562121
commit 8f31302
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/prompt_toolkit/shortcuts/progress_bar/base.py b/prompt_toolkit/shortcuts/progress_bar/base.py
@@ -366,6 +366,11 @@ def item_completed(self, n: int = 1) -> None:
         self.items_completed += n
         self.progress_bar.invalidate()
 
+    def set_items_completed(self, nb_items: int) -> None:
+        if nb_items != self.items_completed:
+            self.items_completed = nb_items
+            self.progress_bar.invalidate()
+
     @property
     def done(self) -> bool:
         """Whether a counter has been completed.