Annotate incident_fits_tristates

src/argus/notificationprofile/models.py

@@ -4,7 +4,7 @@
 from functools import reduce
 import logging
 from operator import or_
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Dict, Optional
 
 from django.conf import settings
 from django.db import models
@@ -20,6 +20,8 @@
 if TYPE_CHECKING:
     from argus.incident.models import Event, Incident
 
+TriState = Optional[bool]
+
 
 LOG = logging.getLogger(__name__)
 
@@ -140,7 +142,7 @@ def is_empty(self):
             and self.is_event_type_empty()
         )
 
-    def incident_fits_tristates(self, incident):
+    def incident_fits_tristates(self, incident) -> Dict[str, TriState]:
         if self.are_tristates_empty():
             return {}
         fits_tristates = {}