From 420313d62369f0602a7e7f846728c390ae382289 Mon Sep 17 00:00:00 2001
From: Dirk Uys <dirkcuys@gmail.com>
Date: Wed, 19 Jun 2024 17:08:35 +0200
Subject: [PATCH] Upload to S3 and generate pre-signed URL

---
 studygroups/tasks.py | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/studygroups/tasks.py b/studygroups/tasks.py
index 8824661b..89e75238 100644
--- a/studygroups/tasks.py
+++ b/studygroups/tasks.py
@@ -33,6 +33,8 @@
 import tempfile
 import unicodecsv as csv
 import json
+import boto3
+from botocore.exceptions import ClientError
 
 
 logger = logging.getLogger(__name__)
@@ -670,6 +672,26 @@ def export_signups(user_id):
             [ signup.communications_opt_in ]
         )
 
-    # TODO - Upload to AWS S3 and generate presigned URL
-    return { "presign_url": "TODO" }
+    # Upload to AWS S3 and generate presigned URL
+    ts = timezone.now().utcnow().isoformat()
+    filename = 'signups-{}.csv'.format(ts)
+    key = '/'.join(['learning-circles', 'exports', filename])
+    bucket = settings.P2PU_RESOURCES_AWS_BUCKET
+
+    s3 = boto3.client('s3', aws_access_key_id=settings.P2PU_RESOURCES_AWS_ACCESS_KEY, aws_secret_access_key=settings.P2PU_RESOURCES_AWS_SECRET_KEY)
+    temp_file.seek(0)
+    response = s3.upload_fileobj(
+        temp_file, bucket, key,
+        ExtraArgs={'Expires': 3600}
+    )
+
+    try:
+        presigned_url = s3.generate_presigned_url(
+            'get_object', Params={'Bucket': bucket, 'Key': key}, ExpiresIn=3600
+        )
+    except ClientError as e:
+        logger.error(e)
+        return None
+
+    return { "presigned_url": presigned_url }