RPf]ddlmZddlZddlZddlZddlmZmZmZm 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 ejRZ*e*jVZ+e*jXZ,GddejRZ-e-j\Z.e-j^Z/e-j`Z0e-jbZ1dZ2 GddZ3y)) annotationsN) GeneratorListOptionalTypeUnion)ConnectionClosedConnectionClosedErrorConnectionClosedOK InvalidState PayloadTooBig ProtocolError) Extension) OK_CLOSE_CODES OP_BINARYOP_CLOSEOP_CONTOP_PINGOP_PONGOP_TEXTClose CloseCodeFrame)RequestResponse) StreamReader) LoggerLikeOrigin Subprotocol)ProtocolSideStateSEND_EOFc&eZdZdZed\ZZy)r"z6A WebSocket connection is either a server or a client.N)__name__ __module__ __qualname____doc__rangeSERVERCLIENTM/var/www/fastapitest/venv/lib/python3.12/site-packages/websockets/protocol.pyr"r"-s@1XNFFr/r"c*eZdZdZed\ZZZZy)r#z6A WebSocket connection is in one of these four states.N) r'r(r)r*r+ CONNECTINGOPENCLOSINGCLOSEDr.r/r0r#r#7s@(-a%Jgvr/r#r/c2eZdZdZeddd ddZeddZejddZeddZ ed d Z ed!d Z d"d Z d#d Z d$d Zd%d$dZd%d$dZd&d'dZd"dZd"dZd(d)dZd*dZd+dZd,dZd-dZd-dZd.dZd.dZd#dZy)/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_sizeloggercHtj|_ |0tjd|j j }||_ |jtj|_ ||_ ||_ ||_ d|_d|_d|_g|_d|_d|_d|_d|_d|_ d|_t1|_g|_g|_|j9|_t=|j:d|_y)Nz websockets.F) uuiduuid4idlogging getLoggernamelowerr: isEnabledForDEBUGdebugsider8r9cur_sizeexpect_continuation_frameorigin extensions subprotocol close_rcvd close_sentclose_rcvd_then_sent handshake_exceof_sentrreadereventswritesparseparsernext parser_exc)selfrFr8r9r:s r0__init__zProtocol.__init__Ws"ZZ\B >&&TYY__5F4G'HIF"( )((7   ! (, */&)- +-26,0+/48!37  #n #% #% jjl  T[[/3r/c|jS)zm WebSocket connection state. Defined in 4.1, 4.2, 7.1.3, and 7.1.4 of :rfc:`6455`. )_staterXs r0r8zProtocol.states{{r/cv|jr&|jjd|j||_y)Nz= connection is %s)rEr:rAr[)rXr8s r0r8zProtocol.states) :: KK  2EJJ ? r/c|jtury|jtjS|jj S)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)r8r6rLrABNORMAL_CLOSUREcoder\s r0 close_codezProtocol.close_codes9 ::V # __ $-- -??'' 'r/cn|jtury|jy|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)r8r6rLreasonr\s r0 close_reasonzProtocol.close_reasons0 ::V # __ $??)) )r/cd|jtusJd|jK|j?|jjt vr#|jjt vrt }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 yet) r8r6rLrMr`rr r rNrW __cause__)rXexc_typeexcs r0 close_exczProtocol.close_excs zzV#B%BB# OO '+$$6$$6)H,H ( OO OO  % %!    r/cd|jj|t|jy)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)rQ feed_datarVrUrXdatas r0 receive_datazProtocol.receive_datas" d# T[[r/cb|jjt|jy)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)rQfeed_eofrVrUr\s r0 receive_eofzProtocol.receive_eofs   T[[r/c|js td| |_|jtt||y)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)rHr send_framerrrXrnfins r0send_continuationzProtocol.send_continuations8"-- ?@ @-0& gtS12r/c|jr td| |_|jtt||y)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)rHrrurrrvs r0 send_textzProtocol.send_text/s8  ) ) ?@ @-0& gtS12r/c|jr td| |_|jtt||y)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. rzN)rHrrurrrvs r0 send_binaryzProtocol.send_binaryDs8  ) ) ?@ @-0& is34r/c*|jr td|-|dk7r tdttjd}d}nt||}|j }|j tt|||_ t|_ y)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 rzNrcz#cannot send a reason without a coder/) rHrrrNO_STATUS_RCVD serializerurrrMr5r8rXr`rdcloserns r0 send_closezProtocol.send_closeYs  ) ) ?@ @ <|#$IJJ)22B7ED$'E??$D h-. r/cB|jtt|y)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)rurrrms r0 send_pingzProtocol.send_pingy gt,-r/cB|jtt|y)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)rurrrms r0 send_pongzProtocol.send_pongrr/c|jtur`|tjk7rMt ||}|j }|j tt|||_ t|_|jtur|js|j|j|_t#|j y)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)r8r4rr_rrrurrrMr5rFr,rPsend_eofdiscardrUrVrs r0failz Protocol.fails& :: y111dF+(h 56"'$ 99 t}} MMO lln  T[[r/c.|jgc}|_|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. )rR)rXrRs r0events_receivedzProtocol.events_receiveds#kk2  r/c.|jgc}|_|S)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. )rS)rXrSs r0 data_to_sendzProtocol.data_to_sends"#kk2  r/cF|jtuxs|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)r8r5rOr\s r0close_expectedzProtocol.close_expecteds#,zzW$F(:(:$(FFr/c#PK |jjEd{r2|jr|jjdt d|j d}n2|j |j }n|j |j z }tj|jj|jtu||jEd{}|jr|jjd||j|77B#t$r;}|jt j"t%|||_Yd}~n/d}~wt$r:}|jt j(t%|||_Yd}~nd}~wt*$rJ}|jt j,|j.d|j0||_Yd}~nd}~wt2$r:}|jt j4t%|||_Yd}~n\d}~wt6$rM}|jj9dd |jt j:||_Yd}~nd}~wwxYwdt=d w) 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)maskr9rJz< %sz at position z parser failed)exc_infoz"parse() shouldn't step after error)rQat_eofrEr:EOFErrorr9rGrrT read_exactrFr,rJ recv_framerrrPROTOCOL_ERRORstrrWr_UnicodeDecodeError INVALID_DATArdstartrMESSAGE_TOO_BIG ExceptionerrorINTERNAL_ERRORAssertionError)rXr9frameris r0rTzProtocol.parses5 "#{{11333zz ))'2 ##=>>==(#H]]*#}}H#}}t}}C*A$C*(C*c|jtus|jturH|j t d|j r d|_nt |j|_n|jturV|j t d|j r d|_n|xjt |jz c_n|jtur-tt|j}|j|nM|jturn9|jtur tj|j|_|j"t$ur|j&Jd|_|j t d|j"t*urL|jtt|j|j |_d|_t$|_|j,t.ur|j1|j3|_t7|j4nt9d|jd|j:j=|y) z- Process an incoming frame. NrzrtFzincomplete fragmented messageTzunexpected opcode: 02x)opcoderrrGrrwlenrnrrrrrurrrTrLr8r5rMrNr4rFr,rrrUrVrrRappend)rXr pong_frames r0rzProtocol.recv_frameas <<7 "elli&?}}(#$CDDyy $ #EJJ \\W $}}$#$CDDyy $  UZZ0 \\W $w 3J OOJ ' \\W $  \\X %$kk%**5DOzzW$222,1)}}(#$CDDzzT! h ;<"&//,0)$ yyF"  ,,.DK  !#6u||C6H!IJ J 5!r/cJ|jtur#td|jjd|jr|j j d||j j|j|jtu|jy)Nz#cannot write to a WebSocket in the z statez> %s)rrJ) r8r4r rArEr:rSrrrFr-rJ)rXrs r0ruzProtocol.send_frames| ::T !5djjoo5FfM  :: KK  fe ,  OOf!4O Q r/c|jrJd|_|jr|jjd|jj t y)NTz> EOF)rPrEr:rSrr$r\s r0rzProtocol.send_eofsA==   :: KK  g & 8$r/) rFr"r8r#r9 Optional[int]r:zOptional[LoggerLike]returnNone)rr#)r8r#rr)rr)rz Optional[str])rr )rnbytesrr)rr)rnrrwboolrr)T)Nrc)r`rrdrrr)rc)r`intrdrrr)rz List[Event])rz List[bytes])rr)rzGenerator[None, None, None])rrrr)r'r(r)r*r4rYpropertyr8setterrarerjrorrrxr{r}rrrrrrrrTrrrurr.r/r0r!r!Gs &"''+ B4B4 B4  B4 % B4 B4H \\ (("**"!!J (3,3*5*@ . .'V  (G4ECNCrs"  99"   &!33  gx&'?4<<  1DLL1     zz --  E} %} %r/