o US`c@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__Z CONNECTINGOPENREMOTE_CLOSING LOCAL_CLOSINGCLOSEDZ REJECTINGr#r#=/usr/local/lib/python3.10/dist-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|jSr.)r2)r4r#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 r6rrr1Z send_datadatamessage_finishedr Zpingpayloadrpongr r closecodereasonr"r2r!r)r4r7r: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)r?zConnection already closed.)r0appendr rZABNORMAL_CLOSURErr"r2r6rr!r1Z receive_bytesr)r4r:r#r#r$r3ps  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<)r?r@)r:frame_finishedr;N) r0popleftr1Zreceived_framesopcoderZPINGrCr;r9r<bytes bytearrayr ZPONGrZCLOSEtupler6rr!r"r2r r ZTEXTstrrBINARYr rr?)r4framer?r@excr#r#r$eventssR          &"zConnection.events)Nr))rrrrr%rrrrFr5propertyrr6r rAr3rrMr#r#r#r$r(;s$   r(N) r collectionsrenumrtypingrrrrrMr r r r r rrr+rZframe_protocolrrrrZ utilitiesrrr%r&r'r(r#r#r#r$s  $