🗃️ sites - models migrations: heritage, relations and datings fields

1 files changed, 23 insertions, 0 deletions

diff --git a/archaeological_operations/migrations/0127_data_migration_current_states.py b/archaeological_operations/migrations/0127_data_migration_current_states.py
new file mode 100644
index 000000000..fe1aa5c25
--- /dev/null
+++ b/archaeological_operations/migrations/0127_data_migration_current_states.py
@@ -0,0 +1,23 @@
+# Generated by Django 4.2.19 on 2026-03-27 12:06
+
+from django.db import migrations
+
+
+def migrate_current_states(apps, __):
+    ArchaeologicalSite = apps.get_model("archaeological_operations", "archaeologicalsite")
+    if not hasattr(ArchaeologicalSite, "current_status"):
+        return
+    q = ArchaeologicalSite.objects.filter(current_status__isnull=False)
+    for site in q.all():
+        site.current_states.add(site.current_status)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_operations', '0126_archaeologicalsite_heritage_relations_datings'),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate_current_states)
+    ]