U
|
[ ±d ã @ sÆ d Z ddlmZ ddlmZ ddlmZmZmZm Z ddl
|
m Z m Z m Z mZmZmZmZ ddlmZ ddlmZmZmZmZ dd lmZ G d
|
d d eZG d d d eZ e j Z e j Z G dd dZ dS )zU
|
wsproto/connection
|
~~~~~~~~~~~~~~~~~~
|
|
An implementation of a WebSocket connection.
|
é )Údeque)ÚEnum)ÚDequeÚ GeneratorÚListÚOptionalé )Ú BytesMessageÚCloseConnectionÚEventÚMessageÚPingÚPongÚ TextMessage)Ú Extension)Ú CloseReasonÚ FrameProtocolÚOpcodeÚ ParseFailed)ÚLocalProtocolErrorc @ s( e Zd ZdZdZdZdZdZdZdZ dS ) ÚConnectionStatez1
|
RFC 6455, Section 4 - Opening Handshake
|
r r é é é é N)
|
Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Z
|
CONNECTINGÚOPENÚREMOTE_CLOSINGÚ LOCAL_CLOSINGÚCLOSEDZ REJECTING© r# r# úId:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\wsproto/connection.pyr s r c @ s e Zd ZdZdZdZdS )ÚConnectionTypez#An enumeration of connection types.r r N)r r r r ÚCLIENTÚSERVERr# r# r# r$ r% - s r% c @ s| e Zd ZdZdeeee eddddZ e
|
e ddd Z e ed
|
d d Zee dd ddZee ddf dddZdS )Ú
|
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_dataÚreturnc C s< |t jk| _t | _t| j|p g | _tj| _ |
|
|¡ d S ©N) r% r& Úclientr Ú_eventsr Ú_protor r Ú_stateÚ receive_data)Úselfr* r+ r, r# r# r$ Ú__init__I s
|
zConnection.__init__)r- c C s | j S r. )r2 )r4 r# r# r$ ÚstateU s zConnection.state)Úeventr- c C sô d}t |tr2| jtjkr2|| j |j|j¡7 }n¾t |t r\| jtjkr\|| j
|
|j ¡7 }nt |t r| jtjkr|| j |j ¡7 }njt |trØ| jtjtjhkrØ|| j |j|j¡7 }| jtjkrÎtj| _qðtj| _ntd| d| j d|S )Nr) zEvent z cannot be sent in state Ú.)Ú
|
isinstancer r6 r r r1 Z send_dataÚdataÚmessage_finishedr ZpingÚpayloadr Zpongr
|
r ÚcloseÚcodeÚreasonr" r2 r! r )r4 r7 r: r# r# r$ ÚsendY s& þ
|
|
ÿzConnection.send)r: r- c C sb |dkr(| j ttjd¡ tj| _dS | jtj tj
|
fkrH| j |¡ n| jtjkr^t dn 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.)r0 Úappendr
|
r ZABNORMAL_CLOSUREr r" r2 r6 r r! r1 Z receive_bytesr )r4 r: r# r# r$ r3 p s
|
zConnection.receive_datac
|
c s° | j r| j ¡ V q z\| j ¡ D ]J}|jtjkrh|jr@|jsDt t
|
|j t t fsXt t|j dV q"|jtjkr¨|jr|jst t
|
|j t t fst t|j dV q"|jtjkrüt
|
|j tsÄt |j \}}| jtjkrätj| _ntj| _t||dV q"|jtjkr4t
|
|j ts t t |j |j|jdV q"|jtj kr"t
|
|j t t fsVt t |j |j|jdV q"q"W n8 t k
|
rª } zt|j t|dV W 5 d}~X Y nX dS )zÅ
|
Return a generator that provides any events that have been generated
|
by protocol activity.
|
|
:returns: generator of :class:`Event <wsproto.events.Event>` subclasses
|
)r< )r> r? )r: Úframe_finishedr; N)!r0 Úpopleftr1 Zreceived_framesÚopcoder ZPINGrB r; ÚAssertionErrorr9 r< ÚbytesÚ bytearrayr ZPONGr ZCLOSEÚtupler6 r r! r" r2 r r
|
ZTEXTÚstrr ÚBINARYr r r> )r4 Úframer> r? Úexcr# r# r$ Úevents sH
|
|
ý
|
ý
|
zConnection.events)Nr) )r r r r r% r r r rF r5 Úpropertyr r6 r r@ r3 r rM r# r# r# r$ r( ; s ü
|
û r( N) r Ú collectionsr Úenumr Útypingr r r r rM r r
|
r r r r r r+ r Zframe_protocolr r r r Z utilitiesr r r% r& r' r( r# r# r# r$ Ú<module> s $
|
|
