o NK&h,@sddlZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z er2ddlmZed ZGd d d ZdS) N) AsyncIterator) TYPE_CHECKINGOptional)FrameOpcode)Data) ServerError)WebsocketImplProtocolzutf-8c@seZdZUdZdZerHded<ejed<ejed<ej ed<ej ed<ej ed <e ed <e e jed <eeed <e ej e eed <e ed<dddZdde ede efddZdeefddZdeddfddZdS)WebsocketFrameAssemblerz Assemble a message from frames. Code borrowed from aaugustin/websockets project: https://github.com/aaugustin/websockets/blob/6eb98dd8fa5b2c896b9f6be7e8d117708da82a39/src/websockets/sync/messages.py ) protocol read_mutex write_mutexmessage_completemessage_fetchedget_in_progressdecodercompleted_queuechunks chunks_queuepausedget_idput_idr r r rrrrrrrrrreturnNcCs^||_t|_t|_tjdd|_t|_t|_ d|_ d|_ g|_ d|_ d|_dS)Nr )maxsizeF)r asyncioLockr rQueuerEventrrrrrrr)selfr r P/var/www/html/venv/lib/python3.10/site-packages/sanic/server/websockets/frame.py__init__7s     z WebsocketFrameAssembler.__init__timeoutc s|j4IdH|dur"|dkr"|js" WdIdHdS|jr)tdd|_|dur9|jIdH}n2|dkrC|j}n(z!ztj|j|dIdHWn tjy]YnwW|j}n|j}w|j rv|j d|_ |js}tdd|_|s WdIdHdS|js WdIdHdS|j |j durdnd }||j}|jrtd |jg|_d|_|WdIdHS1IdHswYdS) ai Read the next message. :meth:`get` returns a single :class:`str` or :class:`bytes`. If the :message was fragmented, :meth:`get` waits until the last frame is received, then it reassembles the message. If ``timeout`` is set and elapses before a complete message is received, :meth:`get` returns ``None``. NrzXCalled get() on Websocket frame assembler while asynchronous get is already in progress.T)r#FZState of Websocket frame assembler was modified while an asynchronous get was in progress.z?Websocket get() found a message when state was already fetched.)r ris_setrrwaitrwait_for TimeoutErrorrr resume_framesclearrjoinrrsetr)rr# completedjoinermessager r r!getYsd      . 1   0zWebsocketFrameAssembler.getc Cs*|j4IdH|jrtdd|_|j}g|_t|_|jr,|j dIdH|D]}|Vq. |j IdH}|durBn|Vq5|j rQ|j d|_ |jsXtdd|_|jsdtd|j|jrrtd|jg|_d|_WdIdHdS1IdHswYdS)z Stream the next message. Iterating the return value of :meth:`get_iter` yields a :class:`str` or :class:`bytes` for each frame in the message. Nz[Called get_iter on Websocket frame assembler while asynchronous get is already in progress.TFr$zIWebsocket frame assembler chunks queue ended before message was complete.zDWebsocket get_iter() found a message when state was already fetched.)r rrrrrrrr'putr2rr r+r,rr.)rrcchunkr r r!get_itersR       .z WebsocketFrameAssembler.get_iterframec s~|j4IdH|jtjurtdd|_n|jtjur d|_n|jtjur'n WdIdHdS|jdurB|j|j |j }n|j }|j durQ|j |n |j |IdH|j si WdIdHdS|jsr|j|_|j dur|j dIdH|jrtd|j|jrtd|jIdH|jd|_WdIdHdS1IdHswYdS)a Add ``frame`` to the next message. When ``frame`` is the final frame in a message, :meth:`put` waits until the message is fetched, either by calling :meth:`get` or by iterating the return value of :meth:`get_iter`. :meth:`put` assumes that the stream of frames respects the protocol. If it doesn't, the behavior is undefined. Nstrict)errorszLWebsocket put() got a new message when a message was already in its chamber.zPWebsocket put() got a new message when the previous message was not yet fetched.)ropcoderTEXT UTF8DecoderrBINARYCONTdecodedatafinrrappendr3rr pause_framesrrr'rr.rr(r,)rr7r@r r r!r3sH           .zWebsocketFrameAssembler.put)rN)N)__name__ __module__ __qualname____doc__ __slots__r__annotations__rrrrboolrcodecsIncrementalDecoderlistrr"floatr2rr6rr3r r r r!r s&        "NEr )rrKcollections.abcrtypingrrwebsockets.framesrrwebsockets.typingrsanic.exceptionsrimplr getincrementaldecoderr<r r r r r!s