Add location model to use json field instead of postgis field

temba/locations/migrations/0033_location_locationalias_location_locations_by_name_and_more.py

@@ -0,0 +1,108 @@
+# Generated by Django 5.1.4 on 2025-01-09 08:50
+
+import mptt.fields
+
+import django.db.models.deletion
+import django.db.models.functions.text
+import django.utils.timezone
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("locations", "0032_squashed"),
+        ("orgs", "0163_squashed"),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Location",
+            fields=[
+                ("id", models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                ("osm_id", models.CharField(max_length=15, unique=True)),
+                ("name", models.CharField(max_length=128)),
+                ("level", models.IntegerField()),
+                ("path", models.CharField(max_length=768)),
+                ("simplified_geometry", models.JSONField(null=True)),
+                ("lft", models.PositiveIntegerField(editable=False)),
+                ("rght", models.PositiveIntegerField(editable=False)),
+                ("tree_id", models.PositiveIntegerField(db_index=True, editable=False)),
+                (
+                    "parent",
+                    mptt.fields.TreeForeignKey(
+                        null=True,
+                        on_delete=django.db.models.deletion.PROTECT,
+                        related_name="children",
+                        to="locations.location",
+                    ),
+                ),
+            ],
+        ),
+        migrations.CreateModel(
+            name="LocationAlias",
+            fields=[
+                ("id", models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                (
+                    "is_active",
+                    models.BooleanField(
+                        default=True, help_text="Whether this item is active, use this instead of deleting"
+                    ),
+                ),
+                (
+                    "created_on",
+                    models.DateTimeField(
+                        blank=True,
+                        default=django.utils.timezone.now,
+                        editable=False,
+                        help_text="When this item was originally created",
+                    ),
+                ),
+                (
+                    "modified_on",
+                    models.DateTimeField(
+                        blank=True,
+                        default=django.utils.timezone.now,
+                        editable=False,
+                        help_text="When this item was last modified",
+                    ),
+                ),
+                ("name", models.CharField(help_text="The name for our alias", max_length=128)),
+                (
+                    "created_by",
+                    models.ForeignKey(
+                        help_text="The user which originally created this item",
+                        on_delete=django.db.models.deletion.PROTECT,
+                        related_name="%(app_label)s_%(class)s_creations",
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
+                (
+                    "location",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.PROTECT, related_name="aliases", to="locations.location"
+                    ),
+                ),
+                (
+                    "modified_by",
+                    models.ForeignKey(
+                        help_text="The user which last modified this item",
+                        on_delete=django.db.models.deletion.PROTECT,
+                        related_name="%(app_label)s_%(class)s_modifications",
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
+                ("org", models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to="orgs.org")),
+            ],
+        ),
+        migrations.AddIndex(
+            model_name="location",
+            index=models.Index(django.db.models.functions.text.Upper("name"), name="locations_by_name"),
+        ),
+        migrations.AddIndex(
+            model_name="locationalias",
+            index=models.Index(django.db.models.functions.text.Upper("name"), name="locationaliases_by_name"),
+        ),
+    ]

temba/locations/models.py

@@ -178,3 +178,50 @@ def create(cls, org, user, boundary, name):
 
     class Meta:
         indexes = [models.Index(Upper("name"), name="boundaryaliases_by_name")]
+
+
+class Location(MPTTModel, models.Model):
+    """
+    Represents a single administrative locations (like a country, state or district)
+    """
+
+    LEVEL_COUNTRY = 0
+    LEVEL_STATE = 1
+    LEVEL_DISTRICT = 2
+    LEVEL_WARD = 3
+
+    MAX_NAME_LEN = 128
+
+    # Used to separate segments in a hierarchy of locations. Has the advantage of being a character in GSM7 and
+    # being very unlikely to show up in an admin region name.
+    PATH_SEPARATOR = ">"
+    PADDED_PATH_SEPARATOR = " > "
+
+    osm_id = models.CharField(max_length=15, unique=True)
+    name = models.CharField(max_length=MAX_NAME_LEN)
+    level = models.IntegerField()
+    parent = TreeForeignKey("self", null=True, on_delete=models.PROTECT, related_name="children", db_index=True)
+    path = models.CharField(max_length=768)  # e.g. Rwanda > Kigali
+    simplified_geometry = models.JSONField(null=True)
+
+    objects = NoGeometryManager()
+    geometries = GeometryManager()
+
+    def __str__(self):
+        return self.name
+
+    class Meta:
+        indexes = [models.Index(Upper("name"), name="locations_by_name")]
+
+
+class LocationAlias(SmartModel):
+    """
+    An org specific alias for a location name
+    """
+
+    org = models.ForeignKey("orgs.Org", on_delete=models.PROTECT)
+    location = models.ForeignKey(Location, on_delete=models.PROTECT, related_name="aliases")
+    name = models.CharField(max_length=Location.MAX_NAME_LEN, help_text="The name for our alias")
+
+    class Meta:
+        indexes = [models.Index(Upper("name"), name="locationaliases_by_name")]