Add status in geo area

apps/geo/migrations/0044_region_status.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.25 on 2024-10-21 08:59
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('geo', '0043_create-unaccent_extension'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='region',
+            name='status',
+            field=models.CharField(choices=[('initiated', 'Initiated'), ('pending', 'Pending'), ('completed', 'Completed'), ('failed', 'Failed')], default='pending', max_length=30),
+        ),
+    ]

apps/geo/models.py

@@ -14,6 +14,11 @@
 
 
 class Region(UserResource):
+    class Status(models.TextChoices):
+        INITIATED = 'initiated', 'Initiated'
+        PENDING = 'pending', 'Pending'
+        COMPLETED = 'completed', 'Completed'
+        FAILED = 'failed', 'Failed'
     """
     Region model
 
@@ -35,6 +40,7 @@ class Region(UserResource):
     cache_index = models.SmallIntegerField(default=0)  # Used to track cache update.
     centroid = models.PointField(blank=True, null=True)  # Admin level 0 centroid
     geo_options = models.JSONField(default=None, blank=True, null=True)
+    status = models.CharField(max_length=30, choices=Status.choices, default=Status.PENDING)
 
     def __init__(self, *args, **kwargs):
         self.id: Union[int, None]

apps/geo/serializers.py

@@ -111,6 +111,7 @@ def create(self, validated_data):
         admin_level.save()
 
         region = admin_level.region
+        region.status = Region.Status.INITIATED
         region.modified_by = self.context['request'].user
         region.save()
 

apps/geo/tasks.py

@@ -181,26 +181,31 @@ def _load_geo_areas(region_id):
     Basically, it  starts with root admin level and iterate through all the
     children.
     """
+    region = Region.objects.get(pk=region_id)
+    try:
+        with reversion.create_revision():
+            if AdminLevel.objects.filter(region=region).count() == 0:
+                return True
 
-    with reversion.create_revision():
-        region = Region.objects.get(pk=region_id)
-
-        if AdminLevel.objects.filter(region=region).count() == 0:
-            return True
-
-        parent_admin_levels = AdminLevel.objects.filter(
-            region=region, parent=None
-        )
-        completed_levels = []
-        _extract_from_admin_levels(
-            parent_admin_levels,
-            None,
-            completed_levels,
-        )
+            parent_admin_levels = AdminLevel.objects.filter(
+                region=region, parent=None
+            )
+            completed_levels = []
+            _extract_from_admin_levels(
+                parent_admin_levels,
+                None,
+                completed_levels,
+            )
 
-        region.calc_cache()
+            region.calc_cache()
+            region.status = Region.Status.COMPLETED
+            region.save()
+        return True
 
-    return True
+    except Exception:
+        logger.error('Load Geo Areas', exc_info=True)
+        region.status = Region.Status.FAILED
+        region.save()
 
 
 @shared_task