Merge pull request #203 from PROCOLLAB-github/dev

Миграция по сжатию аватарок

Author: VeryBigSad

README.md

@@ -26,7 +26,6 @@
 ### Run project
 
 🚀 Run project via `python manage.py runserver`
-
 ## For developers
 
 ### Install pre-commit hooks

files/migrations/0006_alter_userfile_options.py

@@ -0,0 +1,21 @@
+# Generated by Django 4.2.3 on 2023-09-30 09:44
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("files", "0005_alter_userfile_options"),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name="userfile",
+            options={
+                "ordering": ["datetime_uploaded"],
+                "verbose_name": "Файл",
+                "verbose_name_plural": "Файлы",
+            },
+        ),
+    ]

files/migrations/0007_auto_20230929_1727.py

@@ -0,0 +1,85 @@
+# Generated by Django 4.2.3 on 2023-09-29 14:27
+
+from io import BytesIO
+from django.db import migrations
+import requests
+import webp
+from typing import Optional
+
+from PIL import Image
+
+from files.service import SelectelSwiftStorage
+
+
+def convert_image_to_webp(pil_image: Image, quality: int) -> Optional[BytesIO]:
+    """
+    Convert a PIL Image to WebP format.
+
+    Parameters:
+    - pil_image (PIL.Image): A PIL Image object to be converted.
+    - quality (int): The quality of the converted WebP image, 0 to 100.
+
+    Returns:
+    - bytes: The converted WebP image in bytes, or None if the conversion fails.
+    """
+    try:
+        config = webp.WebPConfig.new(preset=webp.WebPPreset.PHOTO, quality=quality)
+        webp_data = webp.WebPPicture.from_pil(pil_image).encode(config=config)
+        return BytesIO(webp_data.buffer())
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return None
+
+
+def migration(apps, schema_editor):
+    UserFile = apps.get_model("files", "UserFile")
+    files = UserFile.objects.all()
+    storage = SelectelSwiftStorage()
+    for i in files:
+        # get all files that end in a .png or .jpg
+        if i.link.endswith(".png") or i.link.endswith(".jpg"):
+            old_link = i.link
+            # convert .png/.jpg to .webp
+            # download file and convert it to PIL image
+            file = requests.get(i.link)
+            pil_image = Image.open(BytesIO(file.content))
+            # make image be 512 by 512. cut coreners if needed
+            # if hasattr(pil_image, '_getexif'):
+            #     # noinspection PyProtectedMember
+            #     orientation = pil_image._getexif().get(0x112, 1)
+            #     if orientation in (3, 6, 8):
+            #         pil_image = pil_image.rotate(90 * (orientation - 2))
+            
+            pil_image.thumbnail((512, 512), Image.ANTIALIAS)
+            # convert image to webp
+            webp_file = convert_image_to_webp(pil_image, quality=80)
+            # delete old file
+            storage.delete(i.link)
+
+            new_url = str(i.link).rsplit(".", 1)[0] + ".webp"
+            token = storage._get_auth_token()
+            requests.put(
+                new_url,
+                headers={
+                    "X-Auth-Token": token,
+                    "Content-Type": "image/webp",
+                },
+                data=webp_file,
+            )
+            i.link = new_url
+            i.extension = "webp"
+            i.mime_type = "image/webp"
+            i.save()
+            # delete the one with the old url
+            UserFile.objects.get(link=old_link).delete()
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("files", "0006_alter_userfile_options"),
+    ]
+
+    operations = [
+        # FIXME
+        # migrations.RunPython(migration, migrations.RunPython.noop),
+    ]

mailing/__init__.py

@@ -0,0 +1,18 @@
+from django.contrib import admin
+from .models import MailingSchema
+
+
+@admin.register(MailingSchema)
+class MailingSchemaAdmin(admin.ModelAdmin):
+    list_display = (
+        "id",
+        "name",
+    )
+    list_display_links = (
+        "id",
+        "name",
+    )
+    search_fields = (
+        "id",
+        "name",
+    )

mailing/admin.py

@@ -0,0 +1,7 @@
+from django.apps import AppConfig
+
+
+class MailingConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "mailing"
+    verbose_name = "Рассылка"

mailing/apps.py

@@ -0,0 +1,9 @@
+def get_default_mailing_schema() -> dict[str, dict[str, str]]:
+    return {
+        "title": {
+            "title": "Заголовок письма",
+            "default": "Рассылка | Procollab",
+        },
+        "text": {"title": "Основной текст письма"},
+        "button_text": {"title": "Текст кнопки", "default": "Кнопка"},
+    }

mailing/constants.py

@@ -0,0 +1,42 @@
+# Generated by Django 4.2.3 on 2023-09-30 09:44
+
+import django.core.validators
+from django.db import migrations, models
+import mailing.models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = []
+
+    operations = [
+        migrations.CreateModel(
+            name="MailingSchema",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("name", models.CharField(max_length=100)),
+                ("schema", models.JSONField()),
+                (
+                    "template",
+                    models.FileField(
+                        upload_to=mailing.models.get_template_path,
+                        validators=[
+                            django.core.validators.FileExtensionValidator(
+                                allowed_extensions=["html"]
+                            )
+                        ],
+                    ),
+                ),
+            ],
+        ),
+    ]

mailing/migrations/0001_initial.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.3 on 2023-09-30 09:49
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("mailing", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="mailingschema",
+            name="schema",
+            field=models.JSONField(default=None, null=True),
+        ),
+    ]

mailing/migrations/0002_alter_mailingschema_schema.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.3 on 2023-09-30 09:50
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("mailing", "0002_alter_mailingschema_schema"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="mailingschema",
+            name="schema",
+            field=models.JSONField(blank=True, default="", null=True),
+        ),
+    ]