✨ finds: change collection to m2m, add owner and ownership (#6082)

1 files changed, 29 insertions, 0 deletions

diff --git a/archaeological_finds/migrations/0139_data_migration_owner_ownership.py b/archaeological_finds/migrations/0139_data_migration_owner_ownership.py
new file mode 100644
index 000000000..9ce23e697
--- /dev/null
+++ b/archaeological_finds/migrations/0139_data_migration_owner_ownership.py
@@ -0,0 +1,29 @@
+import os
+
+from django.db import migrations
+from django.core.management import call_command
+
+
+def load_data(apps, __):
+    OwnershipStatus = apps.get_model("archaeological_finds", "ownershipstatus")
+    if not OwnershipStatus.objects.count():
+        migration = "0139_data_migration_ownership_status.json"
+        json_path = os.sep.join(os.path.abspath(__file__).split(os.sep)[:-1] + [migration])
+        call_command("loaddata", json_path)
+
+    OwnerType = apps.get_model("archaeological_finds", "ownertype")
+    if not OwnerType.objects.count():
+        migration = "0139_data_migration_owner_type.json"
+        json_path = os.sep.join(os.path.abspath(__file__).split(os.sep)[:-1] + [migration])
+        call_command("loaddata", json_path)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_finds', '0138_donors_owner_ownership_status_collections'),
+    ]
+
+    operations = [
+        migrations.RunPython(load_data)
+    ]