Add prescribed field to AlertGroup custom labels

Author: Konstantinov-Innokentii

engine/apps/alerts/models/alert_receive_channel.py

@@ -294,8 +294,8 @@ class AlertReceiveChannel(IntegrationOptionsMixin, MaintainableObject):
     rate_limited_in_slack_at = models.DateTimeField(null=True, default=None)
     rate_limit_message_task_id = models.CharField(max_length=100, null=True, default=None)
 
-    AlertGroupCustomLabels = list[tuple[str, str | None, str | None]] | None
-    alert_group_labels_custom: AlertGroupCustomLabels = models.JSONField(null=True, default=None)
+    AlertGroupCustomLabelsDB = list[tuple[str, str | None, str | None]] | None
+    alert_group_labels_custom: AlertGroupCustomLabelsDB = models.JSONField(null=True, default=None)
     """
     Stores "custom labels" for alert group labels. Custom labels can be either "plain" or "templated".
     For plain labels, the format is: [<LABEL_KEY_ID>, <LABEL_VALUE_ID>, None]

engine/apps/api/serializers/alert_receive_channel.py

@@ -15,6 +15,7 @@
 from apps.base.messaging import get_messaging_backends
 from apps.integrations.legacy_prefix import has_legacy_prefix
 from apps.labels.models import LabelKeyCache, LabelValueCache
+from apps.labels.types import LabelKey
 from apps.user_management.models import Organization
 from common.api_helpers.custom_fields import TeamPrimaryKeyRelatedField
 from common.api_helpers.exceptions import BadRequest
@@ -25,27 +26,29 @@
 from .labels import LabelsSerializerMixin
 
 
-class AlertGroupCustomLabelKey(typing.TypedDict):
-    id: str
-    name: str
-
-
-class AlertGroupCustomLabelValue(typing.TypedDict):
+# AlertGroupCustomLabelValue represents custom alert group label value for API requests
+# It handles two types of label's value:
+# 1. Just Label Value from a label repo for a static label
+# 2. Templated Label value which is actually a jinja template for a dynamic label.
+class AlertGroupCustomLabelValueAPI(typing.TypedDict):
     id: str | None  # None for templated labels, label value ID for plain labels
     name: str  # Jinja template for templated labels, label value name for plain labels
+    prescribed: bool  # Indicates of selected label value is prescribed. Not applicable for templated values.
 
 
-class AlertGroupCustomLabel(typing.TypedDict):
-    key: AlertGroupCustomLabelKey
-    value: AlertGroupCustomLabelValue
+# AlertGroupCustomLabel represents Alert group custom label for API requests
+# Key is just a LabelKey from label repo, while value could be value from repo or a jinja template.
+class AlertGroupCustomLabelAPI(typing.TypedDict):
+    key: LabelKey
+    value: AlertGroupCustomLabelValueAPI
 
 
-AlertGroupCustomLabels = list[AlertGroupCustomLabel]
+AlertGroupCustomLabelsAPI = list[AlertGroupCustomLabelAPI]
 
 
 class IntegrationAlertGroupLabels(typing.TypedDict):
     inheritable: dict[str, bool]
-    custom: AlertGroupCustomLabels
+    custom: AlertGroupCustomLabelsAPI
     template: str | None
 
 
@@ -55,11 +58,13 @@ class CustomLabelSerializer(serializers.Serializer):
     class CustomLabelKeySerializer(serializers.Serializer):
         id = serializers.CharField()
         name = serializers.CharField()
+        prescribed = serializers.BooleanField(required=False, default=False)
 
     class CustomLabelValueSerializer(serializers.Serializer):
         # ID is null for templated labels. For such labels, the "name" value is a Jinja2 template.
         id = serializers.CharField(allow_null=True)
         name = serializers.CharField()
+        prescribed = serializers.BooleanField(required=False, default=False)
 
     key = CustomLabelKeySerializer()
     value = CustomLabelValueSerializer()
@@ -112,16 +117,26 @@ def update(
         return instance
 
     @staticmethod
-    def _create_custom_labels(organization: Organization, labels: AlertGroupCustomLabels) -> None:
+    def _create_custom_labels(organization: Organization, labels: AlertGroupCustomLabelsAPI) -> None:
         """Create LabelKeyCache and LabelValueCache objects for custom labels."""
 
         label_keys = [
-            LabelKeyCache(id=label["key"]["id"], name=label["key"]["name"], organization=organization)
+            LabelKeyCache(
+                id=label["key"]["id"],
+                name=label["key"]["name"],
+                prescribed=label["key"]["prescribed"],
+                organization=organization,
+            )
             for label in labels
         ]
 
         label_values = [
-            LabelValueCache(id=label["value"]["id"], name=label["value"]["name"], key_id=label["key"]["id"])
+            LabelValueCache(
+                id=label["value"]["id"],
+                name=label["value"]["name"],
+                prescribed=label["value"]["prescribed"],
+                key_id=label["key"]["id"],
+            )
             for label in labels
             if label["value"]["id"]  # don't create LabelValueCache objects for templated labels
         ]
@@ -147,8 +162,8 @@ def to_representation(cls, instance: AlertReceiveChannel) -> IntegrationAlertGro
 
     @staticmethod
     def _custom_labels_to_internal_value(
-        custom_labels: AlertGroupCustomLabels,
-    ) -> AlertReceiveChannel.AlertGroupCustomLabels:
+        custom_labels: AlertGroupCustomLabelsAPI,
+    ) -> AlertReceiveChannel.AlertGroupCustomLabelsDB:
         """Convert custom labels from API representation to the schema used by the JSONField on the model."""
 
         return [
@@ -158,8 +173,8 @@ def _custom_labels_to_internal_value(
 
     @staticmethod
     def _custom_labels_to_representation(
-        custom_labels: AlertReceiveChannel.AlertGroupCustomLabels,
-    ) -> AlertGroupCustomLabels:
+        custom_labels: AlertReceiveChannel.AlertGroupCustomLabelsDB,
+    ) -> AlertGroupCustomLabelsAPI:
         """
         Inverse of the _custom_labels_to_internal_value method above.
         Fetches label names from DB cache, so the API response schema is consistent with other label endpoints.
