rewrite for robustness.

progress.py

@@ -0,0 +1,65 @@
+# @File   : progress.py
+# @Time   : 2026/04/12 22:20:53
+# @Author : SilverAg.L
+
+from shutil import get_terminal_size
+from sys import stdout
+
+
+class RawProgressBar:
+    def __init__(self, total=100, unit='%', bar_format=None, **kwargs):
+        self.total = total
+        self.unit = unit
+        self.n = 0
+        self.cols = get_terminal_size((40, 20)).columns
+        self.bar_format = bar_format  # unused
+
+    def update(self, delta):
+        self.n += delta
+        self.refresh()
+
+    def refresh(self):
+        cur = min(self.n, self.total)
+        pct = min(cur / self.total if self.total > 0 else 0, 1.0)
+        pct_str = f"{pct * 100:3.0f}%"
+        rate_str = f"{cur:.2f}/{self.total:.2f}({self.unit})"
+        fixed_len = len(pct_str) + len(rate_str) + 10
+        bar_width = max(self.cols - fixed_len, 10)
+        filled_len = int(bar_width * pct)
+        bar = '>' * filled_len + '-' * (bar_width - filled_len)
+        output = f"\r {pct_str} [{bar}] {rate_str}".ljust(self.cols)
+        stdout.write(output)
+        stdout.flush()
+
+    def close(self):
+        stdout.write('\n')
+        stdout.flush()
+
+
+try:
+    from tqdm import tqdm
+
+    def get_progress_bar(total, unit, bar_format=None):
+        return tqdm(file=stdout, total=total, unit=unit, bar_format=bar_format)
+except ImportError:
+    def get_progress_bar(total, unit, bar_format=None):
+        return RawProgressBar(total=total, unit=unit, bar_format=bar_format)
+
+
+def is_tty():
+    return stdout.isatty()
+
+
+def test_progress_bar():
+    from time import sleep
+    pbar = get_progress_bar(total=100, unit='%')
+    for i in range(101):
+        sleep(0.1)
+        pbar.n = i
+        pbar.refresh()
+    pbar.close()
+    print("Done!")
+
+
+if __name__ == "__main__":
+    test_progress_bar()