configured settings for Image bucket

Author: ElaineShu0312

csm_web/csm_web/settings.py

@@ -11,11 +11,13 @@
 """
 
 import os
+from datetime import datetime
 
 import sentry_sdk
 from factory.django import DjangoModelFactory
 from rest_framework.serializers import ModelSerializer, Serializer
 from sentry_sdk.integrations.django import DjangoIntegration
+from storages.backends.s3boto3 import S3Boto3Storage
 
 # Analogous to RAILS_ENV, is one of {prod, staging, dev}. Defaults to dev. This default can
 # be dangerous, but is worth it to avoid the hassle for developers setting the local ENV var
@@ -67,6 +69,7 @@
     "frontend",
     "django_extensions",
     "django.contrib.postgres",
+    "storages",
 ]
 
 SHELL_PLUS_SUBCLASSES_IMPORT = [ModelSerializer, Serializer, DjangoModelFactory]
@@ -174,11 +177,33 @@
 AWS_S3_VERIFY = True
 AWS_QUERYSTRING_AUTH = False  # public bucket
 DEFAULT_FILE_STORAGE = "storages.backends.s3boto3.S3Boto3Storage"
+PROFILE_IMAGE_STORAGE = "csm_web.settings.ProfileImageStorage"
 
 # Static files (CSS, JavaScript, Images)
 # https://docs.djangoproject.com/en/2.1/howto/static-files/
 
 STATIC_URL = "/static/"
+# Do I need this?
+# MEDIA_URL = "/media/"
+
+
+# xTODO: make sure this actually works
+class ProfileImageStorage(S3Boto3Storage):
+    bucket_name = "csm-web-profile-pictures"
+    file_overwrite = False
+
+    def get_accessed_time(self, name):
+        # Implement logic to get the last accessed time
+        return datetime.now()
+
+    def get_created_time(self, name):
+        # Implement logic to get the creation time
+        return datetime.now()
+
+    def path(self, name):
+        # S3 does not support file paths
+        raise NotImplementedError("This backend does not support absolute paths.")
+
 
 if DJANGO_ENV in (PRODUCTION, STAGING):
     # Enables compression and caching

csm_web/scheduler/models.py

@@ -11,6 +11,8 @@
 from django.utils import functional, timezone
 from rest_framework.serializers import ValidationError
 
+from csm_web.settings import ProfileImageStorage
+
 logger = logging.getLogger(__name__)
 
 logger.info = logger.warning
@@ -53,9 +55,8 @@ class User(AbstractUser):
 
     pronouns = models.CharField(max_length=20, default="", blank=True)
     pronunciation = models.CharField(max_length=50, default="", blank=True)
-    profile_image = models.ImageField(
-        upload_to="csm-profile-images/", null=True, blank=True
-    )
+    # the upload_to field specifies subdivisions within S3 bucket
+    profile_image = models.ImageField(storage=ProfileImageStorage())
     bio = models.CharField(max_length=500, default="", blank=True)
 
     def can_enroll_in_course(self, course, bypass_enrollment_time=False):