add geography to collaborative

Deepthi-Chand · Deepthi-Chand · commit f653c23a016d · 2025-10-04T14:35:43.000+05:30
diff --git a/api/management/commands/populate_geographies.py b/api/management/commands/populate_geographies.py
@@ -0,0 +1,304 @@
+"""
+Django management command to populate Asia Pacific geography data.
+Covers: India, Indonesia, Thailand at state/province level.
+
+Usage:
+    python manage.py populate_geographies
+    python manage.py populate_geographies --clear  # Clear existing data first
+"""
+
+from typing import Any
+
+from django.core.management.base import BaseCommand, CommandParser
+from django.db import transaction
+
+from api.models import Geography
+from api.utils.enums import GeoTypes
+
+
+class Command(BaseCommand):
+    help = "Populate Asia Pacific geography data (India, Indonesia, Thailand)"
+
+    def add_arguments(self, parser: CommandParser) -> None:
+        parser.add_argument(
+            "--clear",
+            action="store_true",
+            help="Clear all existing geography data before populating",
+        )
+
+    def handle(self, *args: Any, **options: Any) -> None:
+        clear_existing = options.get("clear", False)
+
+        if clear_existing:
+            self.stdout.write(self.style.WARNING("Clearing existing geography data..."))
+            Geography.objects.all().delete()
+            self.stdout.write(self.style.SUCCESS("✓ Cleared existing data"))
+
+        with transaction.atomic():
+            self._populate_geographies()
+
+        self.stdout.write(
+            self.style.SUCCESS(
+                f"\n✅ Successfully populated {Geography.objects.count()} geography entries"
+            )
+        )
+
+    def _populate_geographies(self) -> None:
+        """Populate geography data in an idempotent way."""
+
+        # Create or get Asia Pacific region
+        asia_pacific, created = Geography.objects.get_or_create(
+            name="Asia Pacific", defaults={"code": "APAC", "type": GeoTypes.REGION}
+        )
+        if created:
+            self.stdout.write(f"Created region: {asia_pacific.name}")
+        else:
+            self.stdout.write(f"Region already exists: {asia_pacific.name}")
+
+        # India and its states
+        self._populate_india(asia_pacific)
+
+        # Indonesia and its provinces
+        self._populate_indonesia(asia_pacific)
+
+        # Thailand and its provinces
+        self._populate_thailand(asia_pacific)
+
+    def _populate_india(self, parent: Geography) -> None:
+        """Populate India and its states/UTs."""
+        india, created = Geography.objects.get_or_create(
+            name="India",
+            defaults={"code": "IN", "type": GeoTypes.COUNTRY, "parent_id": parent},
+        )
+        if created:
+            self.stdout.write(f"Created country: {india.name}")
+        else:
+            self.stdout.write(f"Country already exists: {india.name}")
+
+        indian_states = [
+            ("Andhra Pradesh", "AP"),
+            ("Arunachal Pradesh", "AR"),
+            ("Assam", "AS"),
+            ("Bihar", "BR"),
+            ("Chhattisgarh", "CT"),
+            ("Goa", "GA"),
+            ("Gujarat", "GJ"),
+            ("Haryana", "HR"),
+            ("Himachal Pradesh", "HP"),
+            ("Jharkhand", "JH"),
+            ("Karnataka", "KA"),
+            ("Kerala", "KL"),
+            ("Madhya Pradesh", "MP"),
+            ("Maharashtra", "MH"),
+            ("Manipur", "MN"),
+            ("Meghalaya", "ML"),
+            ("Mizoram", "MZ"),
+            ("Nagaland", "NL"),
+            ("Odisha", "OR"),
+            ("Punjab", "PB"),
+            ("Rajasthan", "RJ"),
+            ("Sikkim", "SK"),
+            ("Tamil Nadu", "TN"),
+            ("Telangana", "TG"),
+            ("Tripura", "TR"),
+            ("Uttar Pradesh", "UP"),
+            ("Uttarakhand", "UT"),
+            ("West Bengal", "WB"),
+            ("Andaman and Nicobar Islands", "AN"),
+            ("Chandigarh", "CH"),
+            ("Dadra and Nagar Haveli and Daman and Diu", "DN"),
+            ("Delhi", "DL"),
+            ("Jammu and Kashmir", "JK"),
+            ("Ladakh", "LA"),
+            ("Lakshadweep", "LD"),
+            ("Puducherry", "PY"),
+        ]
+
+        created_count = 0
+        for state_name, state_code in indian_states:
+            _, created = Geography.objects.get_or_create(
+                name=state_name,
+                defaults={
+                    "code": state_code,
+                    "type": GeoTypes.STATE,
+                    "parent_id": india,
+                },
+            )
+            if created:
+                created_count += 1
+
+        self.stdout.write(
+            f"Indian states/UTs: {created_count} created, {len(indian_states) - created_count} already existed"
+        )
+
+    def _populate_indonesia(self, parent: Geography) -> None:
+        """Populate Indonesia and its provinces."""
+        indonesia, created = Geography.objects.get_or_create(
+            name="Indonesia",
+            defaults={"code": "ID", "type": GeoTypes.COUNTRY, "parent_id": parent},
+        )
+        if created:
+            self.stdout.write(f"Created country: {indonesia.name}")
+        else:
+            self.stdout.write(f"Country already exists: {indonesia.name}")
+
+        indonesian_provinces = [
+            ("Aceh", "AC"),
+            ("Bali", "BA"),
+            ("Banten", "BT"),
+            ("Bengkulu", "BE"),
+            ("Central Java", "JT"),
+            ("Central Kalimantan", "KT"),
+            ("Central Sulawesi", "ST"),
+            ("East Java", "JI"),
+            ("East Kalimantan", "KI"),
+            ("East Nusa Tenggara", "NT"),
+            ("Gorontalo", "GO"),
+            ("Jakarta", "JK"),
+            ("Jambi", "JA"),
+            ("Lampung", "LA"),
+            ("Maluku", "MA"),
+            ("North Kalimantan", "KU"),
+            ("North Maluku", "MU"),
+            ("North Sulawesi", "SA"),
+            ("North Sumatra", "SU"),
+            ("Papua", "PA"),
+            ("Riau", "RI"),
+            ("Riau Islands", "KR"),
+            ("South Kalimantan", "KS"),
+            ("South Sulawesi", "SN"),
+            ("South Sumatra", "SS"),
+            ("Southeast Sulawesi", "SG"),
+            ("West Java", "JB"),
+            ("West Kalimantan", "KB"),
+            ("West Nusa Tenggara", "NB"),
+            ("West Papua", "PB"),
+            ("West Sulawesi", "SR"),
+            ("West Sumatra", "SB"),
+            ("Yogyakarta", "YO"),
+        ]
+
+        created_count = 0
+        for province_name, province_code in indonesian_provinces:
+            _, created = Geography.objects.get_or_create(
+                name=province_name,
+                defaults={
+                    "code": province_code,
+                    "type": GeoTypes.STATE,
+                    "parent_id": indonesia,
+                },
+            )
+            if created:
+                created_count += 1
+
+        self.stdout.write(
+            f"Indonesian provinces: {created_count} created, {len(indonesian_provinces) - created_count} already existed"
+        )
+
+    def _populate_thailand(self, parent: Geography) -> None:
+        """Populate Thailand and its provinces."""
+        thailand, created = Geography.objects.get_or_create(
+            name="Thailand",
+            defaults={"code": "TH", "type": GeoTypes.COUNTRY, "parent_id": parent},
+        )
+        if created:
+            self.stdout.write(f"Created country: {thailand.name}")
+        else:
+            self.stdout.write(f"Country already exists: {thailand.name}")
+
+        thai_provinces = [
+            ("Bangkok", "BKK"),
+            ("Amnat Charoen", "37"),
+            ("Ang Thong", "15"),
+            ("Bueng Kan", "38"),
+            ("Buriram", "31"),
+            ("Chachoengsao", "24"),
+            ("Chai Nat", "18"),
+            ("Chaiyaphum", "36"),
+            ("Chanthaburi", "22"),
+            ("Chiang Mai", "50"),
+            ("Chiang Rai", "57"),
+            ("Chonburi", "20"),
+            ("Chumphon", "86"),
+            ("Kalasin", "46"),
+            ("Kamphaeng Phet", "62"),
+            ("Kanchanaburi", "71"),
+            ("Khon Kaen", "40"),
+            ("Krabi", "81"),
+            ("Lampang", "52"),
+            ("Lamphun", "51"),
+            ("Loei", "42"),
+            ("Lopburi", "16"),
+            ("Mae Hong Son", "58"),
+            ("Maha Sarakham", "44"),
+            ("Mukdahan", "49"),
+            ("Nakhon Nayok", "26"),
+            ("Nakhon Pathom", "73"),
+            ("Nakhon Phanom", "48"),
+            ("Nakhon Ratchasima", "30"),
+            ("Nakhon Sawan", "60"),
+            ("Nakhon Si Thammarat", "80"),
+            ("Nan", "55"),
+            ("Narathiwat", "96"),
+            ("Nong Bua Lamphu", "39"),
+            ("Nong Khai", "43"),
+            ("Nonthaburi", "12"),
+            ("Pathum Thani", "13"),
+            ("Pattani", "94"),
+            ("Phang Nga", "82"),
+            ("Phatthalung", "93"),
+            ("Phayao", "56"),
+            ("Phetchabun", "67"),
+            ("Phetchaburi", "76"),
+            ("Phichit", "66"),
+            ("Phitsanulok", "65"),
+            ("Phra Nakhon Si Ayutthaya", "14"),
+            ("Phrae", "54"),
+            ("Phuket", "83"),
+            ("Prachinburi", "25"),
+            ("Prachuap Khiri Khan", "77"),
+            ("Ranong", "85"),
+            ("Ratchaburi", "70"),
+            ("Rayong", "21"),
+            ("Roi Et", "45"),
+            ("Sa Kaeo", "27"),
+            ("Sakon Nakhon", "47"),
+            ("Samut Prakan", "11"),
+            ("Samut Sakhon", "74"),
+            ("Samut Songkhram", "75"),
+            ("Saraburi", "19"),
+            ("Satun", "91"),
+            ("Sing Buri", "17"),
+            ("Sisaket", "33"),
+            ("Songkhla", "90"),
+            ("Sukhothai", "64"),
+            ("Suphan Buri", "72"),
+            ("Surat Thani", "84"),
+            ("Surin", "32"),
+            ("Tak", "63"),
+            ("Trang", "92"),
+            ("Trat", "23"),
+            ("Ubon Ratchathani", "34"),
+            ("Udon Thani", "41"),
+            ("Uthai Thani", "61"),
+            ("Uttaradit", "53"),
+            ("Yala", "95"),
+            ("Yasothon", "35"),
+        ]
+
+        created_count = 0
+        for province_name, province_code in thai_provinces:
+            _, created = Geography.objects.get_or_create(
+                name=province_name,
+                defaults={
+                    "code": province_code,
+                    "type": GeoTypes.STATE,
+                    "parent_id": thailand,
+                },
+            )
+            if created:
+                created_count += 1
+
+        self.stdout.write(
+            f"Thai provinces: {created_count} created, {len(thai_provinces) - created_count} already existed"
+        )
diff --git a/api/models/Collaborative.py b/api/models/Collaborative.py
@@ -44,6 +44,9 @@ class Collaborative(models.Model):
         "api.Sector", blank=True, related_name="collaboratives"
     )
     sdgs = models.ManyToManyField("api.SDG", blank=True, related_name="collaboratives")
