Skip to content

Commit 46c2a3f

Browse files
committed
add labels enrich
Signed-off-by: Roi Glinik <groi.tech@gmail.com>
1 parent 03e87b4 commit 46c2a3f

1 file changed

Lines changed: 24 additions & 1 deletion

File tree

playbooks/robusta_playbooks/common_actions.py

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,17 @@
44

55
from pydantic import BaseModel
66

7-
from robusta.api import ActionParams, ExecutionBaseEvent, Finding, FindingSeverity, action
7+
from robusta.api import (
8+
ActionParams,
9+
EnrichmentType,
10+
ExecutionBaseEvent,
11+
Finding,
12+
FindingSeverity,
13+
SlackAnnotations,
14+
TableBlock,
15+
action,
16+
)
17+
from robusta.core.reporting.blocks import TableBlockFormat
818
from robusta.utils.parsing import format_event_templated_string
919

1020

@@ -99,6 +109,19 @@ def customise_finding(event: ExecutionBaseEvent, params: FindingOverrides):
99109
if labels_to_inject:
100110
logging.info(f"[customise_finding] injecting labels into finding: {labels_to_inject}")
101111
event.inject_finding_labels(labels_to_inject)
112+
event.add_enrichment(
113+
[
114+
TableBlock(
115+
[[k, v] for (k, v) in labels_to_inject.items()],
116+
["label", "value"],
117+
table_format=TableBlockFormat.vertical,
118+
table_name="*labels*",
119+
),
120+
],
121+
annotations={SlackAnnotations.ATTACHMENT: True},
122+
enrichment_type=EnrichmentType.alert_labels,
123+
title="labels",
124+
)
102125

103126

104127
class FindingFields(ActionParams):

0 commit comments

Comments
 (0)