U g@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#6/tmp/pip-unpacked-wheel-u83us8q9/wsproto/connection.pyrsrc@seZdZdZdZdZdS)ConnectionTypez#An enumeration of connection types.rrN)rrrrCLIENTSERVERr#r#r#r$r%-sr%c@s|eZdZdZdeeeeeddddZ e e ddd Z e ed d d Zeedd ddZee ddfdddZdS) 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`` N)connection_type extensions trailing_datareturncCs<|tjk|_t|_t|j|p g|_tj|_ | |dSN) r%r&clientr_eventsr_protorr_state receive_data)selfr*r+r,r#r#r$__init__Is  zConnection.__init__)r-cCs|jSr.)r2)r4r#r#r$stateUszConnection.state)eventr-cCsd}t|tr2|jtjkr2||j|j|j7}nt|t r\|jtjkr\||j |j 7}nt|t r|jtjkr||j |j 7}njt|tr|jtjtjhkr||j|j|j7}|jtjkrtj|_qtj|_ntd|d|jd|S)Nr)zEvent z cannot be sent in state .) isinstancer r6rrr1Z send_datadatamessage_finishedr ZpingpayloadrZpongr r closecodereasonr"r2r!r)r4r7r:r#r#r$sendYs&   zConnection.send)r:r-cCsb|dkr(|jttjdtj|_dS|jtj tj fkrH|j |n|jtjkr^t dndS)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|jVqz\|jD]J}|jtjkrh|jr@|jsDt t |j t t fsXt t|j dVq"|jtjkr|jr|jst t |j t t fst t|j dVq"|jtjkrt |j tst |j \}}|jtjkrtj|_ntj|_t||dVq"|jtjkr4t |j tst t|j |j|jdVq"|jtjkr"t |j t t fsVt t|j |j|jdVq"q"Wn8tk r}zt|j t|dVW5d}~XYnXdS)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_framesopcoderZPINGrBr;AssertionErrorr9r<bytes bytearrayr ZPONGrZCLOSEtupler6rr!r"r2r r ZTEXTstrrZBINARYr rr>)r4framer>r?excr#r#r$eventssH         zConnection.events)Nr))rrrrr%rrrrFr5propertyrr6r r@r3rrLr#r#r#r$r(;s  r(N) r collectionsrenumrtypingrrrrrLr r r r r rrr+rZframe_protocolrrrrZ utilitiesrrr%r&r'r(r#r#r#r$s  $