Refactor alert message handling to support multiple links. Introduced LinksLineBlock for rendering multiple links with optional icons, and updated AlertMessageBuilder to utilize this new structure.

Author: MikaKerman

elementary/messages/block_builders.py

@@ -82,8 +82,30 @@ def ItalicTextLineBlock(*, text: SimpleLineBlock) -> LineBlock:
     return TextLineBlock(text=text, style=TextStyle.ITALIC)
 
 
+def LinkInlineBlocks(
+    *, text: str, url: str, icon: Optional[Icon] = None
+) -> list[InlineBlock]:
+    inlines: list[InlineBlock] = []
+    if icon:
+        inlines.append(IconBlock(icon=icon))
+    inlines.append(LinkBlock(text=text, url=url))
+    return inlines
+
+
 def LinkLineBlock(*, text: str, url: str) -> LineBlock:
-    return LineBlock(inlines=[LinkBlock(text=text, url=url)])
+    return LineBlock(inlines=LinkInlineBlocks(text=text, url=url))
+
+
+def LinksLineBlock(*, links: list[tuple[str, str, Optional[Icon]]]) -> LineBlock:
+    return LineBlock(
+        inlines=sum(
+            [
+                LinkInlineBlocks(text=text, url=url, icon=icon)
+                for text, url, icon in links
+            ],
+            [],
+        )
+    )
 
 
 def SummaryLineBlock(

elementary/monitor/alerts/alert_messages/builder.py

@@ -9,7 +9,7 @@
     FactsBlock,
     ItalicTextLineBlock,
     JsonCodeBlock,
-    LinkLineBlock,
+    LinksLineBlock,
     MentionLineBlock,
     NonPrimaryFactBlock,
     PrimaryFactBlock,
@@ -104,7 +104,7 @@ def _get_run_alert_subtitle_block(
         detected_at_str: Optional[str] = None,
         suppression_interval: Optional[int] = None,
         env: Optional[str] = None,
-        report_link: Optional[ReportLinkData] = None,
+        links: list[ReportLinkData] = [],
     ) -> LinesBlock:
         summary = []
         summary.append((type.capitalize() + ":", name))
@@ -117,9 +117,11 @@ def _get_run_alert_subtitle_block(
             summary.append(("Suppression interval:", str(suppression_interval)))
         subtitle_lines = [SummaryLineBlock(summary=summary)]
 
-        if report_link:
+        if links:
             subtitle_lines.append(
-                LinkLineBlock(text="View in Elementary", url=report_link.url)
+                LinksLineBlock(
+                    links=[(link.text, link.url, link.icon) for link in links]
+                )
             )
         return LinesBlock(lines=subtitle_lines)
 
@@ -138,6 +140,7 @@ def _get_run_alert_subtitle_blocks(
         elif isinstance(alert, ModelAlertModel):
             asset_type = "snapshot" if alert.materialization == "snapshot" else "model"
             asset_name = alert.alias
+        report_link = alert.get_report_link()
         return [
             self._get_run_alert_subtitle_block(
                 type=asset_type,
@@ -146,7 +149,7 @@ def _get_run_alert_subtitle_blocks(
                 detected_at_str=alert.detected_at_str,
                 suppression_interval=alert.suppression_interval,
                 env=alert.env,
-                report_link=alert.get_report_link(),
+                links=[report_link] if report_link else [],
             )
         ]
 

elementary/monitor/data_monitoring/alerts/integrations/utils/report_link.py

@@ -1,6 +1,7 @@
 from enum import Enum
 from typing import Optional
 
+from elementary.messages.blocks import Icon
 from elementary.utils.pydantic_shim import BaseModel
 
 TEST_RUNS_LINK_TEXT = "View test runs"
@@ -10,6 +11,7 @@
 class ReportLinkData(BaseModel):
     url: str
     text: str
+    icon: Optional[Icon] = None
 
 
 class ReportPath(Enum):
@@ -31,9 +33,11 @@ def _get_run_history_report_link(
         url = f"{formatted_report_url}/report/{path.value}/{unique_id}/"
         report_link = ReportLinkData(
             url=url,
-            text=TEST_RUNS_LINK_TEXT
-            if path == ReportPath.TEST_RUNS
-            else MODEL_RUNS_LINK_TEXT,
+            text=(
+                TEST_RUNS_LINK_TEXT
+                if path == ReportPath.TEST_RUNS
+                else MODEL_RUNS_LINK_TEXT
+            ),
         )
 
     return report_link
@@ -62,7 +66,7 @@ def get_model_test_runs_link(
 
     if model_unique_id and report_url:
         formatted_report_url = _get_formatted_report_url(report_url)
-        url = f'{formatted_report_url}/report/{ReportPath.TEST_RUNS.value}/?treeNode={{"id":"{model_unique_id}"}}'
+        url = f'{formatted_report_url}/report/{ReportPath.TEST_RUNS.value}/?treeNode={{"id":"{model_unique_id}"}}'  # noqa: E231
         report_link = ReportLinkData(url=url, text=TEST_RUNS_LINK_TEXT)
 
     return report_link