* Allow the user to set and exipration on the exports.

bgroff · Nov 1, 2018 · a916727 · a916727
1 parent 5313523
commit a916727
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/django_kala/django_kala/platforms/aws/documents.py b/django_kala/django_kala/platforms/aws/documents.py
@@ -1,4 +1,7 @@
+import datetime
+
 from django.conf import settings
+from django.utils import timezone
 from uuid import uuid4
 
 import boto3
@@ -25,7 +28,8 @@ def upload_document(self, content, key):
             Body=content,
             Bucket=settings.S3_STORAGE_BUCKET,
             Key='media/documents/{0}'.format(key),
-            ServerSideEncryption='AES256'
+            ServerSideEncryption='AES256',
+
         )
 
     def get_export_url(self, export):
@@ -42,12 +46,16 @@ def get_export_url(self, export):
 
     def upload_export(self, export_path):
         key = 'exports/{0}'.format(uuid4())
+        expires = timezone.now() + datetime.timedelta(days=settings.EXPORT_EXPIRATION_IN_DAYS)
 
         s3 = boto3.resource('s3')
         s3.meta.client.upload_file(
-            export_path,
-            settings.S3_STORAGE_BUCKET,
-            key
+            Filename=export_path,
+            Bucket=settings.S3_STORAGE_BUCKET,
+            Key=key,
+            ExtraArgs={
+                'Expires': expires
+            }
         )
 
         return {'Key': key}

diff --git a/django_kala/django_kala/settings/storage.py b/django_kala/django_kala/settings/storage.py
@@ -3,3 +3,5 @@
 
 if not DEBUG:
     DEFAULT_FILE_STORAGE = 'django_kala.storage.MediaStorage'
+
+EXPORT_EXPIRATION_IN_DAYS=5