Manage X, Y, Z fields for context records, operations and sites

1 files changed, 27 insertions, 0 deletions

diff --git a/archaeological_operations/migrations/0047_point_to_point_2d.py b/archaeological_operations/migrations/0047_point_to_point_2d.py
new file mode 100644
index 000000000..7750fb88a
--- /dev/null
+++ b/archaeological_operations/migrations/0047_point_to_point_2d.py
@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2019-01-22 15:54
+from __future__ import unicode_literals
+
+from django.db import migrations
+
+
+def point_to_point_2d(apps, schema_editor):
+    ArchaeologicalSite = apps.get_model(
+        'archaeological_operations', 'ArchaeologicalSite')
+    Operation = apps.get_model('archaeological_operations', 'Operation')
+    for kls in [ArchaeologicalSite, Operation]:
+        for item in kls.objects.exclude(point__isnull=True).all():
+            item.point_2d = item.point
+            item.point = None
+            item.save()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_operations', '0046_auto_20190122_1550'),
+    ]
+
+    operations = [
+        migrations.RunPython(point_to_point_2d)
+    ]