o /h,@s6ddlmZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lmZmZmZddlm Z ddl!m"Z"ddl#m$Z$gdZ%Gddde"Z&dddddddddedddddddd?d9d:Z'  d@dAd=d>Z(dS)B) annotationsN)AnyOptionalSequenceType)ClientProtocol) HeadersLike)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENT)Response) CONNECTINGOPENEvent) LoggerLikeOrigin Subprotocol) parse_uri) Connection)Deadline)connect unix_connectClientConnectioncsVeZdZdZdddfd d ZdedfdddZdfdd Zdfdd ZZ 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_timeoutsocket socket.socketprotocolrrOptional[float]returnNonecs$|t|_tj|||ddS)Nr) threadingr response_rcvdsuper__init__)selfrr!r __class__O/var/www/html/govbot/env/lib/python3.10/site-packages/websockets/sync/client.pyr(1s  zClientConnection.__init__Nadditional_headersOptional[HeadersLike]user_agent_header Optional[str]timeoutcCs|jtd*|j|_|dur|jj||dur"||jjd<|j|jWdn1s3wY|j |sK| |j t d|jdur]| |j td|jjturs|j |j| |j |jjdur}|jjdS)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)r.r0r2r,r,r- handshake@s.         zClientConnection.handshakeeventrcs<|jdurt|ts J||_|jdSt|dS)z. Process one incoming event. N)r> isinstancerr&setr' process_event)r)rCr*r,r-rFds zClientConnection.process_eventcs(z tW|jdS|jw)zI Read incoming data from the socket and process events. N)r' recv_eventsr&rE)r)r*r,r-rGrs zClientConnection.recv_events)rr r!rrr"r#r$)r.r/r0r1r2r"r#r$)rCrr#r$)r#r$) __name__ __module__ __qualname____doc__r(r rBrFrG __classcell__r,r,r*r-rs $rFdeflateri)sock ssl_contextserver_hostnameunixpathorigin extensions subprotocolsr.r0 compression open_timeoutrmax_sizeloggercreate_connectionuristrrNOptional[socket.socket]rOOptional[ssl.SSLContext]rPr1rQboolrRrSOptional[Origin]rT*Optional[Sequence[ClientExtensionFactory]]rUOptional[Sequence[Subprotocol]]r.r/r0rVrWr"rrX Optional[int]rYOptional[LoggerLike]rZ Optional[Type[ClientConnection]]r#c Cst|}|js|durtd|r*|dur|durtd|dur)|dur)tdn|dus0J|dur8t|| dkrAt|}n | durLtd| t| }|durVt}z|dur|rxttj tj }| | |dusrJ| |n t|j|jf| }| d|s|tjtjd|jr|durt}|dur|j}| | |j||d}| dt||||t||d }|||| d }|| | | W|Sty|dur|w) 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 incompatiblerMzunsupported compression: T)rP)rSrTrUr@rXrYr)rsecure TypeErrorr r ValueErrorrrrAF_UNIX SOCK_STREAM settimeoutr2rrZhostport setsockopt IPPROTO_TCP TCP_NODELAYsslcreate_default_context wrap_socketrrrB Exceptionclose)r[rNrOrPrQrRrSrTrUr.r0rVrWrrXrYrZwsurideadliner! connectionr,r,r-r~sS         rkwargsrcKs6|dur|ddurd}nd}td|d|d|S)a Connect to a WebSocket server listening on a Unix socket. This function is identical to :func:`connect`, except for the additional ``path`` argument. It's only available on Unix. It's mainly useful for debugging servers listening on Unix sockets. Args: path: File system path to the Unix socket. uri: URI of the WebSocket server. ``uri`` defaults to ``ws://localhost/`` or, when a ``ssl_context`` is provided, to ``wss://localhost/``. NrOzws://localhost/zwss://localhost/T)r[rQrRr,)getr)rRr[ryr,r,r-r/s r)$r[r\rNr]rOr^rPr1rQr_rRr1rSr`rTrarUrbr.r/r0r1rVr1rWr"rr"rXrcrYrdrZrer#r)NN)rRr1r[r1ryrr#r)) __future__rrrqr%typingrrrrclientrdatastructuresr extensions.baser extensions.permessage_deflater r6r httpr http11rr!rrrrrrr[rrxrutilsr__all__rrrr,r,r,r-sP           i3