EjP ddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z e jdk\rddl mZnddlmZddlZddlZddlZddlmZmZddlmZdd lmZdd lmZmZmZmZdd l m!Z!d d l"m#Z#m$Z$m%Z%m&Z&m'Z'd dl(m)Z)m*Z*d dl+m,Z,e jZr(ddl.m/Z/m0Z0ddlm1Z1m2Z2m3Z3ddl m4Z4ddl5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;e jxdZ=e jdZ>e jdZ?edddZ@edddZAGddeBZCGddeBZDGd dZEGd!dejZG d*d"ZHGd#d$e je@ZJej d+eeeeddddeeedd% d,d&ZLGd'd(e jeAZMej d+eeeeddddeeedd% d-d)ZOy).) annotationsN) TracebackType) )TypeVar)MemoryObjectReceiveStreamMemoryObjectSendStream) CloseReason)USE_CLIENT_DEFAULT)'DEFAULT_KEEPALIVE_PING_INTERVAL_SECONDS&DEFAULT_KEEPALIVE_PING_TIMEOUT_SECONDSDEFAULT_MAX_MESSAGE_SIZE_BYTESDEFAULT_QUEUE_SIZE)Headers)HTTPXWSExceptionWebSocketDisconnectWebSocketInvalidTypeReceivedWebSocketNetworkErrorWebSocketUpgradeError)AsyncPingManager PingManager)ASGIWebSocketAsyncNetworkStream)AsyncNetworkStream NetworkStream) AsyncClientClientUseClientDefault)Response) AuthTypes CookieTypes HeaderTypesQueryParamTypesRequestExtensions TimeoutTypestextbinary TaskFunction TaskResult SyncSessionWebSocketSession)bounddefault AsyncSessionAsyncWebSocketSessionc eZdZy) ShouldCloseN__name__ __module__ __qualname__T/root/workspace/ytshorts/venv/lib/python3.12/site-packages/httpx2/websockets/_api.pyr3r3@r9r3c eZdZy) EndOfStreamNr4r8r9r:r=r=Dr;r9r=ceZdZUdZded<ded<eeeedd ddZ dd Z dd Z dd Z ddd Z d d Zd!dZd"dZd#d$dZd%d&dZd%d'dZd%d(dZd)d*dZd+d,dZd-dZd%d.dZ d/dZd0dZy)1r-a Sync context manager representing an opened WebSocket session. Attributes: subprotocol (typing.Optional[str]): Optional protocol that has been accepted by the server. response (Response | None): The webSocket handshake response. str | None subprotocolResponse | NoneresponseNmax_message_size_bytes queue_sizekeepalive_ping_interval_secondskeepalive_ping_timeout_secondsrBc||_tjjtjj |_||_|j +|j jjd|_ nd|_ tj||_ t|_tj |_tj$|_d|_d|_||_||_||_||_yNsec-websocket-protocol)streamwsproto connection ConnectionConnectionTypeCLIENTrBheadersgetr@queueQueue_eventsr _ping_manager threadingEvent _should_closeLock _write_lock_should_close_task _executor_max_message_size_bytes _queue_size _keepalive_ping_interval_seconds_keepalive_ping_timeout_secondsselfrKrDrErFrGrBs r:__init__zWebSocketSession.__init__Vs !,,778N8N8U8UV  == $#}}44889QRD #D MR[[YcMd (]&__.$>>+JNGK'=$%0O-/M,r9c|jWtjj|_|jj |j j|_|jJ|j|jfSN)r\ concurrentfuturesThreadPoolExecutorr]submitrYwaitrcs r:_get_executor_should_close_taskz0WebSocketSession._get_executor_should_close_taskusn  " " *'//BBDDN&*nn&;&;D##G, %r9c@ddl} |jj|}|j5|jj |dddy#1swYyxYw#|j $r0}|jtjdt|d}~wwxYw)a Send an Event message. Mainly useful to send events that are not supported by the library. Most of the time, [ping()][httpx_ws.WebSocketSession.ping], [send_text()][httpx_ws.WebSocketSession.send_text], [send_bytes()][httpx_ws.WebSocketSession.send_bytes] and [send_json()][httpx_ws.WebSocketSession.send_json] are preferred. Args: event: The event to send. Raises: WebSocketNetworkError: A network error occured. Examples: Send an event. event = wsproto.events.Message(b"Hello!") ws.send(event) rNStream write error httpcore2rMrr[rKwrite WriteErrorryr INTERNAL_ERRORrrcrrdataes r:rzWebSocketSession.sends,  1??''.D!! ( !!$' ( ( (## 1 JJ{113G H')q 0 1s4'AA AAAAB-+BBcftjj|}|j|y)a Send a text message. Args: data: The text to send. Raises: WebSocketNetworkError: A network error occured. Examples: Send a text message. ws.send_text("Hello!") rNrLr TextMessagerrcrrs r: send_textzWebSocketSession.send_texts'***5 %r9cftjj|}|j|y)a Send a bytes message. Args: data: The data to send. Raises: WebSocketNetworkError: A network error occured. Examples: Send a bytes message. ws.send_bytes(b"Hello!") rNrLr BytesMessagerrs r: send_byteszWebSocketSession.send_bytess'+++6 %r9c|dvsJtj|}|dk(r|j|y|j|j dy)a Send JSON data. Args: data: The data to send. Must be serializable by [json.dumps][json.dumps]. mode: The sending mode. Should either be `'text'` or `'bytes'`. Raises: WebSocketNetworkError: A network error occured. Examples: Send JSON data. data = {"message": "Hello!"} ws.send_json(data) r'r(utf-8Njsondumpsrrencodercrmodeserialized_datas r: send_jsonzWebSocketSession.send_jsonsK&))))**T* 6> NN? + OOO227; $$W-D X %%g.Dzz$r9cddl}|jj|j|jj d|j j tj jjtj jjhvrltjj||}|j j|} |j5|jj!|ddd|jj%y#1swY$xYw#|j"$rY8wxYw)a  Close the WebSocket session. Internally, it'll send the [CloseConnection][wsproto.events.CloseConnection] event. *This method is automatically called when exiting the context manager.* Args: code: The integer close code to indicate why the connection has closed. reason: Additional reasoning for why the connection has closed. Examples: Close the WebSocket session. ws.close() rNF)rrYsetr]shutdownrMstaterLConnectionState LOCAL_CLOSINGCLOSEDrrrr[rKrrryrcrrrrrs r:ryzWebSocketSession.closes(   >> % NN # #E * ??    . . < <    . . 5 5)  NN224@E??''.D %%,KK%%d+,  ,,''  s* D7,D+D7+D40D77E E c&ddl}d}d} |jjs|j|j|}|j j ||j jD]{}t|tjjrZ|j j|j}|j5|jj|dddt|tjj r&|j"j%|j&t|tjj(r|jj+t|tjj,r2|t/t|j0t2r|j0j5n |j0z }||kDrU|j7t8j:d|j<j?tAt8j:dn|jBs| |j0}no||j0z }n_|d}|j<j?|n?tE|}|||j0z}d}d}|j<j?|a|j<j?|~|jjsyy#1swY&xYw#|jF|jHtJf$rF|j7t8jLd|j<j?tOYytP$rYywxYw)ao Background thread listening for data from the server. Internally, it'll: * Answer to Ping events. * Acknowledge Pong events. * Put other events in the [_events][_events] queue that'll eventually be consumed by the user. Args: max_bytes: The maximum chunk size to read at each iteration. rNMessage too big Stream error))rrYis_set_wait_until_closed _read_streamrM receive_datarrrLrrrBr[rKrPongrVackrrrMessagelenrstrrryr MESSAGE_TOO_BIGrUputrmessage_finishedtype ReadErrorrr=rrr3 rc max_bytesrpartial_message_bufferpartial_message_sizerr event_typefull_message_events r:rrz$WebSocketSession._background_receives 59 , ((//1..t/@/@)L,,T2!__335#,E!%)<)<=#33ENN4DE!--4 KK--d34 !%)<)<=**..u}}= !%)G)GH**..0!%)?)?@,:V[V`V`beKfEJJ4E4E4Glqlvlv0ww,/); JJ{'B'BDUV LL,,-@A\A\^o-pq!$555=9> 6 6%** D 63;340 LL,,U3*.eJ1; B LL  24 5   s2C L L/H L L L A#NNNc" |jjs|j|jj|}|r t  |j }|a|j|j|}|sC|jtjd|jjt|jjsyy#t jj$rYywxYw#t$rYywxYwNzKeepalive ping timeout)rYrrrkr3rrL utilitiesLocalProtocolErrorryr rrUrr)rcinterval_secondstimeout_seconds should_close pong_callback acknowledgeds r:rvz+WebSocketSession._background_keepalive_ping*s ((//1#66t7I7I7N7NP`a %-'$(IIKM#.#'#:#:=;M;M#_L' ;#=#=?WX (()>)@A((//1 ((;;   s7A DCA=DC?<D>C??D D DcV |j\}}|j|g|i|}tjj ||ftjj \}}||vr t ||vsJ|j} | S#t$r} t | d} ~ wwxYw)N) return_when) rmrjrgrhrkFIRST_COMPLETEDr3result RuntimeError) rccallablerpkwargsexecutorshould_close_task todo_taskdone_rrs r:rz#WebSocketSession._wait_until_closed<s (*.*N*N*P 'H''B4B6BI!((---.&..>>.GD!!D(!m#$ $$%%'F  '-Q & 's)B B( B##B(cZ|jj|}|dk(r t|SNr9rKreadr=rcrrs r:rzWebSocketSession._read_streamOs*{{ * 3;-  r9)rKrrDintrErrF float | NonerGrrBrAreturnNone)rzMtuple[concurrent.futures.ThreadPoolExecutor, concurrent.futures.Future[bool]])rr-)r{ztype[BaseException] | Noner|zBaseException | Noner}zTracebackType | Nonerrr9)rrrzthreading.Eventrwsproto.events.Eventrrrrrrrrrrr(r typing.AnyrJSONModerrrfrrrrrrrrrrrrNr(rrrrrriNrrrr?rrrrrrrfloatrrrr)rz typing.Callable[..., TaskResult]rprrrrr+rrrr)r5r6r7__doc____annotations__rrr rrdrmrwr~rrrrrrrrrryrrrvrrr8r9r:r-r-Hs" 'E,8_7]$(NN!$ N  N *6 N)5N"N N>7 V7& 8, 8" 8 ! 8  821@$$=4/b&2P&2P. `$L>@$8AKWa &r9ceZdZUdZded<ded<ded<ded <eeeed d  dd Z e jdd Z dddZ ddZd dZd!dZd"d#dZd$d%dZd$d&dZd$d'dZd(d)dZd*d+dZd,dZd$d-dZd.dZy )/r1a Async context manager representing an opened WebSocket session. Internally, this session uses an anyio task group to manage background tasks. As a result, exceptions that are not caught inside the context manager and propagate out of the `async with` block will be wrapped in an [ExceptionGroup][ExceptionGroup]. To handle them, use the `except*` syntax: async with AsyncWebSocketSession(stream) as ws: try: data = await ws.receive_text() except WebSocketDisconnect: # Caught inside the context manager: plain exception. print("Connection closed") # If not caught inside: try: async with AsyncWebSocketSession(stream) as ws: data = await ws.receive_text() except* WebSocketDisconnect: # Propagated out of the context manager: wrapped in ExceptionGroup. print("Connection closed") Attributes: subprotocol (typing.Optional[str]): Optional protocol that has been accepted by the server. response (Response | None): The webSocket handshake response. r?r@rArBz?MemoryObjectSendStream[wsproto.events.Event | HTTPXWSException] _send_eventzBMemoryObjectReceiveStream[wsproto.events.Event | HTTPXWSException]_receive_eventNrCc ||_tjjtjj |_||_|j +|j jjd|_ nd|_ t|_ tj|_tj|_||_||_t'|t(rd|_d|_y||_||_yrI)rKrLrMrNrOrPrBrQrRr@rrVanyiorXrYrZr[r^r_rrr`rarbs r:rdzAsyncWebSocketSession.__init__|s !,,778N8N8U8UV  == $#}}44889QRD #D -/"[[] ::<'=$% f= >48D 137D 04SD 13QD 0r9c Ktjtjjt z\|_|_tj|_ |j 4d{|j4d{|j4d{|jj|j|j|j;|jj|j|j|j ||jj j#tj$d5|j'd{ddddddd{dddd{dddd{y787&77F#1swYExYw#|jj j#tj$d5|j'd{7dddw#1swYwxYwxYw7#1d{7swYxYw7#1d{7swYxYw7#1d{7swYyxYww)NT)shield)rcreate_memory_object_streamrLrrXrrrcreate_task_group_background_task_group start_soonrrr^r`rvra cancel_scopecancel CancelScoperyrls r:__asynccontextmanager__z-AsyncWebSocketSession.__asynccontextmanager__s050Q0Q NN #3 31 1 -$-',&=&=&?### ' 'T%8%8 ' '$:U:U ' '  ' ' 2 243K3KTMiMi j44@++66339988  ' ++88??A&&d3'**,&&' ' ' ' ' ' ' ' ' ''''++88??A&&d3'**,&&''' ' ' ' ' ' ' ' ' ' ' ' 's?A-I0/F'0I03IF*IIF- IA8H-F>:H-F2 )F0 *F2 .H-6 IH+I III I0!I"I0*I-I0F2 2F; 7H->;H( 9H HH  H( H%!H( (H-+I-H? 3H64H? ;III I  I II0I-!I$ "I-)I0cK|jj|\}}tjj |}|j |d{|S7w)aK Send a Ping message. Args: payload: Payload to attach to the Ping event. Internally, it's used to track this specific event. If left empty, a random one will be generated. Returns: An event that can be used to wait for the corresponding Pong response. Examples: Send a Ping and wait for the Pong pong_callback = await ws.ping() # Will block until the corresponding Pong is received. await pong_callback.wait() Nrrs r:rzAsyncWebSocketSession.pingsS(!..55g>##G,ii sAAAAcKddl} |jj|}|j4d{|jj |d{dddd{y7877 #1d{7swYyxYw#|j $r9}|jtjdd{7t|d}~wwxYww)a| Send an Event message. Mainly useful to send events that are not supported by the library. Most of the time, [ping()][httpx_ws.AsyncWebSocketSession.ping], [send_text()][httpx_ws.AsyncWebSocketSession.send_text], [send_bytes()][httpx_ws.AsyncWebSocketSession.send_bytes] and [send_json()][httpx_ws.AsyncWebSocketSession.send_json] are preferred. Args: event: The event to send. Raises: WebSocketNetworkError: A network error occured. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Send an event. event = await wsproto.events.Message(b"Hello!") ws.send(event) rNrrrs r:rzAsyncWebSocketSession.sends6  1??''.D'' . .kk''--- . . .- . . . .## 1**[779MN N N')q 0 1sC+BA,BA2A.A2 B&A0'B+C,B.A20B2B8A; 9BBCBC#C 9B<:C  CCcKtjj|}|j|d{y7w)a Send a text message. Args: data: The text to send. Raises: WebSocketNetworkError: A network error occured. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Send a text message. await ws.send_text("Hello!") rNrrs r:rzAsyncWebSocketSession.send_texts1(***5ii 5?=?cKtjj|}|j|d{y7w)a Send a bytes message. Args: data: The data to send. Raises: WebSocketNetworkError: A network error occured. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Send a bytes message. await ws.send_bytes(b"Hello!") rNrrs r:rz AsyncWebSocketSession.send_bytes s1(+++6iir%cK|dvsJtj|}|dk(r|j|d{y|j|j dd{y7.7w)a Send JSON data. Args: data: The data to send. Must be serializable by [json.dumps][json.dumps]. mode: The sending mode. Should either be `'text'` or `'bytes'`. Raises: WebSocketNetworkError: A network error occured. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Send JSON data. data = {"message": "Hello!"} await ws.send_json(data) r'r(Nrrrs r:rzAsyncWebSocketSession.send_json sd0))))**T* 6>..1 1 1///"8"8"AB B B 2 Bs!5A*A&(A* A(!A*(A*cVKtj|5|jjd{}dddt t r|t |t jjr t|j|j|S7d#1swYcxYww)a: Receive an event from the server. Mainly useful to receive raw [wsproto.events.Event][wsproto.events.Event]. Most of the time, [receive_text()][httpx_ws.AsyncWebSocketSession.receive_text], [receive_bytes()][httpx_ws.AsyncWebSocketSession.receive_bytes], and [receive_json()][httpx_ws.AsyncWebSocketSession.receive_json] are preferred. Args: timeout: Number of seconds to wait for an event. If `None`, will block until an event is available. Returns: A raw [wsproto.events.Event][wsproto.events.Event]. Raises: TimeoutError: No event was received before the timeout delay. WebSocketDisconnect: The server closed the websocket. WebSocketNetworkError: A network error occured. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Wait for an event until one is available. try: event = await ws.receive() except WebSocketDisconnect: print("Connection closed") Wait for an event for 2 seconds. try: event = await ws.receive(timeout=2.) except TimeoutError: print("No event received.") except WebSocketDisconnect: print("Connection closed") N) r fail_afterrrrrrLrrrrrrs r:rzAsyncWebSocketSession.receive?sX  g & 8--5577E 8 e- .K eW^^;; <%ejj%,,? ? 8 8 8s,B)BBBA B)BB&"B)cK|j|d{}t|tjjr |j St |7?w)a  Receive text from the server. Args: timeout: Number of seconds to wait for an event. If `None`, will block until an event is available. Returns: Text data. Raises: TimeoutError: No event was received before the timeout delay. WebSocketDisconnect: The server closed the websocket. WebSocketNetworkError: A network error occured. WebSocketInvalidTypeReceived: The received event was not a text message. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Wait for text until available. try: text = await ws.receive_text() except WebSocketDisconnect: print("Connection closed") Wait for text for 2 seconds. try: event = await ws.receive_text(timeout=2.) except TimeoutError: print("No text received.") except WebSocketDisconnect: print("Connection closed") Nrrs r:rz"AsyncWebSocketSession.receive_textssHPll7++ eW^^77 8:: *511,sAAAAcK|j|d{}t|tjjrt |j St|7Hw)a Receive bytes from the server. Args: timeout: Number of seconds to wait for an event. If `None`, will block until an event is available. Returns: Bytes data. Raises: TimeoutError: No event was received before the timeout delay. WebSocketDisconnect: The server closed the websocket. WebSocketNetworkError: A network error occured. WebSocketInvalidTypeReceived: The received event was not a bytes message. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Wait for bytes until available. try: data = await ws.receive_bytes() except WebSocketDisconnect: print("Connection closed") Wait for bytes for 2 seconds. try: data = await ws.receive_bytes(timeout=2.) except TimeoutError: print("No data received.") except WebSocketDisconnect: print("Connection closed") Nrrs r:rz#AsyncWebSocketSession.receive_bytessMPll7++ eW^^88 9$ $*511,sA"A A A"cK|dvsJ|dk(r|j|d{}n|dk(r|j|d{}tjS787w)a Receive JSON data from the server. The received data should be parseable by [json.loads][json.loads]. Args: timeout: Number of seconds to wait for an event. If `None`, will block until an event is available. mode: Receive mode. Should either be `'text'` or `'bytes'`. Returns: Parsed JSON data. Raises: TimeoutError: No event was received before the timeout delay. WebSocketDisconnect: The server closed the websocket. WebSocketNetworkError: A network error occured. WebSocketInvalidTypeReceived: The received event didn't correspond to the specified mode. Note: Exceptions not caught inside the context manager will be wrapped in an [ExceptionGroup][ExceptionGroup]. Use `except*` to catch them outside the `async with` block. Examples: Wait for data until available. try: data = await ws.receive_json() except WebSocketDisconnect: print("Connection closed") Wait for data for 2 seconds. try: data = await ws.receive_json(timeout=2.) except TimeoutError: print("No data received.") except WebSocketDisconnect: print("Connection closed") r'r(Nr)rrs r:rz"AsyncWebSocketSession.receive_jsonseZ)))) 6>**733D X ++G44Dzz$44s! AAAAAAcKddl}|jj|jjt jj jt jj jhvrt jj||}|jj|} |j4d{|jj|d{dddd{|jj!d{y7Z797+#1d{7swY;xYw#|j$rYOwxYw75w)a Close the WebSocket session. Internally, it'll send the [CloseConnection][wsproto.events.CloseConnection] event. *This method is automatically called when exiting the context manager.* Args: code: The integer close code to indicate why the connection has closed. reason: Additional reasoning for why the connection has closed. Examples: Close the WebSocket session. await ws.close() rN)rrYrrMrrLrrrrrrr[rKrraclosers r:ryzAsyncWebSocketSession.closes(   ??    . . < <    . . 5 5)  NN224@E??''.D ++22++++D11122kk  """ 212222''  "sB8E;E D& ED,.D(/D,3 E>D*?EE E!E&E(D,*E,D>2D5 3D>:EEEEEcKddl}d}d} |jjs*|j|d{}|jj ||jj D]}t|tj jrr|jj|j}|j4d{|jj|d{dddd{t|tj jr&|j j#|j$t|tj j&r|jj)t|tj j*rR|t-t|j.t0r|j.j3n |j.z }||kDre|j5t6j8dd{|j:jt=t6j8dd{n|j>s| |j.}n||j.z }no|&d}|j:j|d{nGtA|}|||j.z}d}d}|j:j|d{|j:j|d{|jjs)yy777_7R#1d{7swYcxYw7D7 77t7Q#|jB|jDtFf$rX|j5t6jHdd{7|j:jtKd{7YywxYww)am Background task listening for data from the server. Internally, it'll: * Answer to Ping events. * Acknowledge Pong events. * Put other events in the [_events][_events] queue that'll eventually be consumed by the user. Args: max_bytes: The maximum chunk size to read at each iteration. rNrr)&rrYrrrMrrrrLrrrBr[rKrrrVrrrrrrrrrryr rrrrrrrr=rrrs r:rrz)AsyncWebSocketSession._background_receive&s7 59 , A((//1!..y99,,T2!__335%7E!%)<)<=#33ENN4DE#'#3#3::"&++"3"3D"999:: !%)<)<=**..u}}= !%)G)GH**..0!%)?)?@,:V[V`V`beKfEJJ4E4E4Glqlvlv0ww,/);"&**[-H-HJ["\\\"&"2"2"7"7 3K4O4OQb c#"$555=9> 6 6%** D 63;340"&"2"2"7"7">>>*.eJ1; N M D&N0M41:N+M7,AN?M:ANM<$N,M>-!NO;NNMNM1 $M'%M1 , N7N:N<N>NAO8O+O8/O20O85O;7O88O;cK|jjstj|d{ |j d{}|7 tj|5|jd{ddd|jjsyy7s7\#t j j$rYywxYw7P#1swYOxYw#t$rX|jtjdd{7|jjt!d{7YwxYwwr)rYrrsleeprrLrrr)rkrryr rrrr)rcrrrs r:rvz0AsyncWebSocketSession._background_keepalive_pingfs $$++-++./ / / &*iik 1 *I))/:3+002223$$++- /!2$$77   333#I**[%?%?AYZZZ**//0E0GHHHIs3EB)EB-B+B-EC+C?CCC E'E+B--C  E C  ECCC,D? D  +D?6D97D?<E>D??EcvK|jj|d{}|dk(r t|S7wrrrs r:rz"AsyncWebSocketSession._read_streamxs6[[%%i00 3;-  1s 979)rKrrDrrErrFrrGrrBrArr)r2typing.AsyncGenerator[AsyncWebSocketSession, None]r)rrrz anyio.Eventrrrrrrfrrrr r r r r rr)r5r6r7rrrrr rrd contextlibasynccontextmanagerr!rrrrrrrrrryrrrvrr8r9r:r1r1Vs@PPVV 'E,8_7]$(R"R!$ R  R *6 R)5R"R RB##'$',2#1J..C>2h+2Z+2Z3 j"#H>A@I$r9cddtjtjdj ddd}|dj ||d<|S) Nupgrade websocketr13)rMr7zsec-websocket-keyzsec-websocket-versionz, rJ)base64 b64encodesecrets token_bytesdecoderz) subprotocolsrQs r: _get_headersrAsY #--g.A.A".EFMMgV!% G ,0IIl,C() Nr9c eZdZdZeeeeed ddZ e jdddde e e dd ddZ y) WebSocketClienta A sync WebSocket client. This class provides an API for connecting to WebSocket. Attributes: client: HTTPX client to use. max_message_size_bytes: Maximum incoming message size in bytes. Larger messages, including fragmented messages whose cumulative size exceeds the limit, close the connection with `MESSAGE_TOO_BIG`. Defaults to 65 KiB. queue_size: Size of the queue where the received messages will be held until they are consumed. If the queue is full, the client will stop receive messages from the server until the queue has room available. Defaults to 512. keepalive_ping_interval_seconds: Interval at which the client will automatically send a Ping event to keep the connection alive. Set it to `None` to disable this mechanism. Defaults to 20 seconds. keepalive_ping_timeout_seconds: Maximum delay the client will wait for an answer to its Ping event. If the delay is exceeded, [WebSocketNetworkError][httpx_ws.WebSocketNetworkError] will be raised and the connection closed. Defaults to 20 seconds. session_class: The session class to use. Defaults to [WebSocketSession][httpx_ws.WebSocketSession]. rDrErFrG session_classcX||_||_||_||_||_||_yrfclientrDrErFrGrErcrHrDrErFrGrEs r:rdzWebSocketClient.__init__2 &<#$/N,.L+*r9Nr@paramsrQcookiesauthfollow_redirectsr extensionsc #K|jjd||t|t|z|||||  5} | jdk7r t | |j | jd|j|j|j|j| } | 5| ddddddy#1swYxYw#1swYyxYww)a Start a sync WebSocket session. It returns a context manager that'll automatically call [close()][httpx_ws.WebSocketSession.close] when exiting. Args: url: The WebSocket URL. subprotocols: Optional list of subprotocols to negotiate with the server. params: Query parameters to include in the handshake request. headers: Headers to include in the handshake request. cookies: Cookies to include in the handshake request. auth: Authentication to use for the handshake request. follow_redirects: Whether to follow redirects on the handshake request. timeout: Timeout configuration for the handshake request. extensions: Request extensions for the handshake request. Returns: A [context manager][contextlib.AbstractContextManager] for [WebSocketSession][httpx_ws.WebSocketSession]. Examples: Initialize the client and connect to a WebSocket. with httpx2.Client() as client: ws_client = WebSocketClient(client) with ws_client.connect("http://localhost:8000/ws") as ws: message = ws.receive_text() print(message) ws.send_text("Hello!") GETrLrQrMrNrOrrPenetwork_streamrCN rHrKrrA status_coderrErPrDrErFrG rcurlr@rLrQrMrNrOrrPrBsessions r:connectzWebSocketClient.connectsj[[    G$|L'AA-!   ##s*+H55((##$45'+'B'B??040T0T/3/R/R! )G   -  ,  -  s5:CA)C%B;*C2 C;C CC C)rHrrDrrErrFrrGrrEztype[SyncSession]rr)rYrr@list[str] | NonerLQueryParamTypes | NonerQHeaderTypes | NonerMCookieTypes | NonerN#AuthTypes | UseClientDefault | NonerObool | UseClientDefaultrTimeoutTypes | UseClientDefaultrPRequestExtensions | Nonerz)typing.Generator[SyncSession, None, None])r5r6r7rrrr rr-rdr4contextmanagerr r[r8r9r:rCrCs!N'E,8_7]+;++!$ +  + *6 +)5+)+ +" *.)-&*&*4F4F3E/3K K' K ' K $ K$K2K2K1K-K 3KKr9rC) rDrErFrGr@rLrQrMrNrOrrPc # K|ddlm}|}ntj|}|5}t |||||}|j ||||| | | | |  5}|ddddddy#1swYxYw#1swYyxYww)a Start a sync WebSocket session. It returns a context manager that'll automatically call [close()][httpx_ws.WebSocketSession.close] when exiting. Args: url: The WebSocket URL. client: HTTPX client to use. If not provided, a default one will be initialized. max_message_size_bytes: Maximum incoming message size in bytes. Larger messages, including fragmented messages whose cumulative size exceeds the limit, close the connection with `MESSAGE_TOO_BIG`. Defaults to 65 KiB. queue_size: Size of the queue where the received messages will be held until they are consumed. If the queue is full, the client will stop receive messages from the server until the queue has room available. Defaults to 512. keepalive_ping_interval_seconds: Interval at which the client will automatically send a Ping event to keep the connection alive. Set it to `None` to disable this mechanism. Defaults to 20 seconds. keepalive_ping_timeout_seconds: Maximum delay the client will wait for an answer to its Ping event. If the delay is exceeded, [WebSocketNetworkError][httpx_ws.WebSocketNetworkError] will be raised and the connection closed. Defaults to 20 seconds. subprotocols: Optional list of subprotocols to negotiate with the server. params: Query parameters to include in the handshake request. headers: Headers to include in the handshake request. cookies: Cookies to include in the handshake request. auth: Authentication to use for the handshake request. follow_redirects: Whether to follow redirects on the handshake request. timeout: Timeout configuration for the handshake request. extensions: Request extensions for the handshake request. Returns: A [context manager][contextlib.AbstractContextManager] for [WebSocketSession][httpx_ws.WebSocketSession]. Examples: Without explicit HTTPX client. with connect_ws("http://localhost:8000/ws") as ws: message = ws.receive_text() print(message) ws.send_text("Hello!") With explicit HTTPX client. with httpx2.Client() as client: with connect_ws("http://localhost:8000/ws", client) as ws: message = ws.receive_text() print(message) ws.send_text("Hello!") Nr )rrHrDrErFrGrK)_clientrr4 nullcontextrCr[)rYrHrDrErFrGr@rLrQrMrNrOrrPr owned_client ws_clientr8s r: connect_wsrksp~$BH( !--f5 ##9!,K+I     %-!  O   s4(B+A7A+A7" B+A4 0A77B<Bc eZdZdZeeeeed ddZ e jdddde e e dd ddZ y) AsyncWebSocketClienta An async WebSocket client. This class provides an API for connecting to WebSocket. Attributes: client: HTTPX client to use. max_message_size_bytes: Maximum incoming message size in bytes. Larger messages, including fragmented messages whose cumulative size exceeds the limit, close the connection with `MESSAGE_TOO_BIG`. Defaults to 65 KiB. queue_size: Size of the queue where the received messages will be held until they are consumed. If the queue is full, the client will stop receive messages from the server until the queue has room available. Defaults to 512. keepalive_ping_interval_seconds: Interval at which the client will automatically send a Ping event to keep the connection alive. Set it to `None` to disable this mechanism. Defaults to 20 seconds. keepalive_ping_timeout_seconds: Maximum delay the client will wait for an answer to its Ping event. If the delay is exceeded, [WebSocketNetworkError][httpx_ws.WebSocketNetworkError] will be raised in an [ExceptionGroup][ExceptionGroup] and the connection closed. Defaults to 20 seconds. session_class: The session class to use. Defaults to [AsyncWebSocketSession][httpx_ws.AsyncWebSocketSession]. rDcX||_||_||_||_||_||_yrfrGrIs r:rdzAsyncWebSocketClient.__init__rJr9NrKc  K|jjd||t|t|z|||||  4d{} | jdk7r t | |j | jd|j|j|j|j| } | 4d{| dddd{dddd{y77,7#1d{7swY)xYw7 #1d{7swYyxYww)a Start an async WebSocket session. It returns an async context manager that'll automatically call [close()][httpx_ws.AsyncWebSocketSession.close] when exiting. Args: url: The WebSocket URL. subprotocols: Optional list of subprotocols to negotiate with the server. params: Query parameters to include in the handshake request. headers: Headers to include in the handshake request. cookies: Cookies to include in the handshake request. auth: Authentication to use for the handshake request. follow_redirects: Whether to follow redirects on the handshake request. timeout: Timeout configuration for the handshake request. extensions: Request extensions for the handshake request. Returns: An [async context manager][contextlib.AbstractAsyncContextManager] for [AsyncWebSocketSession][httpx_ws.AsyncWebSocketSession]. Examples: Initialize the client and connect to a WebSocket. async with httpx2.AsyncClient() as client: ws_client = AsyncWebSocketClient(client) async with ws_client.connect("http://localhost:8000/ws") as ws: message = await ws.receive_text() print(message) await ws.send_text("Hello!") rRrSNrTrUrCrVrXs r:r[zAsyncWebSocketClient.connectsj;;%%  G$|L'AA-!&    ##s*+H55((##$45'+'B'B??040T0T/3/R/R! )G     -   ,     -    s>DCDA-C91C2C95C"; C9C C9 DC7DC9 C9"C4 (C+)C4 0C97D9D ?D D D)rHrrDrrErrFrrGrrEztype[AsyncSession]rr)rYrr@r\rLr]rQr^rMr_rNr`rOrarrbrPrcrz)typing.AsyncGenerator[AsyncSession, None])r5r6r7rrrr rr1rdr4r5r r[r8r9r:rmrms!N'E,8_7],A++!$ +  + *6 +)5+*+ +"## *.)-&*&*4F4F3E/3K K' K ' K $ K$K2K2K1K-K 3K$Kr9rmc K|ddlm}|}ntj|}|4d{}t |||||}|j ||||| | | | |  4d{}|dddd{dddd{y7]7,7#1d{7swY)xYw7 #1d{7swYyxYww)a@ Start an async WebSocket session. It returns an async context manager that'll automatically call [close()][httpx_ws.AsyncWebSocketSession.close] when exiting. Args: url: The WebSocket URL. client: HTTPX client to use. If not provided, a default one will be initialized. max_message_size_bytes: Maximum incoming message size in bytes. Larger messages, including fragmented messages whose cumulative size exceeds the limit, close the connection with `MESSAGE_TOO_BIG`. Defaults to 65 KiB. queue_size: Size of the queue where the received messages will be held until they are consumed. If the queue is full, the client will stop receive messages from the server until the queue has room available. Defaults to 512. keepalive_ping_interval_seconds: Interval at which the client will automatically send a Ping event to keep the connection alive. Set it to `None` to disable this mechanism. Defaults to 20 seconds. keepalive_ping_timeout_seconds: Maximum delay the client will wait for an answer to its Ping event. If the delay is exceeded, [WebSocketNetworkError][httpx_ws.WebSocketNetworkError] will be raised in an [ExceptionGroup][ExceptionGroup] and the connection closed. Defaults to 20 seconds. subprotocols: Optional list of subprotocols to negotiate with the server. params: Query parameters to include in the handshake request. headers: Headers to include in the handshake request. cookies: Cookies to include in the handshake request. auth: Authentication to use for the handshake request. follow_redirects: Whether to follow redirects on the handshake request. timeout: Timeout configuration for the handshake request. extensions: Request extensions for the handshake request. Returns: An [async context manager][contextlib.AbstractAsyncContextManager] for [AsyncWebSocketSession][httpx_ws.AsyncWebSocketSession]. Examples: Without explicit HTTPX client. async with aconnect_ws("http://localhost:8000/ws") as ws: message = await ws.receive_text() print(message) await ws.send_text("Hello!") With explicit HTTPX client. async with httpx2.AsyncClient() as client: async with aconnect_ws("http://localhost:8000/ws", client) as ws: message = await ws.receive_text() print(message) await ws.send_text("Hello!") Nr )rrfrK)rgrr4rhrmr[)rYrHrDrErFrGr@rLrQrMrNrOrrPrrirjr8s r: aconnect_wsrq sp~)LWM !--f5 v(#9!,K+I  $$ %-!%   O       s,B>B B>/B)!B"B)%B+ B)6B7B); B>B'B>B)B)B$ BB$ B)'B>)B;/B2 0B;7B>)r@r\rzdict[str, typing.Any]rf)rYrrHz Client | NonerDrrErrFrrGrr@r\rLr]rQr^rMr_rNr`rOrarrbrPrcrz.typing.Generator[WebSocketSession, None, None])rYrrHzAsyncClient | NonerDrrErrFrrGrr@r\rLr]rQr^rMr_rNr`rOrarrbrPrcrr3)P __future__rr;concurrent.futuresrgr4rrSr=sysrWtypingtypesr version_infortyping_extensionsrrLwsproto.utilitiesanyio.streams.memoryrr wsproto.frame_protocolr rgr _configr rrr_modelsr _exceptionsrrrrr_pingrr _transportr TYPE_CHECKINGrrrrrrr _typesr!r"r#r$r%r&Literalrr*r+r,r0 Exceptionr3r=r-AsyncContextManagerMixinr1rAGenericrCrdrkrmr5rqr8r9r:rs;"    w) R.( 17 ;??" >>* +v~~n- V^^L ) m+=GYZ ~-DNef  )  ) KK\fE::fR "  Afnn[1AH !q#A(4[3Y%)%)"&"&0B0B/A+/q q q q  q &2 q%1q#q #q q q .q.q-q)q 4!qqhA6>>,7AH "&q#A(4[3Y%)%)"&"&0B0B/A+/q q q q  q &2 q%1q#q #q q q .q.q-q)q 8!q qr9