vjl,xddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZmZmZdd lmZddlm Z ddl!m"Z"gdZ#Gdde Z$dddddddeddddddd ddZ% d ddZ&y)) annotationsN)AnySequence)ClientProtocol) HeadersLike)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENTResponse) CONNECTINGEvent) LoggerLikeOrigin Subprotocol) parse_uri) Connection)Deadline)connect unix_connectClientConnectioncneZdZdZdd d fdZdedf d dZd fd Zd fd ZxZ S) ra :mod:`threading` implementation of a WebSocket client connection. :class:`ClientConnection` provides :meth:`recv` and :meth:`send` methods for receiving and sending messages. It supports iteration to receive messages:: for message in websocket: process(message) The iterator exits normally when the connection is closed with close code 1000 (OK) or 1001 (going away) or without a close code. It raises a :exc:`~websockets.exceptions.ConnectionClosedError` when the connection is closed with any other code. Args: socket: Socket connected to a WebSocket server. protocol: Sans-I/O connection. close_timeout: Timeout for closing the connection in seconds.  close_timeoutc`|tj|_t||||y)Nr) threadingr response_rcvdsuper__init__)selfsocketprotocolr __class__s g/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/websockets/sync/client.pyr"zClientConnection.__init__1s2 &__.   '  Nc |jt5|jj|_|%|jj j ||||jj d<|jj|jddd|jj|s td|jj|jjy#1swY\xYw)z1 Perform the opening handshake. )expected_stateNz User-Agentztimed out during handshake) send_contextrr%rrequestheadersupdate send_requestr wait TimeoutError handshake_exc)r#additional_headersuser_agent_headertimeouts r' handshakezClientConnection.handshake@s  j  9==002DL!- $$++,>? ,5F $$\2 MM & &t|| 4 :!!&&w/;< < == & & 2---- - 3: 9s BC99Dc|j4t|tsJ||_|jj yt ||y)z. Process one incoming event. N)response isinstancer r setr! process_event)r#eventr&s r'r;zClientConnection.process_event\sG == eX. ..!DM    " " $ G !% (r(c t||jjy#|jjwxYw)zI Read incoming data from the socket and process events. N)r! recv_eventsr r:)r#r&s r'r>zClientConnection.recv_eventsjs:  % G  !    " " $D   " " $s ,A)r$z socket.socketr%rr float | NonereturnNone)r3HeadersLike | Noner4 str | Noner5r?r@rA)r<rr@rA)r@rA) __name__ __module__ __qualname____doc__r"r r6r;r> __classcell__)r&s@r'rrs}8')    !  $    "26(2 $ ...&. .  .8 ) % %r(rdeflateri)socksslserver_hostnameorigin extensions subprotocolsr3r4 compression open_timeoutrmax_sizeloggercreate_connectionc |/d|vr+|jd}tjdtt |}|j s | t d|jdd}|jdd}|r| | t d| | t d | t|| d k(r t|}n| td | t| }|t} ||rgtjtjtj}|j|j!|J|j#|nL|j%d |j!tj&|j(|j*ffi|}|jd|s/|j-tj.tj0d |j rg|t3j4}| |j(}|j|j!|j7||}|jdt9||||| | }|||| } |j?|||j!|S#t:$r||j=wxYw#t:$r,|jA|jBjEwxYw)a Connect to the WebSocket server at ``uri``. This function returns a :class:`ClientConnection` instance, which you can use to send and receive messages. :func:`connect` may be used as a context manager:: from websockets.sync.client import connect with connect(...) as websocket: ... The connection is closed automatically when exiting the context. Args: uri: URI of the WebSocket server. sock: Preexisting TCP socket. ``sock`` overrides the host and port from ``uri``. You may call :func:`socket.create_connection` to create a suitable TCP socket. ssl: Configuration for enabling TLS on the connection. server_hostname: Host name for the TLS handshake. ``server_hostname`` overrides the host name from ``uri``. origin: Value of the ``Origin`` header, for servers that require it. extensions: List of supported extensions, in order in which they should be negotiated and run. subprotocols: List of supported subprotocols, in order of decreasing preference. additional_headers (HeadersLike | None): Arbitrary HTTP headers to add to the handshake request. user_agent_header: Value of the ``User-Agent`` request header. It defaults to ``"Python/x.y.z websockets/X.Y"``. Setting it to :obj:`None` removes the header. compression: The "permessage-deflate" extension is enabled by default. Set ``compression`` to :obj:`None` to disable it. See the :doc:`compression guide <../../topics/compression>` for details. open_timeout: Timeout for opening the connection in seconds. :obj:`None` disables the timeout. close_timeout: Timeout for closing the connection in seconds. :obj:`None` disables the timeout. max_size: Maximum size of incoming messages in bytes. :obj:`None` disables the limit. logger: Logger for this client. It defaults to ``logging.getLogger("websockets.client")``. See the :doc:`logging guide <../../topics/logging>` for details. create_connection: Factory for the :class:`ClientConnection` managing the connection. Set it to a wrapper or a subclass to customize connection handling. Any other keyword arguments are passed to :func:`~socket.create_connection`. Raises: InvalidURI: If ``uri`` isn't a valid WebSocket URI. OSError: If the TCP connection fails. InvalidHandshake: If the opening handshake fails. TimeoutError: If the opening handshake times out. N ssl_contextzssl_context was renamed to sslz-ssl argument is incompatible with a ws:// URIunixFpathzmissing path argumentz(path and sock arguments are incompatiblerIzunsupported compression: r5T)rL)rMrNrOrRrSr)#popwarningswarnDeprecationWarningrsecure TypeErrorr r ValueErrorrrr$AF_UNIX SOCK_STREAM settimeoutr5r setdefaultrThostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socketr Exceptioncloser6 close_socketrecv_events_threadjoin)urirJrKrLrMrNrOr3r4rPrQrrRrSrTkwargswsurirWrXdeadliner% connections r'rrvsn {}.jj' ,  cNE <rsn"  #(4L+)(44" :Z%zZ%B"&(,"& :>15-1$.'!#"$ $7;-| |  | & |  | |8|/|+|"|||  !|$%|( )|,5-|./|01|@< < <<