o _i\V@s:ddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZmZdd l m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)ddl m*Z*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4gdZ5Gddde2Z6ddddddddedddddddddddddbdDdEZ7  dcdddGdHZ8zddIl9m:Z:ddJl;m$Z<Wne=ydedQdRZ>Ynwe:j?e:j?e:j@e:j@dSZAddTddTdSZBdedUdRZ>dfdWdXZCddddYdgdZd[ZDe d\ZEe d]e d^eEfd_ZFGd`dadaZGdS)h) annotationsN)Sequence)AnyCallableLiteralTypeVarcast)ClientProtocol) HeadersLike)InvalidProxyMessageInvalidProxyStatus ProxyError)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENTResponse) CONNECTINGEvent)Proxy get_proxy parse_proxyprepare_connect_request) StreamReader) BytesLike LoggerLikeOrigin Subprotocol) WebSocketURI parse_uri) Connection)Deadline)connect unix_connectClientConnectioncs\eZdZdZdddddd"fddZdedfd#ddZd$fdd Zd%fd d! ZZ S)&r&a+ :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 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. The ``ping_interval``, ``ping_timeout``, ``close_timeout``, and ``max_queue`` arguments have the same meaning as in :func:`connect`. Args: socket: Socket connected to a WebSocket server. protocol: Sans-I/O connection.   ping_interval ping_timeout close_timeout max_queuesocket socket.socketprotocolr r+ float | Noner,r-r.*int | None | tuple[int | None, int | None]returnNonecs*|t|_tj||||||ddS)Nr*) threadingr response_rcvdsuper__init__)selfr/r1r+r,r-r. __class__X/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/websockets/sync/client.pyr97s  zClientConnection.__init__Nadditional_headersHeadersLike | Noneuser_agent_header str | NonetimeoutcCs|jtd,|j|_|dur|jj||dur$|jjd||j|jWdn1s5wY|j |sDt d|jj durN|jj dS)z1 Perform the opening handshake. )expected_stateNz User-Agentz.timed out while waiting for handshake response) send_contextrr1r$requestheadersupdate setdefault send_requestr7wait TimeoutError handshake_exc)r:r?rArCr=r=r> handshakeLs   zClientConnection.handshakeeventrcs<|jdurt|ts J||_|jdSt|dS)z. Process one incoming event. N)response isinstancerr7setr8 process_event)r:rOr;r=r>rShs zClientConnection.process_eventcs(z tW|jdS|jw)zI Read incoming data from the socket and process events. N)r8 recv_eventsr7rR)r:r;r=r>rTvs zClientConnection.recv_events)r/r0r1r r+r2r,r2r-r2r.r3r4r5)r?r@rArBrCr2r4r5)rOrr4r5)r4r5) __name__ __module__ __qualname____doc__r9rrNrSrT __classcell__r=r=r;r>r&s r&deflateTr(r'ir))socksslserver_hostnameorigin extensions subprotocols compressionr?rAproxy proxy_sslproxy_server_hostname open_timeoutr+r,r-max_sizer.loggercreate_connectionuristrr[socket.socket | Noner\ssl_module.SSLContext | Noner]rBr^ Origin | Noner_'Sequence[ClientExtensionFactory] | Noner`Sequence[Subprotocol] | Nonerar?r@rArbstr | Literal[True] | Nonercrdrer2r+r,r-rfr3r.rgLoggerLike | Nonerhtype[ClientConnection] | Nonekwargsrr4cKs.|durd|vr|d}tdtt|}|js"|dur"td|dd}|dd}|rH|dur<|dur|jdddkr|jdkr| durtdt|||f| | | d|}ntd|d|t j|j|jffi|}|d|s|t jt jd |jrK|durt}|dur%|j}||| dur9|j ||d}n t!|||d}t"t j |}|dt#||||||d}|||||||d}Wnt$yr|durq|%wz |&|| |Wnt$y|'|j()w|*|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. 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. 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. proxy: If a proxy is configured, it is used by default. Set ``proxy`` to :obj:`None` to disable the proxy or to the address of a proxy to override the system configuration. See the :doc:`proxy docs <../../topics/proxies>` for details. proxy_ssl: Configuration for enabling TLS on the proxy connection. proxy_server_hostname: Host name for the TLS handshake with the proxy. ``proxy_server_hostname`` overrides the host name from ``proxy``. open_timeout: Timeout for opening the connection in seconds. :obj:`None` disables the timeout. ping_interval: Interval between keepalive pings in seconds. :obj:`None` disables keepalive. ping_timeout: Timeout for keepalive pings in seconds. :obj:`None` disables timeouts. 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. You may pass a ``(max_message_size, max_fragment_size)`` tuple to set different limits for messages and fragments when you expect long messages sent in short fragments. max_queue: High-water mark of the buffer where frames are received. It defaults to 16 frames. The low-water mark defaults to ``max_queue // 4``. You may pass a ``(high, low)`` tuple to set the high-water and low-water marks. If you want to disable flow control entirely, you may set it to ``None``, although that's a bad idea. 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 incompatiblerZzunsupported compression: Tsockssource_address) local_addrhttphttpsz8proxy_ssl argument is incompatible with an http:// proxyrAr\r]zunsupported proxyrCr])r^r_r`rfrgr*)+popwarningswarnDeprecationWarningr secure ValueErrorrrrr#r&r/AF_UNIX SOCK_STREAM settimeoutrCr$rschemeconnect_socks_proxyconnect_http_proxyAssertionErrorrIrhhostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socket SSLSSLSocketrr ExceptioncloserN close_socketrecv_events_threadjoinstart_keepalive)rir[r\r]r^r_r`rar?rArbrcrdrer+r,r-rfr.rgrhrsws_urirurvdeadline proxy_parsedsock_2r1 connectionr=r=r>r$sp                    r$rvcKsD|dur|ddur|ddurd}nd}td|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/``. Nr\rtzws://localhost/zwss://localhost/T)rirurvr=)getr$)rvrirsr=r=r>r%s r%) ProxyType)rrrrrr#r0cKstd)Nz6connecting through a SOCKS proxy requires python-socks) ImportError)rbrrrsr=r=r>rsr)socks5hsocks5socks4asocks4Fc Kstt|j|j|j|j|jt|j}|d| z |j |j|jfi|WSt t t j fy4tyD}ztd|d}~ww)z0Connect via a SOCKS proxy and return the socket.rCz failed to connect to SOCKS proxyN) SocksProxySOCKS_PROXY_TYPESrrrusernamepasswordSOCKS_PROXY_RDNSrIrCr$OSErrorrLr/rr)rbrrrs socks_proxyexcr=r=r>rs" rc Cst}tj|j|j|jdd}zsz |||d}|r&| |n| t |qt yg}z-t |jts>J|j}d|jkrLdkr_nt||WYd}~W|dSt|d}~wtjyrtdty}ztd|d}~ww|dw)NT)rbii,z(timed out while connecting to HTTP proxyz0did not receive a valid HTTP response from proxy)rrparse read_line read_exact read_to_eofrrCrecv feed_datafeed_eofnext StopIterationrQvalue status_coder r/rLrr )r[rreaderparserdatarrPr=r=r>read_connect_responsesH    rr~cKs|d|tj|j|jffi|}|jdkr<|dur"t}|dur)|j}| ||j ||d}| d| t |||zt ||W|StyY|w)NrCr}r)rIrCr/rhrrrrrrrsendallrrrr)rbrrrAr\r]rsr[r=r=r>rs$     rTF.)boundc@sTeZdZdZdZ d$d%d d Zd&ddZd'ddZd(ddZd)ddZ d*d"d#Z dS)+rzr Socket-like object providing TLS-in-TLS. Only methods that are used by websockets are implemented. iNr[r0rtssl_module.SSLContextr]rBr4r5cCsBt|_t|_||_|j|j|j|d|_||jjdS)Nr) r MemoryBIOincomingoutgoing ssl_socketwrap_bio ssl_objectrun_io do_handshake)r:r[rtr]r=r=r>r96s  zSSLSSLSocket.__init__funcCallable[..., T]argsrrcGs d}d}z||}Wntjyd}Yn tjy"d}Ynw|j}|r0|j||rH|j|j}|rB|j |n|j q|rKq|S)NTF) rSSLWantReadErrorSSLWantWriteErrorrreadrrr recv_bufsizerwrite write_eof)r:rr want_read want_writeresultrr=r=r>rFs,    zSSLSSLSocket.run_iobuflenintbytescCs*z ||jj|WStjyYdSw)N)rrrr SSLEOFError)r:rr=r=r>rds zSSLSSLSocket.recvrrcCs||jj|SN)rrr)r:rr=r=r>sendjszSSLSSLSocket.sendc Csd}t|=}|d}t|}||kr$||||d7}||ksWdn1s.wYWddSWddS1sFwYdS)NrB) memoryviewrlenr)r:rcountview byte_viewamountr=r=r>rmsPzSSLSSLSocket.sendallnamerjcCs t|j|Sr)getattrr)r:rr=r=r> __getattr__xs zSSLSSLSocket.__getattr__r)r[r0rtrr]rBr4r5)rrrrr4r)rrr4r)rrr4r)rrr4r5)rrjr4r) rUrVrWrXrr9rrrrrr=r=r=r>r,s     r).rirjr[rkr\rlr]rBr^rmr_rnr`rorarBr?r@rArBrbrprcrlrdrBrer2r+r2r,r2r-r2rfr3r.r3rgrqrhrrrsrr4r&)NN)rvrBrirBrsrr4r&) rbrrrrr#rsrr4r0)r[r0rr#r4r)rbrrrrr#rArBr\rlr]rBrsrr4r0)H __future__rr/r\rr6rcollections.abcrtypingrrrrrclientr datastructuresr exceptionsr r rextensions.baserextensions.permessage_deflaterrGrhttp11rrr1rrrbrrrrstreamsrrrrrrirr rr"utilsr#__all__r&r$r% python_socksrpython_socks.syncrrrSOCKS5SOCKS4rrrrrrrr=r=r=r>s          i       '&