darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-02 17:48:50 -04:00
Code Issues Packages Projects Releases Wiki Activity

Better bucket name pending possible change to uuid for service primary

Browse Source 
key.

Also small bugfix around file name and original file name
This commit is contained in:
Adam Shimali
2016-01-29 12:19:50 +00:00
parent 0dc6f7be1a
commit 5196436645
5 changed files with 22 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/main/uploader.py
View File

@@ -6,7 +6,7 @@ from boto3 import resource
def s3upload(service_id, filedata):
upload_id = str(uuid.uuid4())
s3 = resource('s3')
bucket_name = 'service-{}-notify'.format(service_id)
bucket_name = 'service-{}-{}-notify'.format(service_id, upload_id)
s3.create_bucket(Bucket=bucket_name)
contents = '\n'.join(filedata['data'])
key = s3.Object(bucket_name, upload_id)
@@ -16,7 +16,7 @@ def s3upload(service_id, filedata):
def s3download(service_id, upload_id):
s3 = resource('s3')
bucket_name = 'service-{}-notify'.format(service_id)
bucket_name = 'service-{}-{}-notify'.format(service_id, upload_id)
key = s3.Object(bucket_name, upload_id)
contents = key.get()['Body'].read().decode('utf-8')
return contents