2016-02-02 22:26:49 +00:00
|
|
|
import botocore
|
2016-01-13 17:32:40 +00:00
|
|
|
from boto3 import resource
|
|
|
|
|
|
|
|
|
|
|
2016-02-02 22:26:49 +00:00
|
|
|
def s3upload(upload_id, service_id, filedata, region):
|
2016-01-13 17:32:40 +00:00
|
|
|
s3 = resource('s3')
|
2016-02-02 22:26:49 +00:00
|
|
|
bucket_name = 'service-{}-notify'.format(service_id)
|
2016-01-14 16:00:13 +00:00
|
|
|
contents = '\n'.join(filedata['data'])
|
2016-02-02 22:26:49 +00:00
|
|
|
|
|
|
|
|
bucket = s3.Bucket(bucket_name)
|
|
|
|
|
exists = True
|
|
|
|
|
try:
|
|
|
|
|
s3.meta.client.head_bucket(Bucket=bucket_name)
|
|
|
|
|
except botocore.exceptions.ClientError as e:
|
|
|
|
|
error_code = int(e.response['Error']['Code'])
|
|
|
|
|
if error_code == 404:
|
|
|
|
|
exists = False
|
|
|
|
|
|
|
|
|
|
if not exists:
|
|
|
|
|
s3.create_bucket(Bucket=bucket_name,
|
|
|
|
|
CreateBucketConfiguration={'LocationConstraint': region})
|
|
|
|
|
|
2016-01-14 16:00:13 +00:00
|
|
|
key = s3.Object(bucket_name, upload_id)
|
|
|
|
|
key.put(Body=contents, ServerSideEncryption='AES256')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def s3download(service_id, upload_id):
|
|
|
|
|
s3 = resource('s3')
|
2016-02-02 22:26:49 +00:00
|
|
|
bucket_name = 'service-{}-notify'.format(service_id)
|
2016-01-14 16:00:13 +00:00
|
|
|
key = s3.Object(bucket_name, upload_id)
|
|
|
|
|
contents = key.get()['Body'].read().decode('utf-8')
|
|
|
|
|
return contents
|