feat: tqdm for S3-list

2023-02-18 20:54:35 +03:00
parent 07c8bea489
commit 894d0b24c5
1 changed files with 3 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -22,7 +22,9 @@ s3_config_data = json.loads(open(expanduser("~/.mc/config.json")).read())["alias
 def get_files(s3, bucket, prefix) -> Mapping[str, Tuple[int, str]]:
    res = {}
    prefix_len = len(prefix)
-    for obj in s3.list_objects(bucket, prefix=prefix, recursive=True):
+    for obj in tqdm.tqdm(
+        s3.list_objects(bucket, prefix=prefix, recursive=True), desc="S3 list objects"
+    ):
        if obj.is_dir:
            continue
        res[obj.object_name[prefix_len:].lstrip("/")] = (obj.size, obj.etag)