jshbrmn/irt-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

envs fixed

Browse Source
This commit is contained in:
Josh Burman
2021-11-03 17:18:02 +00:00
parent 10dbed501b
commit b005efbe2e
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/helpers/aws_helper.py
View File

@ -3,12 +3,12 @@ import os
import json import json
session = boto3.Session( session = boto3.Session(
aws_access_key_id=os.environ['SOLVER_AWS_ACCESS_KEY_ID'], aws_access_key_id=os.environ['AWS_ACCESS_KEY_ID'],
aws_secret_access_key=os.environ['SOLVER_AWS_SECRET_ACCESS_KEY'] aws_secret_access_key=os.environ['AWS_SECRET_ACCESS_KEY']
) )
s3 = session.resource('s3', region_name=os.environ['SOLVER_AWS_REGION']) s3 = session.resource('s3', region_name=os.environ['AWS_REGION'])
sqs = session.client('sqs', region_name=os.environ['SOLVER_AWS_REGION']) sqs = session.client('sqs', region_name=os.environ['AWS_REGION'])
def get_key_from_message(body): def get_key_from_message(body):
return body['Records'][0]['s3']['object']['key'] return body['Records'][0]['s3']['object']['key']

2
app/services/loft_service.py
View File

@ -18,7 +18,7 @@ class LoftService(Base):
logging.info('Retrieving attributes from message...') logging.info('Retrieving attributes from message...')
# get s3 object # get s3 object
self.key = aws_helper.get_key_from_message(self.source) self.key = aws_helper.get_key_from_message(self.source)
s3_object = aws_helper.get_object(self.key, os.environ['SOLVER_INGEST_BUCKET']) s3_object = aws_helper.get_object(self.key, os.environ['INGEST_BUCKET'])
# convert to tar # convert to tar
self.tar = tar_helper.raw_to_tar(s3_object) self.tar = tar_helper.raw_to_tar(s3_object)