Merge pull request #21 from PROCOLLAB-github/feature/files

Feature/files

Author: yakser

.env.example

@@ -12,4 +12,9 @@ DATABASE_NAME=
 DATABASE_USER=
 DATABASE_PASSWORD=
 DATABASE_HOST=
-DATABASE_PORT=
+DATABASE_PORT=
+
+SELECTEL_ACCOUNT_ID=
+SELECTEL_CONTAINER_NAME=
+SELECTEL_CONTAINER_PASSWORD=
+SELECTEL_CONTAINER_USERNAME=

files/__init__.py

@@ -0,0 +1,13 @@
+from django.contrib import admin
+
+from files.models import UserFile
+
+
+@admin.register(UserFile)
+class UserFileAdmin(admin.ModelAdmin):
+    list_display = (
+        "link",
+        "user",
+        "datetime_uploaded",
+    )
+    list_display_links = ("link",)

files/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class FilesConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "files"

files/apps.py

@@ -0,0 +1,2 @@
+class SelectelUploadError(Exception):
+    pass

files/exceptions.py

@@ -0,0 +1,86 @@
+import requests
+import time
+from files.exceptions import SelectelUploadError
+
+from procollab.settings import (
+    DEBUG,
+    SELECTEL_ACCOUNT_ID,
+    SELECTEL_CONTAINER_NAME,
+    SELECTEL_CONTAINER_PASSWORD,
+    SELECTEL_CONTAINER_USERNAME,
+)
+
+
+class FileAPI:
+    def __init__(self, file, user) -> None:
+        self.file = file
+        self.user = user
+
+    @staticmethod
+    def delete(url: str) -> int:
+        """Deletes file from selcdn"""
+        token = FileAPI._get_selectel_swift_token()
+        response = requests.delete(url, headers={"X-Auth-Token": token})
+        return response.status_code
+
+    def upload(self):
+        return self._upload_via_selectel_swift()
+
+    def _upload_via_selectel_swift(self) -> tuple[int, str]:
+        token = self._get_selectel_swift_token()
+        url = self._generate_selectel_swift_file_url()
+
+        with self.file.open(mode="rb") as file_object:
+            response = requests.put(
+                url,
+                headers={"X-Auth-Token": token, "Content-Type": file_object.content_type},
+                data=file_object.read(),
+            )
+
+        return response.status_code, url
+
+    def _generate_selectel_swift_link(sefl):
+        link = f"https://api.selcdn.ru/v1/SEL_{SELECTEL_ACCOUNT_ID}/{SELECTEL_CONTAINER_NAME}/"
+        if DEBUG:
+            link += "debug/"
+        return link
+
+    @staticmethod
+    def _get_selectel_swift_token():
+        """Returns auth token for selcdn"""
+        data = {
+            "auth": {
+                "identity": {
+                    "methods": ["password"],
+                    "password": {
+                        "user": {
+                            "id": SELECTEL_CONTAINER_USERNAME,
+                            "password": SELECTEL_CONTAINER_PASSWORD,
+                        }
+                    },
+                }
+            }
+        }
+        r = requests.post("https://api.selcdn.ru/v3/auth/tokens", json=data)
+        if r.status_code not in [200, 201]:
+            raise SelectelUploadError("Couldn't generate a token for selcdn")
+        return r.headers["x-subject-token"]
+
+    def _get_file_extension(self) -> str:
+        if len(self.file.name.split(".")) > 1:
+            return "." + self.file.name.split(".")[1]
+        return ""
+
+    def _generate_selectel_swift_file_url(self) -> str:
+        """
+        Generates url for selcdn
+
+        Returns:
+            url: str looks like /hashedEmail/hashedFilename_hashedTime.extension
+        """
+        link = self._generate_selectel_swift_link()
+        extension = self._get_file_extension()
+        return (
+            link
+            + f"{abs(hash(self.user.email))}/{abs(hash(self.file.name))}_{abs(hash(time.time()))}{extension}"
+        )

files/helpers.py

@@ -0,0 +1,41 @@
+# Generated by Django 4.1.2 on 2022-11-11 22:37
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="UserFile",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("link", models.URLField()),
+                ("datetime_uploaded", models.DateTimeField(auto_now_add=True)),
+                (
+                    "user",
+                    models.ForeignKey(
+                        null=True,
+                        on_delete=django.db.models.deletion.SET_NULL,
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
+            ],
+        ),
+    ]

files/migrations/0001_initial.py

@@ -0,0 +1,22 @@
+# Generated by Django 4.1.2 on 2022-11-12 13:56
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("files", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="userfile",
+            name="id",
+        ),
+        migrations.AlterField(
+            model_name="userfile",
+            name="link",
+            field=models.URLField(primary_key=True, serialize=False),
+        ),
+    ]

files/migrations/0002_remove_userfile_id_alter_userfile_link.py

@@ -0,0 +1,22 @@
+from django.contrib.auth import get_user_model
+from django.db import models
+
+User = get_user_model()
+
+
+class UserFile(models.Model):
+    """
+    UserFile model
+
+    Attributes:
+        user: User who uploaded the file
+        link: Link to the file on the CDN
+        datetime_uploaded: Datetime when the file was uploaded
+    """
+
+    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)
+    link = models.URLField(primary_key=True, null=False)
+    datetime_uploaded = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f"UserFile by {self.user}, {self.link}"