Geodata redesign: site migration

1 files changed, 93 insertions, 73 deletions

diff --git a/ishtar_common/management/commands/migrate_to_geo_v4.py b/ishtar_common/management/commands/migrate_to_geo_v4.py
index 8daf921d9..ed1a877f3 100644
--- a/ishtar_common/management/commands/migrate_to_geo_v4.py
+++ b/ishtar_common/management/commands/migrate_to_geo_v4.py
@@ -59,84 +59,104 @@ def migrate(quiet=False, log=True):
         sys.stdout.write(f"\r[{get_time()}] Towns migrated\n")
         sys.stdout.flush()
 
-    # manage operation vector sources
-    operation_content_type = ContentType.objects.get(
-        app_label="archaeological_operations", model="operation"
-    )
-    q = Operation.objects.exclude(main_geodata__isnull=False)
-    nb = q.count()
-    data_type_area, __ = models_common.GeoDataType.objects.get_or_create(
-        txt_idx="operation-area", defaults={"label": "Emprise de l'opération"}
-    )
-    data_type_center, __ = models_common.GeoDataType.objects.get_or_create(
-        txt_idx="operation-center", defaults={"label": "Centre de l'opération"}
-    )
-    for idx, operation in enumerate(q.all()):
-        if not quiet:
-            sys.stdout.write(f"\r[{percent(idx, nb)}] Migrate operations {idx + 1}/{nb}")
-            sys.stdout.flush()
-
-        operation._no_move = True
-        operation.skip_history_when_saving = True
-        operation.save()  # auto manage geo town association
-        q_towns = operation.towns.filter(main_geodata__multi_polygon__isnull=False)
-        if q_towns.count() > 1:
-            changed.append(
-                ["operation", str(operation), operation.pk,
-                 "Association géo de zone communale"])
-        elif q_towns.count() == 1:
-            changed.append(
-                ["operation", str(operation), operation.pk,
-                 "Association géo de commune"])
-        if operation.multi_polygon_source == "P" and operation.multi_polygon:
-            attrs = {
-                "name": f"{_('Operation')}{_(':')} {str(operation)}",
-                "source_content_type": operation_content_type,
-                "source_id": operation.pk,
-                "multi_polygon": operation.multi_polygon,
-                "data_type": data_type_area,
-            }
-            data = models_common.GeoVectorData.objects.create(**attrs)
-            operation.main_geodata = data
-            operation.save()
-            changed.append(
-                    ["geovectordata", data.name, data.pk, "Multi-polygone opération"])
-        if operation.point_source == "P" and operation.point_2d:
-            if operation.x and operation.y:
-                attrs = {
-                    "name": f"{_('Operation')}{_(':')} {str(operation)}",
-                    "source_content_type": operation_content_type,
-                    "source_id": operation.pk,
-                    "data_type": data_type_center,
-                    "x": operation.x,
-                    "y": operation.y,
-                    "z": operation.z,
-                }
-                data = models_common.GeoVectorData.objects.create(**attrs)
-                operation.main_geodata = data
-                operation.save()
+    model_list = [
+        ("operation", "opération", "de l'opération", Operation),
+        ("archaeologicalsite", "site", "du site", ArchaeologicalSite),
+    ]
+    for model_slug, model_name, model_full_name, model in model_list:
+        # manage operation vector sources
+        model_content_type = ContentType.objects.get(
+            app_label="archaeological_operations", model=model_slug
+        )
+        q = model.objects.exclude(main_geodata__isnull=False)
+        nb = q.count()
+        data_type_area, __ = models_common.GeoDataType.objects.get_or_create(
+            txt_idx=f"{model_slug}-area",
+            defaults={"label": f"Emprise {model_full_name}"},
+        )
+        data_type_center, __ = models_common.GeoDataType.objects.get_or_create(
+            txt_idx="operation-center", defaults={"label": f"Centre {model_full_name}"}
+        )
+        for idx, obj in enumerate(q.all()):
+            if not quiet:
+                sys.stdout.write(
+                    f"\r[{percent(idx, nb)}] Migrate {model_name}s {idx + 1}/{nb}"
+                )
+                sys.stdout.flush()
+
+            obj._no_move = True
+            obj.skip_history_when_saving = True
+            obj.save()  # auto manage geo town association
+            q_towns = obj.towns.filter(main_geodata__multi_polygon__isnull=False)
+            if q_towns.count() > 1:
+                changed.append(
+                    [model_slug, str(obj), obj.pk, "Association géo de zone communale"]
+                )
+            elif q_towns.count() == 1:
                 changed.append(
-                    ["geovectordata", data.name, data.pk, "Coordonnées opération"])
-            elif operation.point_2d:
+                    [model_slug, str(obj), obj.pk, "Association géo de commune"]
+                )
+            if obj.multi_polygon_source == "P" and obj.multi_polygon:
                 attrs = {
-                    "name": f"{_('Operation')}{_(':')} {str(operation)}",
-                    "source_content_type": operation_content_type,
-                    "source_id": operation.pk,
-                    "data_type": data_type_center,
+                    "name": f"{_(model_name.capitalize())}{_(':')} {str(obj)}",
+                    "source_content_type": model_content_type,
+                    "source_id": obj.pk,
+                    "multi_polygon": obj.multi_polygon,
+                    "data_type": data_type_area,
                 }
-                if operation.point:
-                    attrs["point_3d"] = operation.point
-                else:
-                    attrs["point_2d"] = operation.point_2d
                 data = models_common.GeoVectorData.objects.create(**attrs)
-                operation.main_geodata = data
-                operation.save()
+                obj.main_geodata = data
+                obj.save()
                 changed.append(
-                    ["geovectordata", data.name, data.pk, "Point opération"])
-    if not quiet and nb:
-        sys.stdout.write(f"\r[{get_time()}] Operation migrated\n")
-        sys.stdout.flush()
-
+                    [
+                        "geovectordata",
+                        data.name,
+                        data.pk,
+                        f"Multi-polygone {model_name}",
+                    ]
+                )
+            if obj.point_source == "P" and obj.point_2d:
+                if obj.x and obj.y:
+                    attrs = {
+                        "name": f"{_(model_name.capitalize())}{_(':')} {str(obj)}",
+                        "source_content_type": model_content_type,
+                        "source_id": obj.pk,
+                        "data_type": data_type_center,
+                        "x": obj.x,
+                        "y": obj.y,
+                        "z": obj.z,
+                    }
+                    data = models_common.GeoVectorData.objects.create(**attrs)
+                    obj.main_geodata = data
+                    obj.save()
+                    changed.append(
+                        [
+                            "geovectordata",
+                            data.name,
+                            data.pk,
+                            f"Coordonnées {model_name}",
+                        ]
+                    )
+                elif obj.point_2d:
+                    attrs = {
+                        "name": f"{_(model_name.capitalize())}{_(':')} {str(obj)}",
+                        "source_content_type": model_content_type,
+                        "source_id": obj.pk,
+                        "data_type": data_type_center,
+                    }
+                    if obj.point:
+                        attrs["point_3d"] = obj.point
+                    else:
+                        attrs["point_2d"] = obj.point_2d
+                    data = models_common.GeoVectorData.objects.create(**attrs)
+                    obj.main_geodata = data
+                    obj.save()
+                    changed.append(
+                        ["geovectordata", data.name, data.pk, f"Point {model_name}"]
+                    )
+        if not quiet and nb:
+            sys.stdout.write(f"\r[{get_time()}] {model_name.capitalize()} migrated\n")
+            sys.stdout.flush()
 
     if log and changed:
         filename = f"geo_migration-created-{get_time().replace(':', '')}.csv"