1 files changed, 29 insertions, 0 deletions
diff --git a/archaeological_finds/migrations/0125_data_migration_communicability_functionalarea.py b/archaeological_finds/migrations/0125_data_migration_communicability_functionalarea.py
new file mode 100644
index 000000000..3734e46d5
--- /dev/null
+++ b/archaeological_finds/migrations/0125_data_migration_communicability_functionalarea.py
@@ -0,0 +1,29 @@
+import os
+
+from django.db import migrations
+from django.core.management import call_command
+
+
+def load_data(apps, __):
+    CommunicabilityType = apps.get_model("archaeological_finds", "communicabilitytype")
+    if not CommunicabilityType.objects.count():
+        migration = "0125_data_migration_communicability.json"
+        json_path = os.sep.join(os.path.abspath(__file__).split(os.sep)[:-1] + [migration])
+        call_command("loaddata", json_path)
+
+    FunctionalArea = apps.get_model("archaeological_finds", "functionalarea")
+    if not FunctionalArea.objects.count():
+        migration = "0125_data_migration_functionalarea.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', '0124_verbose_names'),
+    ]
+
+    operations = [
+        migrations.RunPython(load_data)
+    ]