Create Progress the way tqdm.rich made it

Taxel · Jan 6, 2024 · 5c77e1a · 5c77e1a
1 parent 0800ebb
commit 5c77e1a
Showing 1 changed file with 28 additions and 3 deletions.
diff --git a/plextraktsync/rich/RichProgressBar.py b/plextraktsync/rich/RichProgressBar.py
@@ -27,6 +27,31 @@ def __exit__(self, *exc):
 
     @cached_property
     def progress(self):
-        from rich.progress import Progress
-
-        return Progress(**self.options)
+        from tqdm.rich import FractionColumn, RateColumn
+
+        from rich.progress import (BarColumn, Progress, TimeElapsedColumn,
+                                   TimeRemainingColumn)
+
+        args = (
+            "[progress.description]{task.description}"
+            "[progress.percentage]{task.percentage:>4.0f}%",
+            BarColumn(bar_width=None),
+            FractionColumn(
+                unit_scale=False,
+                unit_divisor=1000,
+            ),
+            "[",
+            TimeElapsedColumn(),
+            "<",
+            TimeRemainingColumn(),
+            ",",
+            RateColumn(
+                unit="it",
+                unit_scale=False,
+                unit_divisor=1000,
+            ),
+            "]"
+        )
+        progress = Progress(*args, **self.options)
+
+        return progress