+    geographies = models.ManyToManyField(
+        "api.Geography", blank=True, related_name="collaboratives"
+    )
     contributors = models.ManyToManyField(
         "authorization.User", blank=True, related_name="contributed_collaboratives"
     )
@@ -79,5 +82,9 @@ def tags_indexing(self):
     def sdgs_indexing(self):
         return [sdg.code for sdg in self.sdgs.all()]  # type: ignore
 
+    @property
+    def geographies_indexing(self):
+        return [geo.name for geo in self.geographies.all()]  # type: ignore
+
     class Meta:
         db_table = "collaborative"
diff --git a/api/models/Geography.py b/api/models/Geography.py
@@ -1,3 +1,4 @@
+import uuid
 from typing import Optional
 
 from django.db import models
@@ -6,7 +7,7 @@
 
 
 class Geography(models.Model):
-    id = models.AutoField(primary_key=True)
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
     name = models.CharField(max_length=75, unique=True)
     code = models.CharField(
         max_length=100, null=True, blank=True, unique=False, default=""
diff --git a/api/schema/collaborative_schema.py b/api/schema/collaborative_schema.py
@@ -21,6 +21,7 @@
     CollaborativeMetadata,
     CollaborativeOrganizationRelationship,
     Dataset,
+    Geography,
     Metadata,
     Organization,
     Sector,
@@ -66,6 +67,7 @@ class UpdateCollaborativeMetadataInput:
     tags: Optional[List[str]]
     sectors: List[uuid.UUID]
     sdgs: Optional[List[uuid.UUID]]
+    geographies: Optional[List[uuid.UUID]]
 
 
 @strawberry_django.partial(Collaborative, fields="__all__", exclude=["datasets"])
@@ -393,6 +395,10 @@ def add_update_collaborative_metadata(
         _update_collaborative_sectors(collaborative, update_metadata_input.sectors)
         if update_metadata_input.sdgs is not None:
             _update_collaborative_sdgs(collaborative, update_metadata_input.sdgs)
+        if update_metadata_input.geographies is not None:
+            _update_collaborative_geographies(
+                collaborative, update_metadata_input.geographies
+            )
         return TypeCollaborative.from_django(collaborative)
 
     @strawberry_django.mutation(handle_django_errors=False)
@@ -959,3 +965,12 @@ def update_collaborative_organization_relationships(
             )
 
         return TypeCollaborative.from_django(collaborative)
+
+
+def _update_collaborative_geographies(
+    collaborative: Collaborative, geography_ids: List[uuid.UUID]
+) -> None:
+    """Update geographies for a collaborative."""
+    collaborative.geographies.clear()
+    geographies = Geography.objects.filter(id__in=geography_ids)
+    collaborative.geographies.add(*geographies)
diff --git a/api/utils/enums.py b/api/utils/enums.py
@@ -24,9 +24,10 @@ class OrganizationTypes(models.TextChoices):
 
 
 class GeoTypes(models.TextChoices):
-    DISTRICT = "DISTRICT"
-    STATE = "STATE"
+    REGION = "REGION"
     COUNTRY = "COUNTRY"
+    STATE = "STATE"
+    DISTRICT = "DISTRICT"
     UT = "UT"
 
 
