@@ -67,10 +67,12 @@ def __init__(
6767 # Mapping from stream ID to response stream events.
6868 self ._events : dict [
6969 int ,
70- h2 .events .ResponseReceived
71- | h2 .events .DataReceived
72- | h2 .events .StreamEnded
73- | h2 .events .StreamReset ,
70+ list [
71+ h2 .events .ResponseReceived
72+ | h2 .events .DataReceived
73+ | h2 .events .StreamEnded
74+ | h2 .events .StreamReset ,
75+ ],
7476 ] = {}
7577
7678 # Connection terminated events are stored as state since
@@ -102,9 +104,11 @@ async def handle_async_request(self, request: Request) -> Response:
102104 async with self ._init_lock :
103105 if not self ._sent_connection_init :
104106 try :
105- kwargs = {"request" : request }
106- async with Trace ("send_connection_init" , logger , request , kwargs ):
107- await self ._send_connection_init (** kwargs )
107+ sci_kwargs = {"request" : request }
108+ async with Trace (
109+ "send_connection_init" , logger , request , sci_kwargs
110+ ):
111+ await self ._send_connection_init (** sci_kwargs )
108112 except BaseException as exc :
109113 with AsyncShieldCancellation ():
110114 await self .aclose ()
@@ -293,6 +297,7 @@ async def _receive_response(
293297
294298 status_code = 200
295299 headers = []
300+ assert event .headers is not None
296301 for k , v in event .headers :
297302 if k == b":status" :
298303 status_code = int (v .decode ("ascii" , errors = "ignore" ))
@@ -310,6 +315,8 @@ async def _receive_response_body(
310315 while True :
311316 event = await self ._receive_stream_event (request , stream_id )
312317 if isinstance (event , h2 .events .DataReceived ):
318+ assert event .flow_controlled_length is not None
319+ assert event .data is not None
313320 amount = event .flow_controlled_length
314321 self ._h2_state .acknowledge_received_data (amount , stream_id )
315322 await self ._write_outgoing_data (request )
@@ -380,7 +387,9 @@ async def _receive_events(
380387
381388 await self ._write_outgoing_data (request )
382389
383- async def _receive_remote_settings_change (self , event : h2 .events .Event ) -> None :
390+ async def _receive_remote_settings_change (
391+ self , event : h2 .events .RemoteSettingsChanged
392+ ) -> None :
384393 max_concurrent_streams = event .changed_settings .get (
385394 h2 .settings .SettingCodes .MAX_CONCURRENT_STREAMS
386395 )
0 commit comments