o /h]@sDddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z mZmZmZmZddlmZddlmZmZmZmZmZmZmZmZmZmZddlm Z m!Z!dd l"m#Z#dd lm$Z$m%Z%m&Z&gd Z'e e e!efZ( Gd d d ej)Z*e*j+Z+e*j,Z,Gdddej)Z-e-j.Z.e-j/Z/e-j0Z0e-j1Z1dZ2 GdddZ3dS)) annotationsN) GeneratorListOptionalTypeUnion)ConnectionClosedConnectionClosedErrorConnectionClosedOK InvalidState PayloadTooBig ProtocolError) Extension) OK_CLOSE_CODES OP_BINARYOP_CLOSEOP_CONTOP_PINGOP_PONGOP_TEXTClose CloseCodeFrame)RequestResponse) StreamReader) LoggerLikeOrigin Subprotocol)ProtocolSideStateSEND_EOFc@seZdZdZed\ZZdS)r!z6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENTr,r,L/var/www/html/govbot/env/lib/python3.10/site-packages/websockets/protocol.pyr!-sr!c@s eZdZdZed\ZZZZdS)r"z6A WebSocket connection is in one of these four states.N) r%r&r'r(r) CONNECTINGOPENCLOSINGCLOSEDr,r,r,r-r"7sr"c@seZdZdZeddddMddZedNddZejdOddZedPddZ edQddZ edRddZ dSddZ dTd d!Z dUd$d%ZdVdUd'd(ZdVdUd)d*ZdWdXd/d0ZdSd1d2ZdSd3d4ZdYdZd6d7Zd[d9d:Zd\dd?Zd^dAdBZd^dCdDZd_dGdHZd_dIdJZdTdKdLZdS)`r a6 Sans-I/O implementation of a WebSocket connection. Args: side: :attr:`~Side.CLIENT` or :attr:`~Side.SERVER`. state: initial state of the WebSocket connection. max_size: maximum size of incoming messages in bytes; :obj:`None` disables the limit. logger: logger for this connection; depending on ``side``, defaults to ``logging.getLogger("websockets.client")`` or ``logging.getLogger("websockets.server")``; see the :doc:`logging guide <../../topics/logging>` for details. iN)statemax_sizeloggersider!r4r"r5 Optional[int]r6Optional[LoggerLike]returnNonecCst|_ |durtd|j}||_ |tj |_ ||_ ||_ ||_ d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t|_g|_g|_||_t|jd|_dS)Nz websockets.F) uuiduuid4idlogging getLoggernamelowerr6 isEnabledForDEBUGdebugr7r4r5cur_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfr7r4r5r6r,r,r-__init__Ws6    zProtocol.__init__cCs|jS)zm WebSocket connection state. Defined in 4.1, 4.2, 7.1.3, and 7.1.4 of :rfc:`6455`. )_staterWr,r,r-r4szProtocol.statecCs |jr |jd|j||_dS)Nz= connection is %s)rEr6rArY)rWr4r,r,r-r4s cCs&|jturdS|jdurtjS|jjS)z `WebSocket close code`_. .. _WebSocket close code: https://www.rfc-editor.org/rfc/rfc6455.html#section-7.1.5 :obj:`None` if the connection isn't closed yet. N)r4r2rKrABNORMAL_CLOSUREcoderZr,r,r- close_codes  zProtocol.close_code Optional[str]cCs$|jturdS|jdurdS|jjS)z `WebSocket close reason`_. .. _WebSocket close reason: https://www.rfc-editor.org/rfc/rfc6455.html#section-7.1.6 :obj:`None` if the connection isn't closed yet. N)r4r2rKreasonrZr,r,r- close_reasons  zProtocol.close_reasonr cCsf|jtus Jd|jdur"|jdur"|jjtvr"|jjtvr"t}nt}||j|j|j}|j |_ |S)a Exception to raise when trying to interact with a closed connection. Don't raise this exception while the connection :attr:`state` is :attr:`~websockets.protocol.State.CLOSING`; wait until it's :attr:`~websockets.protocol.State.CLOSED`. Indeed, the exception includes the close code and reason, which are known only once the connection is closed. Raises: AssertionError: if the connection isn't closed yet. zconnection isn't closed yetN) r4r2rKrLr\rr r rMrV __cause__)rWexc_typeexcr,r,r- close_excs    zProtocol.close_excdatabytescCs|j|t|jdS)aH Receive data from the network. After calling this method: - You must call :meth:`data_to_send` and send this data to the network. - You should call :meth:`events_received` and process resulting events. Raises: EOFError: if :meth:`receive_eof` was called earlier. N)rP feed_datarUrTrWrfr,r,r- receive_datas zProtocol.receive_datacCs|jt|jdS)a Receive the end of the data stream from the network. After calling this method: - You must call :meth:`data_to_send` and send this data to the network; it will return ``[b""]``, signaling the end of the stream, or ``[]``. - You aren't expected to call :meth:`events_received`; it won't return any new events. Raises: EOFError: if :meth:`receive_eof` was called earlier. N)rPfeed_eofrUrTrZr,r,r- receive_eofs zProtocol.receive_eoffinboolcCs,|jstd| |_|tt||dS)a Send a `Continuation frame`_. .. _Continuation frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.6 Parameters: data: payload containing the same kind of data as the initial frame. fin: FIN bit; set it to :obj:`True` if this is the last frame of a fragmented message and to :obj:`False` otherwise. Raises: ProtocolError: if a fragmented message isn't in progress. unexpected continuation frameN)rGr send_framerrrWrfrmr,r,r-send_continuationszProtocol.send_continuationTcC,|jrtd| |_|tt||dS)a Send a `Text frame`_. .. _Text frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.6 Parameters: data: payload containing text encoded with UTF-8. fin: FIN bit; set it to :obj:`False` if this is the first frame of a fragmented message. Raises: ProtocolError: if a fragmented message is in progress. expected a continuation frameN)rGrrprrrqr,r,r- send_text/zProtocol.send_textcCrs)a Send a `Binary frame`_. .. _Binary frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.6 Parameters: data: payload containing arbitrary binary data. fin: FIN bit; set it to :obj:`False` if this is the first frame of a fragmented message. Raises: ProtocolError: if a fragmented message is in progress. rtN)rGrrprrrqr,r,r- send_binaryDrvzProtocol.send_binaryr_r\r`strcCsj|jrtd|dur|dkrtdttjd}d}n t||}|}|tt|||_ t |_ dS)a Send a `Close frame`_. .. _Close frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.1 Parameters: code: close code. reason: close reason. Raises: ProtocolError: if a fragmented message is being sent, if the code isn't valid, or if a reason is provided without a code rtNr_z#cannot send a reason without a coder3) rGrrrNO_STATUS_RCVD serializerprrrLr1r4rWr\r`closerfr,r,r- send_closeYs   zProtocol.send_closecC|tt|dS)z Send a `Ping frame`_. .. _Ping frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.2 Parameters: data: payload containing arbitrary binary data. N)rprrrir,r,r- send_pingy zProtocol.send_pingcCr~)z Send a `Pong frame`_. .. _Pong frame: https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.3 Parameters: data: payload containing arbitrary binary data. N)rprrrir,r,r- send_pongrzProtocol.send_pongintcCsr|jtur!|tjkr!t||}|}|tt|||_ t |_|j t ur-|j s-|||_t|jdS)a? `Fail the WebSocket connection`_. .. _Fail the WebSocket connection: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.7 Parameters: code: close code reason: close reason Raises: ProtocolError: if the code isn't valid. N)r4r0rr[rrzrprrrLr1r7r*rOsend_eofdiscardrTrUr{r,r,r-fails    z Protocol.fail List[Event]cC|jg}|_|S)a7 Fetch events generated from data received from the network. Call this method immediately after any of the ``receive_*()`` methods. Process resulting events, likely by passing them to the application. Returns: List[Event]: Events read from the connection. )rQ)rWrQr,r,r-events_receiveds zProtocol.events_received List[bytes]cCr)a Obtain data to send to the network. Call this method immediately after any of the ``receive_*()``, ``send_*()``, or :meth:`fail` methods. Write resulting data to the connection. The empty bytestring :data:`~websockets.protocol.SEND_EOF` signals the end of the data stream. When you receive it, half-close the TCP connection. Returns: List[bytes]: Data to write to the connection. )rR)rWrRr,r,r- data_to_sendszProtocol.data_to_sendcCs|jtup |jduS)a Tell if the TCP connection is expected to close soon. Call this method immediately after any of the ``receive_*()``, ``send_close()``, or :meth:`fail` methods. If it returns :obj:`True`, schedule closing the TCP connection after a short timeout if the other side hasn't already closed it. Returns: bool: Whether the TCP connection is expected to close soon. N)r4r1rNrZr,r,r-close_expectedszProtocol.close_expectedGenerator[None, None, None]c cszO |jEdHr|jr|jdtd|jdur d}n|jdur)|j}n|j|j}tj|jj |j t u||j dEdH}|jrK|jd|| |qtyn}z|tjt|||_WYd}~nd}~wty}z|tjt|||_WYd}~nfd}~wty}z|tj|jd|j||_WYd}~nDd}~wty}z|tjt|||_WYd}~n'd}~wty}z|jjddd |tj||_WYd}~nd}~wwdVtd ) a- Parse incoming data into frames. :meth:`receive_data` and :meth:`receive_eof` run this generator coroutine until it needs more data or reaches EOF. :meth:`parse` never raises an exception. Instead, it sets the :attr:`parser_exc` and yields control. TN< EOFzunexpected end of stream)maskr5rIz< %sz at position z parser failed)exc_infoz"parse() shouldn't step after error)rPat_eofrEr6EOFErrorr5rFrrS read_exactr7r*rI recv_framerrrPROTOCOL_ERRORrxrVr[UnicodeDecodeError INVALID_DATAr`startr MESSAGE_TOO_BIG ExceptionerrorINTERNAL_ERRORAssertionError)rWr5framerdr,r,r-rSs\         zProtocol.parseccsx|jtu|jks J|jEdHs |j|jEdHr|jr)|jd|jtur2| t |_ dVt d)z Discard incoming data. This coroutine replaces :meth:`parse`: - after receiving a close frame, during a normal closure (1.4); - after sending a close frame, during an abnormal closure (7.1.7). Nrz"discard() shouldn't step after EOF) r7r*rOrPrrrEr6r+rr2r4rrZr,r,r-rCs   zProtocol.discardrrcCsr|jtus |jtur!|jdurtd|jrd|_nt|j|_n|jturA|jdur/td|jr6d|_n{|jt|j7_np|jt urRt t |j}| |n_|jt urXnY|jt urt|j|_|jturs|jduspJd|_|jdur|td|jtur| t t |j|j|_d|_t|_|jtur|||_t|jn td|jd|j|dS) z- Process an incoming frame. NrtroFzincomplete fragmented messageTzunexpected opcode: 02x)opcoderrrFrrmlenrfrrrrrprrrSrKr4r1rLrMr0r7r*rrrTrUrrQappend)rWr pong_framer,r,r-rasF              zProtocol.recv_framecCsT|jturtd|jjd|jr|jd||j|j|j t u|j ddS)Nz#cannot write to a WebSocket in the z statez> %s)rrI) r4r0r rArEr6rRrrzr7r+rI)rWrr,r,r-rps zProtocol.send_framecCs2|jrJd|_|jr|jd|jtdS)NTz> EOF)rOrEr6rRrr#rZr,r,r-rs  zProtocol.send_eof) r7r!r4r"r5r8r6r9r:r;)r:r")r4r"r:r;)r:r8)r:r^)r:r )rfrgr:r;)r:r;)rfrgrmrnr:r;)T)Nr_)r\r8r`rxr:r;)r_)r\rr`rxr:r;)r:r)r:r)r:rn)r:r)rrr:r;)r%r&r'r(r0rXpropertyr4setterr]rarerjrlrrrurwr}rrrrrrrSrrrprr,r,r,r-r GsBD      %      +    G  R r )4 __future__renumr?r<typingrrrrr exceptionsr r r r r rrIrframesrrrrrrrrrrhttp11rrstreamsrrrr__all__EventIntEnumr!r*r+r"r/r0r1r2r#r r,r,r,r-s2   0