add tqdm

adchia · adchia · commit 4d858d646466 · 2022-08-15T10:35:29.000-04:00
Signed-off-by: Danny Chiao &lt;danny@tecton.ai&gt;
diff --git a/infra/scripts/cleanup_dynamo_ci.py b/infra/scripts/cleanup_dynamo_ci.py
@@ -1,15 +1,21 @@
 import boto3
+from tqdm import tqdm
 
 
 def main() -> None:
     db = boto3.resource("dynamodb")
 
-    num_deleted = 0
-    for table in db.tables.all():
+    num_to_delete = 0
+    all_tables = db.tables.all()
+    for table in all_tables:
         if "integration_test" in table.name:
-            table.delete()
-            num_deleted += 1
-    print(f"Deleted {num_deleted} CI DynamoDB tables")
+            num_to_delete += 1
+    with tqdm(total=num_to_delete) as progress:
+        for table in all_tables:
+            if "integration_test" in table.name:
+                table.delete()
+                progress.update()
+    print(f"Deleted {num_to_delete} CI DynamoDB tables")
 
 
 if __name__ == "__main__":
