1 files changed, 23 insertions, 0 deletions
diff --git a/archaeological_finds/migrations/0130_data_discovery_method.py b/archaeological_finds/migrations/0130_data_discovery_method.py
new file mode 100644
index 000000000..56a1efb1f
--- /dev/null
+++ b/archaeological_finds/migrations/0130_data_discovery_method.py
@@ -0,0 +1,23 @@
+import os
+
+from django.db import migrations
+from django.core.management import call_command
+
+
+def load_data(apps, __):
+    DiscoveryMethod = apps.get_model("archaeological_finds", "discoverymethod")
+    if not DiscoveryMethod.objects.count():
+        migration = "0130_data_discovery_method.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', '0129_discovery_method'),
+    ]
+
+    operations = [
+        migrations.RunPython(load_data)
+    ]