wip

engine/apps/alerts/incident_appearance/renderers/base_renderer.py

@@ -4,12 +4,6 @@
 
 
 class AlertBaseRenderer(ABC):
-
-    # TODO: update this wording
-    # potential restricted title/message wording
-    IS_RESTRICTED_TITLE = "RESTRICTED TITLE TODO TODO"  # upgrade to see this
-    IS_RESTRICTED_MESSAGE = "RESTRICTED MESSAGE TODO TODO"
-
     def __init__(self, alert):
         self.alert = alert
 

engine/apps/alerts/incident_appearance/renderers/classic_markdown_renderer.py

@@ -1,5 +1,6 @@
 from apps.alerts.incident_appearance.renderers.base_renderer import AlertBaseRenderer, AlertGroupBaseRenderer
 from apps.alerts.incident_appearance.templaters import AlertClassicMarkdownTemplater
+from common.constants.alert_group_restrictions import IS_RESTRICTED_MESSAGE, IS_RESTRICTED_TITLE
 from common.utils import str_or_backup
 
 
@@ -13,8 +14,8 @@ def render(self):
         is_restricted = self.alert.group.is_restricted
 
         return {
-            "title": self.IS_RESTRICTED_TITLE if is_restricted else str_or_backup(templated_alert.title, "Alert"),
-            "message": self.IS_RESTRICTED_MESSAGE if is_restricted else str_or_backup(templated_alert.message, ""),
+            "title": IS_RESTRICTED_TITLE if is_restricted else str_or_backup(templated_alert.title, "Alert"),
+            "message": IS_RESTRICTED_MESSAGE if is_restricted else str_or_backup(templated_alert.message, ""),
             "image_url": None if is_restricted else str_or_backup(templated_alert.image_url, None),
             "source_link": None if is_restricted else str_or_backup(templated_alert.source_link, None),
         }

engine/apps/alerts/incident_appearance/renderers/web_renderer.py

@@ -1,5 +1,6 @@
 from apps.alerts.incident_appearance.renderers.base_renderer import AlertBaseRenderer, AlertGroupBaseRenderer
 from apps.alerts.incident_appearance.templaters import AlertWebTemplater
+from common.constants.alert_group_restrictions import IS_RESTRICTED_MESSAGE, IS_RESTRICTED_TITLE
 from common.utils import str_or_backup
 
 
@@ -13,8 +14,8 @@ def render(self):
         is_restricted = self.alert.group.is_restricted
 
         return {
-            "title": self.IS_RESTRICTED_TITLE if is_restricted else str_or_backup(templated_alert.title, "Alert"),
-            "message": self.IS_RESTRICTED_MESSAGE if is_restricted else str_or_backup(templated_alert.message, ""),
+            "title": IS_RESTRICTED_TITLE if is_restricted else str_or_backup(templated_alert.title, "Alert"),
+            "message": IS_RESTRICTED_MESSAGE if is_restricted else str_or_backup(templated_alert.message, ""),
             "image_url": None if is_restricted else str_or_backup(templated_alert.image_url, None),
             "source_link": None if is_restricted else str_or_backup(templated_alert.source_link, None),
         }

engine/apps/api/serializers/alert.py

@@ -61,10 +61,15 @@ def get_render_for_web(self, obj):
 
 class AlertRawSerializer(serializers.ModelSerializer):
     id = serializers.CharField(read_only=True, source="public_primary_key")
+    raw_request_data = serializers.SerializerMethodField()
 
     class Meta:
         model = Alert
         fields = [
             "id",
             "raw_request_data",
         ]
+
+    def get_raw_request_data(self, obj):
+        # TODO:
+        return {} if obj.group.is_restricted else obj.raw_request_data

engine/apps/public_api/serializers/alerts.py

@@ -7,7 +7,7 @@
 class AlertSerializer(EagerLoadingMixin, serializers.ModelSerializer):
     id = serializers.CharField(read_only=True, source="public_primary_key")
     alert_group_id = serializers.CharField(read_only=True, source="group.public_primary_key")
-    payload = serializers.JSONField(read_only=True, source="raw_request_data")
+    payload = serializers.SerializerMethodField(read_only=True)
 
     SELECT_RELATED = ["group"]
 
@@ -19,3 +19,6 @@ class Meta:
             "created_at",
             "payload",
         ]
+
+    def get_payload(self, obj):
+        return {} if obj.group.is_restricted else obj.raw_request_data

engine/apps/public_api/serializers/incidents.py

@@ -4,6 +4,7 @@
 from apps.alerts.models import AlertGroup
 from apps.telegram.models.message import TelegramMessage
 from common.api_helpers.mixins import EagerLoadingMixin
+from common.constants.alert_group_restrictions import IS_RESTRICTED_TITLE
 
 
 class IncidentSerializer(EagerLoadingMixin, serializers.ModelSerializer):
@@ -13,7 +14,7 @@ class IncidentSerializer(EagerLoadingMixin, serializers.ModelSerializer):
     route_id = serializers.SerializerMethodField()
     created_at = serializers.DateTimeField(source="started_at")
     alerts_count = serializers.SerializerMethodField()
-    title = serializers.CharField(source="web_title_cache")
+    title = serializers.SerializerMethodField()
     state = serializers.SerializerMethodField()
 
     SELECT_RELATED = ["channel", "channel_filter", "slack_message", "channel__organization"]
@@ -41,6 +42,9 @@ class Meta:
             "permalinks",
         ]
 
+    def get_title(self, obj):
+        return IS_RESTRICTED_TITLE if obj.is_restricted else obj.web_title_cache
+
     def get_alerts_count(self, obj):
         return len(obj.alerts.all())
 

engine/common/constants/alert_group_restrictions.py

@@ -0,0 +1,2 @@
+IS_RESTRICTED_TITLE = "UPGRADE TO SEE MORE"
+IS_RESTRICTED_MESSAGE = "UPGRADE TO SEE MORE"