Add status in regionType

apps/geo/enums.py

@@ -1,4 +1,10 @@
 import graphene
+from geo.models import Region
+
+from utils.graphene.enums import (
+    convert_enum_to_graphene_enum,
+    get_enum_name_from_django_field,
+)
 
 
 class GeoAreaOrderingEnum(graphene.Enum):
@@ -8,3 +14,16 @@ class GeoAreaOrderingEnum(graphene.Enum):
     # DESC
     DESC_ID = f'-{ASC_ID}'
     DESC_ADMIN_LEVEL = f'-{ASC_ADMIN_LEVEL}'
+
+
+RegionStatusEnum = convert_enum_to_graphene_enum(
+    Region.Status, name='RegionStatusEnum'
+)
+
+
+enum_map = {
+    get_enum_name_from_django_field(field): enum
+    for field, enum in (
+        (Region.status, RegionStatusEnum),
+    )
+}

apps/geo/mutations.py

@@ -155,6 +155,8 @@ def mutate(root, info, region_id):
             error_data.append('Geo Area already generated')
         elif instance.created_by != info.context.user:
             error_data.append('Authorized User only update region')
+        elif instance.is_published == True:
+            error_data.append("Published Region cannot be retiggered")
 
         if error_data:
             return ReTriggeredFailedRegion(errors=[

apps/geo/schema.py

@@ -1,3 +1,4 @@
+from geo.enums import RegionStatusEnum
 import graphene
 from graphene_django import DjangoObjectType
 from graphene_django_extras import DjangoObjectField, PageGraphqlPagination
@@ -53,12 +54,14 @@ def get_custom_queryset(queryset, info, **kwargs):
 
 
 class RegionType(DjangoObjectType, ClientIdMixin):
+    status = graphene.Field(RegionStatusEnum)
+
     class Meta:
         model = Region
         only_fields = (
             'id', 'title', 'public', 'regional_groups',
             'key_figures', 'population_data', 'media_sources',
-            'centroid', 'is_published',
+            'centroid', 'is_published', 'status'
         )
 
     @staticmethod

apps/geo/tasks.py

@@ -211,7 +211,11 @@ def load_geo_areas(region_id):
     if not have_lock:
         return False
     region = Region.objects.filter(id=region_id).first()
-    region.status = Region.Status.INITIATED
+    if not region:
+      logger.error("Region not found", exc_info=True)
+      return False
+
+
     try:
         return_value = _load_geo_areas(region_id)
         region.status = Region.Status.COMPLETED

schema.graphql

@@ -6011,6 +6011,7 @@ type RegionDetailType {
   mediaSources: GenericScalar
   centroid: PointScalar
   clientId: ID!
+  status: RegionStatusEnum
   adminLevels: [AdminLevelType!]!
 }
 
@@ -6050,6 +6051,13 @@ input RegionProjectFilterData {
   isTest: Boolean
 }
 
+enum RegionStatusEnum {
+  INITIATED
+  PENDING
+  COMPLETED
+  FAILED
+}
+
 type RegionType {
   id: ID!
   title: String!
@@ -6060,6 +6068,7 @@ type RegionType {
   populationData: GenericScalar
   mediaSources: GenericScalar
   centroid: PointScalar
+  status: RegionStatusEnum
   clientId: ID!
 }