@@ -68,11 +68,11 @@ async def __aenter__(
6868 self ._exit_stack = stack .pop_all ()
6969 return retval
7070
71- async def __aexit__ (self , exc_type , exc_val , exc_tb ) -> typing . Union [ bool , None ] :
71+ async def __aexit__ (self , exc_type , exc_val , exc_tb ) -> bool | None :
7272 return await self ._exit_stack .__aexit__ (exc_type , exc_val , exc_tb )
7373
7474 async def read (
75- self , max_bytes : int , timeout : typing . Optional [ float ] = None
75+ self , max_bytes : int , timeout : float | None = None
7676 ) -> bytes :
7777 message : Message = await self .receive (timeout = timeout )
7878 type = message ["type" ]
@@ -82,10 +82,10 @@ async def read(
8282
8383 event : wsproto .events .Event
8484 if type == "websocket.send" :
85- data_str : typing . Optional [ str ] = message .get ("text" )
85+ data_str : str | None = message .get ("text" )
8686 if data_str is not None :
8787 event = wsproto .events .TextMessage (data_str )
88- data_bytes : typing . Optional [ bytes ] = message .get ("bytes" )
88+ data_bytes : bytes | None = message .get ("bytes" )
8989 if data_bytes is not None :
9090 event = wsproto .events .BytesMessage (data_bytes )
9191 elif type == "websocket.close" :
@@ -94,7 +94,7 @@ async def read(
9494 return self .connection .send (event )
9595
9696 async def write (
97- self , buffer : bytes , timeout : typing . Optional [ float ] = None
97+ self , buffer : bytes , timeout : float | None = None
9898 ) -> None :
9999 self .connection .receive_data (buffer )
100100 for event in self .connection .events ():
@@ -121,7 +121,7 @@ async def aclose(self) -> None:
121121 async def send (self , message : Message ) -> None :
122122 self ._receive_queue .put (message )
123123
124- async def receive (self , timeout : typing . Optional [ float ] = None ) -> Message :
124+ async def receive (self , timeout : float | None = None ) -> Message :
125125 while self ._send_queue .empty ():
126126 await anyio .sleep (0 )
127127 return self ._send_queue .get (timeout = timeout )
@@ -178,9 +178,9 @@ async def __aenter__(self) -> "ASGIWebSocketTransport":
178178
179179 async def __aexit__ (
180180 self ,
181- exc_type : typing . Optional [ type [BaseException ]] = None ,
182- exc_val : typing . Optional [ BaseException ] = None ,
183- exc_tb : typing . Optional [ TracebackType ] = None ,
181+ exc_type : type [BaseException ] | None = None ,
182+ exc_val : BaseException | None = None ,
183+ exc_tb : TracebackType | None = None ,
184184 ) -> None :
185185 await super ().__aexit__ (exc_type , exc_val , exc_tb )
186186 await self .exit_stack .__aexit__ (exc_type , exc_val , exc_tb )
0 commit comments