RPf,ddlmZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lmZmZmZddlm Z ddl!m"Z"ddl#m$Z$gdZ%Gdde"Z&dddddddddeddddddd ddZ' d ddZ(y)) annotationsN)AnyOptionalSequenceType)ClientProtocol) HeadersLike)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENT)Response) CONNECTINGOPENEvent) LoggerLikeOrigin Subprotocol) parse_uri) Connection)Deadline)connect unix_connectClientConnectioncneZdZdZdd d fdZdedf d dZd fd Zd fd ZxZ S) ra Threaded 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 P/var/www/fastapitest/venv/lib/python3.12/site-packages/websockets/sync/client.pyr%zClientConnection.__init__1s2 &__.   '  Nc|jt5|jj|_|%|jj j ||||jj d<|jj|jddd|jj|s5|j|jjtd|j5|j|jjtd|jj t"urO|jj|j$|j|jj|jj&|jj&y#1swY3xYw)z1 Perform the opening handshake. )expected_stateNz User-Agentztimed out during handshakez"connection closed during handshake) send_contextrr(rrequestheadersupdate send_requestr#wait close_socketrecv_events_threadjoin TimeoutErrorresponseConnectionErrorstaterr handshake_exc)r&additional_headersuser_agent_headertimeouts r* handshakezClientConnection.handshake@sp  j  9 5==002DL!- $$++,>? ,5F $$\2 MM & &t|| 4  5!!&&w/      # # ( ( *;< < ==      # # ( ( *!"FG G ==  d *  # # ( ();); <      # # ( ( * == & & 2---- - 3/ 5 5s BGGc|j4t|tsJ||_|jj yt ||y)z. Process one incoming event. N)r8 isinstancerr#setr$ process_event)r&eventr)s r*rCzClientConnection.process_eventdsG == eX. ..!DM    " " $ G !% (r+c t||jjy#|jjwxYw)zI Read incoming data from the socket and process events. N)r$ recv_eventsr#rB)r&r)s r*rFzClientConnection.recv_eventsrs:  % G  !    " " $D   " " $s ,A)r'z socket.socketr(r r Optional[float]returnNone)r<Optional[HeadersLike]r= Optional[str]r>rGrHrI)rDrrHrI)rHrI) __name__ __module__ __qualname____doc__r%rr?rCrF __classcell__)r)s@r*rrs~8*,    !  '    "59+5#' ".1".)".! ".  ".H ) % %r+rFdeflateri)sock ssl_contextserver_hostnameunixpathorigin extensions subprotocolsr<r= compression open_timeoutr max_sizeloggercreate_connectionc t|}|js | td|r| | td||td|J| t|| dk(r t |}n| t d| t | }|t} ||rgtjtjtj}|j|j|J|j|n:tj|j|j f|j}|jd|s/|j#tj$tj&d|jrg|t)j*}| |j}|j|j|j-||}|jdt/||||t0|| }|||| }|j3| | |j|S#t4$r||j7wxYw) 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:: async with websockets.sync.client.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_context: 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. 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. Nz5ssl_context argument is incompatible with a ws:// URIzmissing path argumentz(path and sock arguments are incompatiblerQzunsupported compression: T)rT)rWrXrYr:r\r]r)rsecure TypeErrorr r ValueErrorrrr'AF_UNIX SOCK_STREAM settimeoutr>rr^hostport setsockopt IPPROTO_TCP TCP_NODELAYsslcreate_default_context wrap_socketr rr? Exceptionclose)urirRrSrTrUrVrWrXrYr<r=rZr[r r\r]r^wsurideadliner( connections r*rr~sFf cNE <rs" 00#(4L+..44" :b%zb%R%),0%)#=A4804'1!*$&%'##':>1n n " n * n # n n n n;n2n.n%nn""#n$#%n()n, !-n081n23nd< < <<