feat: adding a notification limit

thirdeye-dev · Mar 24, 2024 · 60bc85a · 60bc85a
1 parent db24d91
commit 60bc85a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/backend/api_app/monitoring/tasks.py b/backend/api_app/monitoring/tasks.py
@@ -356,6 +356,16 @@ def trace_transaction(transaction_hash):
                 continue
 
             for alert in alerts:
+                notifications = Notification.objects.filter(
+                    alert__smart_contract__owner_organization=monitoring_task.SmartContract.owner_organization,
+                ).count()
+
+                if notifications >= monitoring_task.SmartContract.owner_organization.notification_limit:
+                    logger.info(
+                        f"[DEBUG] Notification limit reached for organization {monitoring_task.SmartContract.owner_organization}"
+                    )
+                    return # we don't want to run more alerts
+
                 logger.info(f"[DEBUG] Running alert {alert.id}")
                 alert_runner = BlockchainAlertRunner(alert, transaction)
                 alert_runner.run()

diff --git a/backend/authentication/organizations/models.py b/backend/authentication/organizations/models.py
@@ -30,6 +30,7 @@ def invite_user_from_email(email, invited_by, organization=None, token=None):
 class Organization(BaseMixin):
     id = models.UUIDField(primary_key=True, editable=False)
     name = models.CharField(max_length=100)
+    notification_limit = models.PositiveIntegerField(default=5)
 
     class Meta:
         verbose_name = _("Organization")