Alert group column/label selector

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+### Added
+
+- Add options to customize table columns in AlertGroup page ([3281](https://github.com/grafana/oncall/pull/3281))
+
 ### Fixed
 
 - User profile UI tweaks ([#3443](https://github.com/grafana/oncall/pull/3443))

engine/apps/api/alert_group_table_columns.py

@@ -0,0 +1,31 @@
+import typing
+
+from apps.user_management.constants import AlertGroupTableColumns, default_columns
+
+if typing.TYPE_CHECKING:
+    from apps.user_management.models import User
+
+
+def alert_group_table_user_settings(user: "User") -> AlertGroupTableColumns:
+    """
+    Returns user settings for alert group table columns. The flag "default" shows that user has default settings for
+    visible columns. It's used by frontend to enable/disable `reset` button.
+    This function uses lazy update to update columns settings for organization and for user.
+    """
+    default_organization_columns = default_columns()
+    if not user.organization.alert_group_table_columns:
+        user.organization.update_alert_group_table_columns(default_organization_columns)
+    organization_columns = user.organization.alert_group_table_columns
+    if user.alert_group_table_selected_columns:
+        visible_columns = [
+            column for column in user.alert_group_table_selected_columns if column in organization_columns
+        ]
+    else:
+        visible_columns = default_organization_columns
+    user.update_alert_group_table_selected_columns(visible_columns)
+    hidden_columns = [column for column in organization_columns if column not in visible_columns]
+    return {
+        "visible": visible_columns,
+        "hidden": hidden_columns,
+        "default": visible_columns == default_organization_columns,
+    }

engine/apps/api/serializers/alert_group_table_settings.py

@@ -0,0 +1,60 @@
+from rest_framework import serializers
+from rest_framework.exceptions import ValidationError
+
+from apps.user_management.constants import (
+    AlertGroupTableColumnTypeChoices,
+    AlertGroupTableDefaultColumnChoices,
+    default_columns,
+)
+
+
+class AlertGroupTableColumnSerializer(serializers.Serializer):
+    name = serializers.CharField(max_length=200)
+    id = serializers.CharField(max_length=200)
+    type = serializers.ChoiceField(choices=AlertGroupTableColumnTypeChoices.choices)
+
+    def validate(self, data):
+        self._validate_id(data)
+        return data
+
+    def _validate_id(self, data):
+        """Validate if `id` of column with `default` type is in the list of available default columns"""
+        if (
+            data["type"] == AlertGroupTableColumnTypeChoices.DEFAULT.value
+            and data["id"] not in AlertGroupTableDefaultColumnChoices.values
+        ):
+            raise ValidationError("Invalid column id format")
+
+
+class AlertGroupTableColumnsOrganizationSerializer(serializers.Serializer):
+    visible = AlertGroupTableColumnSerializer(many=True)
+    hidden = AlertGroupTableColumnSerializer(many=True)
+
+    def validate(self, data):
+        """
+        Validate that at least one column is selected as visible and that all default columns are in the list.
+        """
+        columns = data["visible"] + data["hidden"]
+        request_columns_ids = [column["id"] for column in columns]
+        if len(data["visible"]) == 0:
+            raise ValidationError("At least one column should be selected as visible")
+        elif not set(request_columns_ids) >= set(AlertGroupTableDefaultColumnChoices.values):
+            raise ValidationError("Default column cannot be removed")
+        elif len(request_columns_ids) > len(set(request_columns_ids)):
+            raise ValidationError("Duplicate column")
+        return data
+
+
+class AlertGroupTableColumnsUserSerializer(AlertGroupTableColumnsOrganizationSerializer):
+    def validate(self, data):
+        """
+        Validate that all columns exist in organization alert group table columns list.
+        """
+        data = super().validate(data)
+        columns = data["visible"] + data["hidden"]
+        request_columns_ids = [column["id"] for column in columns]
+        organization_columns = self.context["request"].auth.organization.alert_group_table_columns or default_columns()
+        organization_columns_ids = [column["id"] for column in organization_columns]
+        if set(organization_columns_ids) != set(request_columns_ids):
+            raise ValidationError("Invalid settings")
+        return data