o ]i#@sdZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZmZddlmZmZmZmZdd lmZerLdd lmZdd lmZGd d d eZGdddeZejZejZGdddZ dS)zU wsproto/connection ~~~~~~~~~~~~~~~~~~ An implementation of a WebSocket connection. ) annotations)deque)Enum) TYPE_CHECKING) BytesMessageCloseConnectionEventMessagePingPong TextMessage) CloseReason FrameProtocolOpcode ParseFailed)LocalProtocolError) Generator) Extensionc@s(eZdZdZdZdZdZdZdZdZ dS) ConnectionStatez1 RFC 6455, Section 4 - Opening Handshake rrN) __name__ __module__ __qualname____doc__ CONNECTINGOPENREMOTE_CLOSING LOCAL_CLOSINGCLOSED REJECTINGr$r$W/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/wsproto/connection.pyrsrc@seZdZdZdZdZdS)ConnectionTypez#An enumeration of connection types.rrN)rrrrCLIENTSERVERr$r$r$r%r&2sr&c@sLeZdZdZ  ddd d ZedddZdddZd ddZd!ddZ dS)" 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. Nconnection_typer& extensionslist[Extension] | None trailing_databytesreturnNonecCs<|tju|_t|_t|j|pg|_tj|_ | |dS)a Constructor :param wsproto.connection.ConnectionType connection_type: Whether this object is on the client- or server-side of a connection. To initialise as a client pass ``CLIENT`` otherwise pass ``SERVER``. :param list extensions: The proposed extensions. :param bytes trailing_data: Data that has been received, but not yet processed. N) r&r'clientr_eventsr_protorr_state receive_data)selfr+r,r.r$r$r%__init__Is zConnection.__init__rcCs|jS)N)r5)r7r$r$r%state_szConnection.stateeventr cCsd}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|_|Sd|d|jd}t|)Nr*zEvent z cannot be sent in state .) isinstancer r9rrr4 send_datadatamessage_finishedr pingpayloadr pongrr closecodereasonr"r5r!r)r7r:r>msgr$r$r%sendcs0 zConnection.sendr> bytes | NonecCsh|dur|jttjdtj|_dS|jtj tj fvr%|j |dS|jtjur1d}t | 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)rDzConnection already closed.)r3appendrrABNORMAL_CLOSURErr"r5r9rr!r4 receive_bytesr)r7r>rFr$r$r%r6{s  zConnection.receive_dataGenerator[Event, None, None]c cs|jr |jV|jsz|jD]}|jtjur7|js J|js%Jt |j t t fs/Jt |j dVq|jtjurY|jsBJ|jsGJt |j t t fsQJt|j dVq|jtjurt |j tsgJ|j \}}|jtjurwtj|_ntj|_t||dVq|jtjurt |j tsJt|j |j|jdVq|jtjurt |j t t fsJtt |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 )rA)rDrE)r>frame_finishedr?N) r3popleftr4received_framesopcoderPINGrMr?r<rAr/ bytearrayr PONGr CLOSEtupler9rr!r"r5r rTEXTstrr BINARYrrrD)r7framerDrEexcr$r$r%eventssV              ("zConnection.events)Nr*)r+r&r,r-r.r/r0r1)r0r)r:r r0r/)r>rHr0r1)r0rL) rrrrr8propertyr9rGr6r[r$r$r$r%r)@s     r)N)!r __future__r collectionsrenumrtypingrr[rrr r r r r frame_protocolrrrr utilitiesrcollections.abcrr,rrr&r'r(r)r$r$r$r%s     $