added another progress bar

Australian-Imaging-Service · Feb 9, 2024 · 307958f · 307958f
1 parent 4b0113f
commit 307958f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/xnat_ingest/cli/upload.py b/xnat_ingest/cli/upload.py
@@ -202,7 +202,10 @@ def iter_staged_sessions():
                     # we recreate the project/subject/sesssion directory structure
                     session_tmp_dir = tmp_download_dir.joinpath(*ids)
                     session_tmp_dir.mkdir(parents=True, exist_ok=True)
-                    for relpath, obj in objs:
+                    for relpath, obj in tqdm(
+                        objs,
+                        desc=f"Downloading scans in {ids} session from S3 bucket"
+                    ):
                         obj_path = session_tmp_dir.joinpath(*relpath)
                         obj_path.parent.mkdir(parents=True, exist_ok=True)
                         logger.debug("Downloading %s to %s", obj, obj_path)