fix: avoid KeyError in extract_team_id for malformed payloads

Author: saschabuehrle

slack_bolt/request/internals.py

@@ -112,9 +112,9 @@ def extract_team_id(payload: Dict[str, Any]) -> Optional[str]:
     if isinstance(payload.get("event"), dict):
         return extract_team_id(payload["event"])
     if isinstance(payload.get("user"), dict):
-        return payload["user"]["team_id"]
+        return payload["user"].get("team_id")
     if isinstance(payload.get("view"), dict):
-        return payload["view"]["team_id"]
+        return payload["view"].get("team_id")
     return None
 
 

tests/slack_bolt/request/test_internals.py

@@ -1253,8 +1253,10 @@ def test_extraction_functions_invalid_dict_keys(self):
         invalid_payloads = {
             "event": {"event": "some_event_type"},
             "user": {"user": "U12345"},
+            "user_missing_team_id": {"user": {"id": "U12345"}},
             "team": {"team": "T12345"},
             "view": {"view": "V12345"},
+            "view_missing_team_id": {"view": {"id": "V12345"}},
             "message": {"message": "some text"},
             "item": {"item": "item_id"},
             "function_data": {"function_data": "fd_123"},