ID

0f9d749

date

2023-04-30 08:16:20+00:00

author

Alex Chan <alex@alexwlchan.net>

parent

9a1f970

message

use tqdm for downloading messages from SQS

changed files

1 file, 2 additions, 1 deletion

• aws/download_sqs_messages

Changed files

aws/download_sqs_messages (2042) → aws/download_sqs_messages (2065)

diff --git a/aws/download_sqs_messages b/aws/download_sqs_messages
index 1b63445..46e19a4 100755
--- a/aws/download_sqs_messages
+++ b/aws/download_sqs_messages
@@ -5,6 +5,7 @@ import json
 import os
 
 import boto3
+import tqdm
 
 from _common import ACCOUNT_NAMES, get_aws_session
 
@@ -75,5 +76,5 @@ def download_messages(*, queue_url):
 
 if __name__ == "__main__":
     args = parse_args()
-    for message in download_messages(queue_url=args.QUEUE_URL):
+    for message in tqdm.tqdm(download_messages(queue_url=args.QUEUE_URL)):
         print(json.dumps(message))