Add ProgressBar class

plextraktsync/rich/RichProgressBar.py

@@ -0,0 +1,32 @@
+from functools import cached_property
+
+
+class RichProgressBar:
+    def __init__(self, iterable, total, options=None, desc=""):
+        self.iter = iterable
+        self.options = options or {}
+        self.desc = desc
+        self.total = total
+
+    def __iter__(self):
+        p = self.progress
+        task_id = p.add_task(self.desc, total=self.total)
+
+        i = 0
+        for it in self.iter:
+            yield it
+            i += 1
+            p.update(task_id, completed=i)
+
+    def __enter__(self):
+        self.progress.__enter__()
+        return self
+
+    def __exit__(self, *exc):
+        self.progress.__exit__(*exc)
+
+    @cached_property
+    def progress(self):
+        from rich.progress import Progress
+
+        return Progress(**self.options)