ID

3de1b63

date

2023-04-29 10:03:46+00:00

author

Alex Chan <alex@alexwlchan.net>

parent

3044833

message

run black over bulk_sns_publish

changed files

1 file, 4 additions, 4 deletions

• aws/bulk_sns_publish

Changed files

aws/bulk_sns_publish (4637) → aws/bulk_sns_publish (4614)

diff --git a/aws/bulk_sns_publish b/aws/bulk_sns_publish
index b25e32a..b14c436 100755
--- a/aws/bulk_sns_publish
+++ b/aws/bulk_sns_publish
@@ -123,14 +123,14 @@ def parse_args():
 
 def publish_batch(sns_client, topic_arn, batch_entries):
     resp = sns_client.publish_batch(
-                    TopicArn=topic_arn, PublishBatchRequestEntries=batch_entries
-                )
+        TopicArn=topic_arn, PublishBatchRequestEntries=batch_entries
+    )
 
     # This is to account for any failures in sending messages to SNS.
     # I've never actually had this happen in practice so I've not written
     # any code to handle it (I'd probably just retry the whole script)
     # but I include it just in case.
-    assert len(resp['Failed']) == 0, resp
+    assert len(resp["Failed"]) == 0, resp
 
 
 def publish_messages(*, input_file, topic_arn):
@@ -149,7 +149,7 @@ def publish_messages(*, input_file, topic_arn):
         for (batch, _) in concurrently(
             handler=functools.partial(publish_batch, sns_client, topic_arn),
             inputs=get_batch_entries(input_file),
-            max_concurrency=8
+            max_concurrency=8,
         ):
             pbar.update(len(batch))