@@ -61,42 +61,41 @@ class APIGatewayEventRequestContext(BaseRequestContext):
61
61
@property
62
62
def connected_at (self ) -> int | None :
63
63
"""The Epoch-formatted connection time. (WebSocket API)"""
64
- return self [ "requestContext" ] .get ("connectedAt" )
64
+ return self .get ("connectedAt" )
65
65
66
66
@property
67
67
def connection_id (self ) -> str | None :
68
68
"""A unique ID for the connection that can be used to make a callback to the client. (WebSocket API)"""
69
- return self [ "requestContext" ] .get ("connectionId" )
69
+ return self .get ("connectionId" )
70
70
71
71
@property
72
72
def event_type (self ) -> str | None :
73
73
"""The event type: `CONNECT`, `MESSAGE`, or `DISCONNECT`. (WebSocket API)"""
74
- return self [ "requestContext" ] .get ("eventType" )
74
+ return self .get ("eventType" )
75
75
76
76
@property
77
77
def message_direction (self ) -> str | None :
78
78
"""Message direction (WebSocket API)"""
79
- return self [ "requestContext" ] .get ("messageDirection" )
79
+ return self .get ("messageDirection" )
80
80
81
81
@property
82
82
def message_id (self ) -> str | None :
83
83
"""A unique server-side ID for a message. Available only when the `eventType` is `MESSAGE`."""
84
- return self [ "requestContext" ] .get ("messageId" )
84
+ return self .get ("messageId" )
85
85
86
86
@property
87
87
def operation_name (self ) -> str | None :
88
88
"""The name of the operation being performed"""
89
- return self [ "requestContext" ] .get ("operationName" )
89
+ return self .get ("operationName" )
90
90
91
91
@property
92
92
def route_key (self ) -> str | None :
93
93
"""The selected route key."""
94
- return self [ "requestContext" ] .get ("routeKey" )
94
+ return self .get ("routeKey" )
95
95
96
96
@property
97
97
def authorizer (self ) -> APIGatewayEventAuthorizer :
98
- authz_data = self ._data .get ("requestContext" , {}).get ("authorizer" , {})
99
- return APIGatewayEventAuthorizer (authz_data )
98
+ return APIGatewayEventAuthorizer (self .get ("authorizer" ) or {})
100
99
101
100
102
101
class APIGatewayProxyEvent (BaseProxyEvent ):
@@ -129,7 +128,7 @@ def resolved_headers_field(self) -> dict[str, Any]:
129
128
130
129
@property
131
130
def request_context (self ) -> APIGatewayEventRequestContext :
132
- return APIGatewayEventRequestContext (self . _data )
131
+ return APIGatewayEventRequestContext (self [ "requestContext" ] )
133
132
134
133
@property
135
134
def path_parameters (self ) -> dict [str , str ]:
@@ -241,8 +240,7 @@ def iam(self) -> RequestContextV2AuthorizerIam:
241
240
class RequestContextV2 (BaseRequestContextV2 ):
242
241
@property
243
242
def authorizer (self ) -> RequestContextV2Authorizer :
244
- ctx = self .get ("requestContext" ) or {} # key might exist but can be `null`
245
- return RequestContextV2Authorizer (ctx .get ("authorizer" , {}))
243
+ return RequestContextV2Authorizer (self .get ("authorizer" ) or {})
246
244
247
245
248
246
class APIGatewayProxyEventV2 (BaseProxyEvent ):
@@ -284,7 +282,7 @@ def cookies(self) -> list[str]:
284
282
285
283
@property
286
284
def request_context (self ) -> RequestContextV2 :
287
- return RequestContextV2 (self . _data )
285
+ return RequestContextV2 (self [ "requestContext" ] )
288
286
289
287
@property
290
288
def path_parameters (self ) -> dict [str , str ]:
0 commit comments