Merge pull request #1760 from domwhewell-sage/dev

Fix Teams output module
blacklanternsecurity · Sep 9, 2024 · 8056c0a · 8056c0a
2 parents 2bf9d6c + 465d089
commit 8056c0a
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/bbot/modules/output/teams.py b/bbot/modules/output/teams.py
@@ -15,26 +15,10 @@ class Teams(WebhookOutputModule):
         "min_severity": "Only allow VULNERABILITY events of this severity or higher",
     }
     _module_threads = 5
-    adaptive_card = {
-        "type": "message",
-        "attachments": [
-            {
-                "contentType": "application/vnd.microsoft.card.adaptive",
-                "contentUrl": None,
-                "content": {
-                    "$schema": "http://adaptivecards.io/schemas/adaptive-card.json",
-                    "type": "AdaptiveCard",
-                    "version": "1.2",
-                    "msteams": {"width": "full"},
-                    "body": [],
-                },
-            }
-        ],
-    }
 
     async def handle_event(self, event):
         while 1:
-            data = self.format_message(self.adaptive_card.copy(), event)
+            data = self.format_message(event)
 
             response = await self.helpers.request(
                 url=self.webhook_url,
@@ -89,7 +73,23 @@ def get_severity_color(self, event):
                 color = "Good"
         return color
 
-    def format_message(self, adaptive_card, event):
+    def format_message(self, event):
+        adaptive_card = {
+            "type": "message",
+            "attachments": [
+                {
+                    "contentType": "application/vnd.microsoft.card.adaptive",
+                    "contentUrl": None,
+                    "content": {
+                        "$schema": "http://adaptivecards.io/schemas/adaptive-card.json",
+                        "type": "AdaptiveCard",
+                        "version": "1.2",
+                        "msteams": {"width": "full"},
+                        "body": [],
+                    },
+                }
+            ],
+        }
         heading = {"type": "TextBlock", "text": f"{event.type}", "wrap": True, "size": "Large", "style": "heading"}
         body = adaptive_card["attachments"][0]["content"]["body"]
         body.append(heading)