add a progress bar

msys2 · Dec 9, 2023 · aaca202 · aaca202
1 parent 56fbad1
commit aaca202
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/msys2-repo-sigstats b/msys2-repo-sigstats
@@ -14,6 +14,7 @@ from typing import NamedTuple, Iterator
 from pgpdump import BinaryData
 from pgpdump.utils import PgpdumpException
 from tabulate import tabulate
+from fastprogress.fastprogress import progress_bar
 
 from msys2_devtools.db import ExtTarFile
 
@@ -103,7 +104,7 @@ def get_signature_paths(root_path: str, include_all: bool) -> list[str]:
 
 
 def parse_signatures(paths: list[str]) -> Iterator[tuple[str, Signature]]:
-    for p in paths:
+    for p in progress_bar(paths, leave=False):
         with open(p, "rb") as h:
             data = h.read()
         yield (p, parse_signature(data))

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,6 +15,7 @@ zstandard = "^0.22.0"
 pyyaml = "^6.0"
 pydantic = "^2.0"
 pgpdump = "^1.5"
+fastprogress = "^1.0.3"
 
 [tool.poetry.scripts]
 msys2-srcinfo-cache = 'msys2_devtools.srcinfo_cache:run'