refactor(db): combine migration into 1 and add migrate_data()

cal-itp · Sep 30, 2024 · d1e2225 · d1e2225
1 parent b2876ea
commit d1e2225
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 22 deletions.
diff --git a/...nsitagency_enrollmentflow_relationship.py → ...ransitagency_enrollment_flows_and_more.py b/...nsitagency_enrollmentflow_relationship.py → ...ransitagency_enrollment_flows_and_more.py
@@ -1,12 +1,11 @@
-# Generated by Django 5.1.1 on 2024-09-25 16:47
+# Generated by Django 5.1 on 2024-09-30 23:25
 
 import django.db.models.deletion
 from django.db import migrations, models
 
 
 def migrate_data(apps, schema_editor):
     TransitAgency = apps.get_model("core", "TransitAgency")
-
     for agency in TransitAgency.objects.all():
         for flow in agency.enrollment_flows.all():
             flow.transit_agency = agency
@@ -26,11 +25,15 @@ class Migration(migrations.Migration):
             field=models.ForeignKey(
                 blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, to="core.transitagency"
             ),
-            preserve_default=False,
         ),
         migrations.RunPython(migrate_data),
         migrations.RemoveField(
             model_name="transitagency",
             name="enrollment_flows",
         ),
+        migrations.AlterField(
+            model_name="enrollmentflow",
+            name="label",
+            field=models.TextField(help_text="A human readable label, used as the display text in Admin.", null=True),
+        ),
     ]
diff --git a/benefits/core/migrations/0029_alter_enrollmentflow_label.py b/benefits/core/migrations/0029_alter_enrollmentflow_label.py
diff --git a/benefits/core/models.py b/benefits/core/models.py
@@ -381,7 +381,7 @@ class EnrollmentFlow(models.Model):
         default=[EnrollmentMethods.DIGITAL, EnrollmentMethods.IN_PERSON],
         help_text="If the flow is supported by digital enrollment, in-person enrollment, or both",
     )
-    transit_agency = models.ForeignKey(TransitAgency, on_delete=models.PROTECT)
+    transit_agency = models.ForeignKey(TransitAgency, on_delete=models.PROTECT, null=True, blank=True)
 
     class Meta:
         ordering = ["display_order"]