Files
notifications-api/app/aws/s3.py
Nicholas Staples 4cc0028b01 Remove csv after process job is finished.
Fixed new tests.
2016-04-05 14:55:03 +01:00

17 lines
421 B
Python

from boto3 import resource
def get_s3_job_object(bucket_name, job_id):
s3 = resource('s3')
return s3.Object(bucket_name, '{}.csv'.format(job_id))
def get_job_from_s3(bucket_name, job_id):
obj = get_s3_job_object(bucket_name, job_id)
return obj.get()['Body'].read().decode('utf-8')
def remove_job_from_s3(bucket_name, job_id):
obj = get_s3_job_object(bucket_name, job_id)
return obj.delete()