Add status field in alert

Author: susilnem

deploy/helm/ifrcgo-helm/values.yaml

@@ -111,6 +111,8 @@ cronjobs:
     schedule: '0 9 * * *'
   - command: 'ingest_country_plan_file'
     schedule: '1 0 * * *'
+  - command: 'update_alert_status'
+    schedule: '0 */12 * * *'
 
 elasticsearch:
   enabled: true

main/enums.py

@@ -5,6 +5,7 @@
 from flash_update import enums as flash_update_enums
 from deployments import enums as deployments_enums
 from per import enums as per_enums
+from notifications import enums as notifications_enums
 
 
 apps_enum_register = [
@@ -13,6 +14,7 @@
     ('flash_update', flash_update_enums.enum_register),
     ('deployments', deployments_enums.enum_register),
     ('per', per_enums.enum_register),
+    ('notifications', notifications_enums.enum_register),
 ]
 
 

notifications/drf_views.py

@@ -62,7 +62,7 @@ class SurgeAlertViewset(viewsets.ReadOnlyModelViewSet):
         prefetch_related('molnix_tags', 'molnix_tags__groups').\
         select_related('event', 'country').all()
     filterset_class = SurgeAlertFilter
-    ordering_fields = ('created_at', 'atype', 'category', 'event', 'is_stood_down',)
+    ordering_fields = ('created_at', 'atype', 'category', 'event', 'is_stood_down', 'status', 'opens')
     search_fields = ('operation', 'message', 'event__name',)  # for /docs
 
     def get_serializer_class(self):

notifications/enums.py

@@ -0,0 +1,5 @@
+from . import models
+
+enum_register = {
+    'surge_alert_status': models.SurgeAlertStatus,
+}

notifications/management/commands/update_alert_status.py

@@ -0,0 +1,34 @@
+import logging
+from django.db import models
+
+from django.core.management.base import BaseCommand
+from django.utils import timezone
+from notifications.models import SurgeAlert, SurgeAlertStatus
+
+logger = logging.getLogger(__name__)
+
+class Command(BaseCommand):
+    '''
+        Updating the Alert Status according:
+        If the alert status is marked as stood_down, then the status is Stood Down.
+        If the closing timestamp (closes) is earlier than the current date, the status is displayed as Closed. Otherwise, it is displayed as Open.
+    '''
+    
+    help = 'Update alert status'
+
+    def handle(self, *args, **options):
+        now = timezone.now()
+        try:
+            SurgeAlert.objects.exclude(status=SurgeAlertStatus.CLOSED).filter(
+                models.Q(opens__lte=now, closes__gte=now) | models.Q(is_stood_down=True)
+            ).update(
+                status=models.Case(
+                    models.When(opens__lte=now, closes__gte=now, then=models.Value(SurgeAlertStatus.OPEN)),
+                    models.When(closes__lt=now, then=models.Value(SurgeAlertStatus.CLOSED)),
+                    models.When(is_stood_down=True, then=models.Value(SurgeAlertStatus.STOOD_DOWN)),
+                    default=models.F('status'),
+                    output_field=models.IntegerField()
+                )
+            )
+        except Exception as e:
+            logger.error('Error while updating alerts status: %s' % str(e))

notifications/migrations/0014_surgealert_status.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.23 on 2024-02-09 10:18
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('notifications', '0013_auto_20230410_0720'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='surgealert',
+            name='status',
+            field=models.IntegerField(choices=[(0, 'open'), (1, 'stood down'), (2, 'closed')], default=0, verbose_name='alert status'),
+        ),
+    ]

notifications/models.py

@@ -23,6 +23,10 @@ class SurgeAlertCategory(models.IntegerChoices):
     SHELTER = 3, _('shelter')
     STAND_DOWN = 4, _('stand down')
 
+class SurgeAlertStatus(models.IntegerChoices):
+    OPEN = 0, _('open')
+    STOOD_DOWN = 1, _('stood down')
+    CLOSED = 2, _('closed')
 
 class SurgeAlert(models.Model):
 
@@ -54,6 +58,7 @@ class SurgeAlert(models.Model):
 
     # Don't set `auto_now_add` so we can modify it on save
     created_at = models.DateTimeField(verbose_name=_('created at'))
+    status = models.IntegerField(choices=SurgeAlertStatus.choices, verbose_name=_('alert status'), default=0)
 
     class Meta:
         ordering = ['-created_at']
@@ -65,6 +70,13 @@ def save(self, *args, **kwargs):
         if (not self.id and not self.created_at) or (self.created_at > timezone.now()):
             self.created_at = timezone.now()
         self.is_stood_down = self.molnix_status == 'unfilled'
+        
+        if self.is_stood_down:
+            self.status = SurgeAlertStatus.STOOD_DOWN
+        elif self.closes and self.closes < timezone.now():
+            self.status = SurgeAlertStatus.CLOSED
+        else:
+            self.status = SurgeAlertStatus.OPEN
         return super(SurgeAlert, self).save(*args, **kwargs)
 
     def __str__(self):

notifications/serializers.py

@@ -11,6 +11,7 @@ class SurgeAlertSerializer(ModelSerializer):
     event = SurgeEventSerializer()
     country = MiniCountrySerializer()
     atype_display = serializers.CharField(source='get_atype_display', read_only=True)
+    status_display = serializers.CharField(source='get_status_display', read_only=True)
     category_display = serializers.CharField(source='get_category_display', read_only=True)
     molnix_tags = MolnixTagSerializer(many=True, read_only=True)
 
@@ -19,7 +20,7 @@ class Meta:
         fields = (
             'operation', 'country', 'message', 'deployment_needed', 'is_private', 'event', 'created_at', 'id',
             'atype', 'atype_display', 'category', 'category_display', 'molnix_id', 'molnix_tags',
-            'molnix_status', 'opens', 'closes', 'start', 'end', 'is_active', 'is_stood_down'
+            'molnix_status', 'opens', 'closes', 'start', 'end', 'is_active', 'is_stood_down','status_display'
         )