o Rh@sdZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z mZmZmZmZddlmZddlmZmZmZmZdd lmZGd d d eZGd d d eZejZejZGdddZdS)zU wsproto/connection ~~~~~~~~~~~~~~~~~~ An implementation of a WebSocket connection. )deque)Enum)Deque GeneratorListOptional) BytesMessageCloseConnectionEventMessagePingPong TextMessage) Extension) CloseReason FrameProtocolOpcode ParseFailed)LocalProtocolErrorc@s(eZdZdZdZdZdZdZdZdZ dS) ConnectionStatez1 RFC 6455, Section 4 - Opening Handshake rrN) __name__ __module__ __qualname____doc__ CONNECTINGOPENREMOTE_CLOSING LOCAL_CLOSINGCLOSED REJECTINGr%r%T/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/wsproto/connection.pyrsrc@seZdZdZdZdZdS)ConnectionTypez#An enumeration of connection types.rrN)rrrrCLIENTSERVERr%r%r%r&r'-sr'c @seZdZdZ  ddedeeededdfdd Z e de fd d Z d e defd dZdeeddfddZdee ddffddZdS) Connectiona A low-level WebSocket connection object. This wraps two other protocol objects, an HTTP/1.1 protocol object used to do the initial HTTP upgrade handshake and a WebSocket frame protocol object used to exchange messages and other control frames. :param conn_type: Whether this object is on the client- or server-side of a connection. To initialise as a client pass ``CLIENT`` otherwise pass ``SERVER``. :type conn_type: ``ConnectionType`` Nconnection_type extensions trailing_datareturncCs<|tju|_t|_t|j|pg|_tj|_ | |dSN) r'r(clientr_eventsr_protorr _state receive_data)selfr,r-r.r%r%r&__init__Is zConnection.__init__cCs|jSr0)r4)r6r%r%r&stateUszConnection.stateeventcCsd}t|tr|jtjkr||j|j|j7}|St|t r0|jtjkr0||j |j 7}|St|t rF|jtjkrF||j |j 7}|St|trq|jtjtjhvrq||j|j|j7}|jtjkrktj|_|Stj|_|Std|d|jd)Nr+zEvent z cannot be sent in state .) isinstancer r8rr r3 send_datadatamessage_finishedr pingpayloadrpongr r!closecodereasonr#r4r"r)r6r9r=r%r%r&sendYs. zConnection.sendr=cCsd|dur|jttjdtj|_dS|jtj tj fvr%|j |dS|jtjur/t d dS)aO Pass some received data to the connection for handling. A list of events that the remote peer triggered by sending this data can be retrieved with :meth:`~wsproto.connection.Connection.events`. :param data: The data received from the remote peer on the network. :type data: ``bytes`` N)rCzConnection already closed.)r2appendr rABNORMAL_CLOSURErr#r4r8r r"r3 receive_bytesr)r6r=r%r%r&r5ps  zConnection.receive_datac cs|jr |jV|jsz|jD]}|jtjur5|jr!|js#Jt |j t t fs-Jt |j dVq|jtjurU|jrA|jsCJt |j t t fsMJt|j dVq|jtjurt |j tscJ|j \}}|jtjurstj|_ntj|_t||dVq|jtjurt |j tsJt|j |j|jdVq|jtjurt |j t t fsJt|j |j|jdVq qWdSty}zt|jt|dVWYd}~dSd}~ww)z Return a generator that provides any events that have been generated by protocol activity. :returns: generator of :class:`Event ` subclasses )r@)rCrD)r=frame_finishedr>N) r2popleftr3received_framesopcoderPINGrIr>r;r@bytes bytearrayr PONGrCLOSEtupler8rr"r#r4r!r TEXTstrrBINARYr rrC)r6framerCrDexcr%r%r&eventssR          &"zConnection.events)Nr+)rrrrr'rrrrNr7propertyrr8r rEr5rrXr%r%r%r&r*;s$   r*N) r collectionsrenumrtypingrrrrrXr r r r r rrr-rframe_protocolrrrr utilitiesrrr'r(r)r*r%r%r%r&s  $