Add context manager to progress dialogs

anxdpanic · Dec 15, 2023 · 5c27828 · 5c27828
1 parent 764caec
commit 5c27828
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/resources/lib/youtube_plugin/kodion/ui/abstract_progress_dialog.py b/resources/lib/youtube_plugin/kodion/ui/abstract_progress_dialog.py
@@ -19,6 +19,12 @@ def __init__(self, dialog, heading, text, total=100):
         self._position = 1
         self.update(steps=-1)
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
     def get_total(self):
         return self._total