use bucket name from payload instead of env var

2021-11-03 17:58:48 +00:00
parent 1c7fb41ca5
commit a42ae76e72
3 changed files with 5 additions and 2 deletions
--- a/app/helpers/aws_helper.py
+++ b/app/helpers/aws_helper.py
@ -13,6 +13,9 @@ sqs = session.client('sqs', region_name=os.environ['AWS_REGION'])
 def get_key_from_message(body):
  return body['Records'][0]['s3']['object']['key']

+def get_bucket_from_message(body):
+  return body['Records'][0]['s3']['bucket']['name']
+
 def get_object(key, bucket):
  return s3.Object(
    bucket_name=bucket,