Merge pull request #9 from devind-team/6-использование-специфичных-для-postgres-типов

6 использование специфичных для postgres типов

Author: SquakR

devind_notifications/migrations/0001_initial.py

@@ -1,6 +1,5 @@
 # Generated by Django 3.2.12 on 2022-02-09 23:14
 
-import django.contrib.postgres.fields
 import django.db.models.deletion
 from django.conf import settings
 from django.db import migrations, models
@@ -49,11 +48,11 @@ class Migration(migrations.Migration):
             name='Mailing',
             fields=[
                 ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('dispatchers', django.contrib.postgres.fields.ArrayField(base_field=models.CharField(max_length=30), default=devind_notifications.models.mailing.default_dispatchers, help_text='Средства отправки', size=None)),
+                ('dispatchers', models.JSONField(default=list, help_text='Массив прикрепленных файлов', null=True)),
                 ('address', models.EmailField(default=devind_notifications.models.mailing.mailing_default_email, help_text='Адрес отправки', max_length=254)),
                 ('header', models.CharField(help_text='Заголовок сообщения', max_length=256)),
                 ('text', models.TextField(help_text='Текст сообщения')),
-                ('attachments', django.contrib.postgres.fields.ArrayField(base_field=models.FileField(help_text='Файлы', max_length=1024, upload_to=devind_notifications.models.mailing.attachments_directory_path), help_text='Массив прикрепленных файлов', null=True, size=None)),
+                ('attachments', models.JSONField(default=devind_notifications.models.mailing.default_dispatchers, help_text='Средства отправки')),
                 ('created_at', models.DateTimeField(auto_now_add=True, help_text='Дата добавления')),
                 ('user', models.ForeignKey(help_text='Пользователь', on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
             ],

devind_notifications/migrations/0002_auto_20220401_1155.py

@@ -0,0 +1,20 @@
+# Generated by Django 3.2.12 on 2022-04-01 08:55
+
+from django.db import migrations
+
+
+def forward(_, schema_editor):  # Чтобы не править руками боевые бд
+    if schema_editor.connection.vendor.startswith('postgres'):
+        schema_editor.execute('alter table devind_notifications_mailing alter column dispatchers type jsonb using to_jsonb(dispatchers);')
+        schema_editor.execute('alter table devind_notifications_mailing alter column attachments type jsonb using to_jsonb(attachments);')
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('devind_notifications', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.RunPython(forward, lambda *_: None)
+    ]

devind_notifications/models/mailing.py

@@ -1,7 +1,6 @@
 import importlib
 
 from django.conf import settings
-from django.contrib.postgres.fields import ArrayField
 from django.db import models
 
 
@@ -43,16 +42,15 @@ class AbstractMailing(models.Model):
         - TelegramDispatch - отправка по средства телеграмма
     """
 
-    dispatchers = ArrayField(
-        models.CharField(max_length=30),
+    dispatchers = models.JSONField(
         default=default_dispatchers,
         help_text='Средства отправки'
     )
     address = models.EmailField(default=mailing_default_email, help_text='Адрес отправки')
     header = models.CharField(max_length=256, help_text='Заголовок сообщения')
     text = models.TextField(help_text='Текст сообщения')
-    attachments = ArrayField(
-        models.FileField(upload_to=attachments_directory_path, max_length=1024, help_text='Файлы'),
+    attachments = models.JSONField(
+        default=list,
         help_text='Массив прикрепленных файлов',
         null=True
     )