o ?Hhl,@s&ddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZmZmZdd lmZddlm Z ddl!m"Z"gdZ#Gddde Z$dddddddedddddddd;d6d7Z%  d) annotationsN)AnySequence)ClientProtocol) HeadersLike)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENTResponse) CONNECTINGEvent) LoggerLikeOrigin Subprotocol) parse_uri) Connection)Deadline)connect unix_connectClientConnectioncsVeZdZdZdddfd d ZdedfdddZdfdd Zdfdd 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_timeoutsocket socket.socketprotocolrr float | NonereturnNonecs$|t|_tj|||ddS)Nr) threadingr response_rcvdsuper__init__)selfrrr __class__V/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/websockets/sync/client.pyr%1s  zClientConnection.__init__Nadditional_headersHeadersLike | Noneuser_agent_header str | NonetimeoutcCs|jtd*|j|_|dur|jj||dur"||jjd<|j|jWdn1s3wY|j |sBt d|jj durL|jj dS)z1 Perform the opening handshake. )expected_stateNz User-Agentztimed out during handshake) send_contextr rrrequestheadersupdate send_requestr#wait TimeoutError handshake_exc)r&r+r-r/r)r)r* handshake@s    zClientConnection.handshakeeventrcs<|jdurt|ts J||_|jdSt|dS)z. Process one incoming event. N)response isinstancer r#setr$ process_event)r&r:r'r)r*r>\s zClientConnection.process_eventcs(z tW|jdS|jw)zI Read incoming data from the socket and process events. N)r$ recv_eventsr#r=)r&r'r)r*r?js zClientConnection.recv_events)rrrrrrr r!)r+r,r-r.r/rr r!)r:rr r!)r r!) __name__ __module__ __qualname____doc__r%r r9r>r? __classcell__r)r)r'r*rs rdeflateri)socksslserver_hostnameorigin extensions subprotocolsr+r- compression open_timeoutrmax_sizeloggercreate_connectionuristrrFsocket.socket | NonerGssl_module.SSLContext | NonerHr.rI Origin | NonerJ'Sequence[ClientExtensionFactory] | NonerKSequence[Subprotocol] | Noner+r,r-rLrMrrrN int | NonerOLoggerLike | NonerPtype[ClientConnection] | Nonekwargsrr cKs8|durd|vr|d}tdtt|}|js"|dur"td|dd}|dd}|rH|dur<|dur` 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 incompatiblerEzunsupported compression: r/T)rH)rIrJrKrNrOr)#popwarningswarnDeprecationWarningrsecure TypeErrorr r ValueErrorrrrAF_UNIX SOCK_STREAM settimeoutr/r setdefaultrPhostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socketr Exceptioncloser9 close_socketrecv_events_threadjoin)rQrFrGrHrIrJrKr+r-rLrMrrNrOrPr[wsurir]r^deadliner connectionr)r)r*rvsW            rr^cKsD|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/``. NrGr\zws://localhost/zwss://localhost/T)rQr]r^r))getr)r^rQr[r)r)r*r5s r)"rQrRrFrSrGrTrHr.rIrUrJrVrKrWr+r,r-r.rLr.rMrrrrNrXrOrYrPrZr[rr r)NN)r^r.rQr.r[rr r)' __future__rrrGror"r`typingrrclientrdatastructuresrextensions.baserextensions.permessage_deflater r3r http11r r rr rrrrrQrryrutilsr__all__rrrr)r)r)r*sL         aA