@@ -170,33 +185,37 @@ def _custom_labels_to_representation(
         if custom_labels is None:
             return []
 
-        # get up-to-date label key names
-        label_key_names = {
-            k.id: k.name
-            for k in LabelKeyCache.objects.filter(id__in=[label[0] for label in custom_labels]).only("id", "name")
+        # build index of keys id to name and prescribed flag
+        label_key_index = {
+            k.id: {"name": k.name, "prescribed": k.prescribed}
+            for k in LabelKeyCache.objects.filter(id__in=[label[0] for label in custom_labels]).only(
+                "id", "name", "prescribed"
+            )
         }
 
-        # get up-to-date label value names
-        label_value_names = {
-            v.id: v.name
+        # build index of values id to name and prescribed flag
+        label_value_index = {
+            v.id: {"name": v.name, "prescribed": v.prescribed}
             for v in LabelValueCache.objects.filter(id__in=[label[1] for label in custom_labels if label[1]]).only(
-                "id", "name"
+                "id", "name", "prescribed"
             )
         }
 
         return [
             {
                 "key": {
                     "id": key_id,
-                    "name": label_key_names[key_id],
+                    "name": label_key_index[key_id]["name"],
+                    "prescribed": label_key_index[key_id]["prescribed"],
                 },
                 "value": {
                     "id": value_id if value_id else None,
-                    "name": label_value_names[value_id] if value_id else typing.cast(str, template),
+                    "name": label_value_index[value_id]["name"] if value_id else typing.cast(str, template),
+                    "prescribed": label_value_index[key_id]["prescribed"],
                 },
             }
             for key_id, value_id, template in custom_labels
-            if key_id in label_key_names and (value_id in label_value_names or not value_id)
+            if key_id in label_value_index and (value_id in label_value_index or not value_id)
         ]
 
 

engine/apps/api/tests/test_alert_receive_channel.py

@@ -1629,7 +1629,12 @@ def test_alert_group_labels_post(alert_receive_channel_internal_api_setup, make_
     labels = [{"key": {"id": "test", "name": "test"}, "value": {"id": "123", "name": "123"}}]
     alert_group_labels = {
         "inheritable": {"test": False},
-        "custom": [{"key": {"id": "test", "name": "test"}, "value": {"id": "123", "name": "123"}}],
+        "custom": [
+            {
+                "key": {"id": "test", "name": "test", "prescribed": False},
+                "value": {"id": "123", "name": "123", "prescribed": False},
+            }
+        ],
         "template": "{{ payload.labels | tojson }}",
     }
     data = {