feat: added python code in migrations to generate objects.

bmstu-itstech · Jul 21, 2024 · 57cad35 · 57cad35
1 parent 0473368
commit 57cad35
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/rs_back/achievement/admin.py b/rs_back/achievement/admin.py
@@ -47,10 +47,3 @@ class AchievementOrderAdmin(admin.ModelAdmin):
     readonly_fields = ('order',)
     form = AchievementOrderForm
     search_fields = ('order', 'achievement',)
-
-    def __init__(self, model, admin_site):
-        try:
-            AchievementOrder.generate()
-        except OperationalError:
-            pass
-        super().__init__(model, admin_site)
diff --git a/rs_back/achievement/migrations/0008_alter_achievementorder_achievement_and_more.py b/rs_back/achievement/migrations/0008_alter_achievementorder_achievement_and_more.py
@@ -5,6 +5,19 @@
 import django.db.models.deletion
 
 
+def forward_func(apps, schema_editor):
+    AchievementOrder = apps.get_model('achievement', 'AchievementOrder')
+    db_alias = schema_editor.connection.alias
+    AchievementOrder.objects.using(db_alias).bulk_create(
+        [
+            AchievementOrder(achievement=None, order=1),
+            AchievementOrder(achievement=None, order=2),
+            AchievementOrder(achievement=None, order=3),
+            AchievementOrder(achievement=None, order=4),
+        ]
+    )
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
@@ -22,4 +35,5 @@ class Migration(migrations.Migration):
             name='order',
             field=models.IntegerField(validators=[django.core.validators.MinValueValidator(1)], verbose_name='порядок'),
         ),
+        migrations.RunPython(forward_func),
     ]