diff options
Diffstat (limited to 'archaeological_finds/migrations/0139_data_migration_owner_ownership.py')
-rw-r--r-- | archaeological_finds/migrations/0139_data_migration_owner_ownership.py | 29 |
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) + ] |