U gl,@sRddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZmZmZdd lmZddlm Z ddl!m"Z"dddgZ#Gddde Z$dddddddeddddddddddddddd ddd!d!d"d#d$d%dd&d'dZ%d*ddd%dd(d)dZ&dS)+) annotationsN)AnySequence)ClientProtocol) HeadersLike)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENTResponse) CONNECTINGEvent) LoggerLikeOrigin Subprotocol) parse_uri) Connection)Deadlineconnect unix_connectClientConnectioncsteZdZdZdddddddfd d Zd ed fd d dddddZdddfdd Zddfdd ZZ 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_timeoutz socket.socketr float | NoneNone)socketprotocolrreturncs$|t|_tj|||ddS)Nr) threadingr response_rcvdsuper__init__)selfrrr __class__:/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/sync/client.pyr$1s zClientConnection.__init__NHeadersLike | None str | None)additional_headersuser_agent_headertimeoutr c Cs|jtdJ|j|_|dk r0|jj||dk rD||jjd<|j|jW5QRX|j |spt d|jj dk r|jj dS)z1 Perform the opening handshake. )Zexpected_stateNz User-Agentztimed out during handshake) Z send_contextr rrrequestheadersupdate send_requestr"wait TimeoutErrorZ handshake_exc)r%r,r-r.r(r(r) handshake@s     zClientConnection.handshaker)eventr cs:|jdkr*t|tst||_|jn t|dS)z. Process one incoming event. N)response isinstancer AssertionErrorr"setr# process_event)r%r6r&r(r)r;\s   zClientConnection.process_event)r cs ztW5|jXdS)zI Read incoming data from the socket and process events. N)r"r:r# recv_events)r%r&r(r)r<jszClientConnection.recv_events) __name__ __module__ __qualname____doc__r$r r5r;r< __classcell__r(r(r&r)rsdeflateri)socksslserver_hostnameorigin extensions subprotocolsr,r- compression open_timeoutrmax_sizeloggercreate_connectionstrzsocket.socket | Nonezssl_module.SSLContext | Noner+z Origin | Nonez'Sequence[ClientExtensionFactory] | NonezSequence[Subprotocol] | Noner*rz int | NonezLoggerLike | Noneztype[ClientConnection] | Noner)urirCrDrErFrGrHr,r-rIrJrrKrLrMkwargsr cKsX|dkr&d|kr&|d}tdtt|}|jsD|dk rDtd|dd}|dd}|r|dkrz|dkrztdn|dk r|dk rtd |dk rt|| d krt|}n| dk rt d | t | }|dkrt }z|dkrZ|r*t t j t j}|||dk st||n&|d |t j|j|jff|}|d|sr|t jt jd |jr|dkrt}|dkr|j}|||j||d}|dt||||| | d}|||| d}Wn*tk r|dk r |YnXz| |||Wn*tk rR|!|j"#YnX|S)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 incompatiblerBzunsupported compression: r.T)rE)rFrGrHrKrLr)$popwarningswarnDeprecationWarningrsecure TypeErrorr r ValueErrorrrrAF_UNIX SOCK_STREAM settimeoutr.r9r setdefaultrMhostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socketr Exceptioncloser5Z close_socketZrecv_events_threadjoin)rOrCrDrErFrGrHr,r-rIrJrrKrLrMrPZwsurirRrSdeadliner connectionr(r(r)rvsW               )rSrOrPr cKsD|dkr.|ddkr*|ddkr*d}nd}tf|d|d|S)a Connect to a WebSocket server listening on a Unix socket. This function accepts the same keyword arguments as :func:`connect`. 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`` is provided, to ``wss://localhost/``. NrDrQzws://localhost/zwss://localhost/T)rOrRrS)getr)rSrOrPr(r(r)r5s )NN)' __future__rrrDrdr!rUtypingrrclientrZdatastructuresrZextensions.baserZextensions.permessage_deflater r0r Zhttp11r r rr rrrrrOrrkrutilsr__all__rrrr(r(r(r)sJ          a2A