feat: add private media storage

chats/apps/discussions/migrations/0002_alter_discussionmessagemedia_media_file.py

@@ -0,0 +1,26 @@
+# Generated by Django 4.1.2 on 2024-10-17 18:34
+
+import chats.core.storage_backends
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("discussions", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="discussionmessagemedia",
+            name="media_file",
+            field=models.FileField(
+                blank=True,
+                max_length=300,
+                null=True,
+                storage=chats.core.storage_backends.PrivateMediaStorage(),
+                upload_to="discussionmedia/%Y/%m/%d/",
+                verbose_name="Media File",
+            ),
+        ),
+    ]

chats/apps/discussions/models/discussion_message.py

@@ -3,6 +3,7 @@
 from django.utils.translation import gettext_lazy as _
 
 from chats.core.models import BaseModel, WebSocketsNotifiableMixin
+from chats.core.storage_backends import PrivateMediaStorage
 
 
 class DiscussionMessage(WebSocketsNotifiableMixin, BaseModel):
@@ -65,6 +66,7 @@ class DiscussionMessageMedia(BaseModel):
         blank=True,
         max_length=300,
         upload_to="discussionmedia/%Y/%m/%d/",
+        storage=PrivateMediaStorage(),
     )
 
     class Meta:

chats/apps/msgs/migrations/0008_alter_messagemedia_media_file.py

@@ -0,0 +1,26 @@
+# Generated by Django 4.1.2 on 2024-10-17 18:34
+
+import chats.core.storage_backends
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("msgs", "0007_alter_message_options"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="messagemedia",
+            name="media_file",
+            field=models.FileField(
+                blank=True,
+                max_length=300,
+                null=True,
+                storage=chats.core.storage_backends.PrivateMediaStorage(),
+                upload_to="",
+                verbose_name="Media File",
+            ),
+        ),
+    ]

chats/apps/msgs/models.py

@@ -8,6 +8,7 @@
 
 from chats.core.models import BaseModel
 from chats.core.requests import get_request_session_with_retries
+from chats.core.storage_backends import PrivateMediaStorage
 
 
 class Message(BaseModel):
@@ -114,7 +115,11 @@ class MessageMedia(BaseModel):
     )
     content_type = models.CharField(_("Content Type"), max_length=300)
     media_file = models.FileField(
-        _("Media File"), null=True, blank=True, max_length=300
+        _("Media File"),
+        null=True,
+        blank=True,
+        max_length=300,
+        storage=PrivateMediaStorage(),
     )
     media_url = models.TextField(_("Media url"), null=True, blank=True)
 

chats/core/excel_storage.py

@@ -2,6 +2,11 @@
 
 
 class ExcelStorage(S3Boto3Storage):  # pragma: no cover
+    location = "private"
+    default_acl = "private"
+    file_overwrite = False
+    custom_domain = False
+
     def get_default_settings(self):
         default_settings = super().get_default_settings()
         default_settings["location"] = "dashboard_data/excel"

chats/core/storage_backends.py

@@ -0,0 +1,8 @@
+from storages.backends.s3boto3 import S3Boto3Storage
+
+
+class PrivateMediaStorage(S3Boto3Storage):
+    location = "private"
+    default_acl = "private"
+    file_overwrite = False
+    custom_domain = False