first commit

swingmusic/utils/progressbar.py

@@ -0,0 +1,15 @@
+from tqdm import tqdm as _tqdm
+
+
+def tqdm(*args, **kwargs):
+    """
+    Wrapper for tqdm that sets globals.
+    """
+    bar_format = "{percentage:3.0f}%|{bar:45}|{n_fmt}/{total_fmt}{desc}"
+    kwargs["bar_format"] = bar_format
+
+    if "desc" in kwargs:
+        print(f"INFO|{kwargs['desc'].capitalize()} ...")
+        kwargs["desc"] = ""
+
+    return _tqdm(*args, **kwargs)