PR Fixes

apps/geo/models.py

@@ -14,11 +14,11 @@
 
 
 class Region(UserResource):
-    class Status(models.TextChoices):
-        INITIATED = 'initiated', 'Initiated'
-        PENDING = 'pending', 'Pending'
-        COMPLETED = 'completed', 'Completed'
-        FAILED = 'failed', 'Failed'
+    class Status(models.IntegerChoices):
+        INITIATED = 0, 'Initiated'
+        PENDING = 1, 'Pending'
+        COMPLETED = 2, 'Completed'
+        FAILED = 3, 'Failed'
     """
     Region model
 
@@ -40,7 +40,7 @@ class Status(models.TextChoices):
     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)
+    status = models.PositiveSmallIntegerField(choices=Status.choices, default=Status.PENDING)
 
     def __init__(self, *args, **kwargs):
         self.id: Union[int, None]

apps/geo/tasks.py

@@ -181,7 +181,10 @@ 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)
+    region = Region.objects.filter(pk=region_id).first()
+    if not region:
+        logger.error("Region not found", exc_info=True)
+        return False
     try:
         with reversion.create_revision():
             if AdminLevel.objects.filter(region=region).count() == 0: