a یxd %@szddlmZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z ddlm Z dgZedZGd ddZdS) ) annotationsN)IteratorListOptionalcast)FrameOpcode)Data Assemblerzutf-8c@s\eZdZdZddddZdddd d d Zd dd dZdddddZddddZdS)r z) Assemble messages from frames. None)returncCsFt|_t|_t|_d|_d|_d|_g|_ d|_ d|_ dS)NF) threadingLockmutexEventmessage_completemessage_fetchedget_in_progressput_in_progressdecoderchunks chunks_queueclosedselfrP/var/www/html/Ranjet/env/lib/python3.9/site-packages/websockets/sync/messages.py__init__s    zAssembler.__init__NzOptional[float]r )timeoutr cCs|j2|jrtd|jr$tdd|_Wdn1s>0Y|j|}|jd|_|sxtd|dd|jrtd|jsJ|j |j durd nd }| |j }|j rJ|j g|_ |jdusJ|WdS1s0YdS) a Read the next message. :meth:`get` returns a single :class:`str` or :class:`bytes`. If the message is fragmented, :meth:`get` waits until the last frame is received, then it reassembles the message and returns it. To receive messages frame by frame, use :meth:`get_iter` instead. Args: timeout: If a timeout is provided and elapses before a complete message is received, :meth:`get` raises :exc:`TimeoutError`. Raises: EOFError: If the stream of frames has ended. RuntimeError: If two threads run :meth:`get` or :meth:``get_iter` concurrently. stream of frames ended"get or get_iter is already runningTNFz timed out in z.1fs)rrEOFErrorr RuntimeErrorrwait TimeoutErroris_setclearrjoinrrsetr)rr completedjoinermessagerrrget;s,$    z Assembler.getzIterator[Data]ccs$|jd|jrtd|jr$td|j}g|_tdt|_ |j rV|j dd|_Wdn1sp0Y|EdH|j }|durq|Vq|jhd|_|j sJ|j |jrtd|j rJ|j|jgksJd|_ Wdn1s0YdS)aw 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. The iterator must be fully consumed before calling :meth:`get_iter` or :meth:`get` again. Else, :exc:`RuntimeError` is raised. This method only makes sense for fragmented messages. If messages aren't fragmented, use :meth:`get` instead. Raises: EOFError: If the stream of frames has ended. RuntimeError: If two threads run :meth:`get` or :meth:``get_iter` concurrently. r r!z!queue.SimpleQueue[Optional[Data]]NTF)rrr%rr&rrqueue SimpleQueuerrr)putr0r*rr,)rrchunkrrrget_iterws:  $    zAssembler.get_iterr)framer cCs|j|jrtd|jr&td|jtjur@tdd|_ n2|jtj urTd|_ n|jtj urbnWddS|j dur|j |j |j}n|j }|jdur|j|n |j||jsWddS|jrJ|j|jdur|jd|jrJd|_Wdn1s*0Y|j|jFd|_|js\J|j|jrvtdd|_ Wdn1s0YdS)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 fully consuming 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. Raises: EOFError: If the stream of frames has ended. RuntimeError: If two threads run :meth:`put` concurrently. r zput is already runningstrict)errorsNTF)rrr%rr&opcoder ZTEXT UTF8DecoderrBINARYZCONTdecodedataZfinrrappendr3rr)r,rr'r*)rr6r=rrrr3sD          &  z Assembler.putcCs||jb|jrWddSd|_|jrJ|j|jdurJ|jd|jrZ|jWdn1sn0YdS)z End the stream of frames. Callling :meth:`close` concurrently with :meth:`get`, :meth:`get_iter`, or :meth:`put` is safe. They will raise :exc:`EOFError`. NT) rrrrr,rr3rrrrrrcloses   zAssembler.close)N) __name__ __module__ __qualname____doc__rr0r5r3r?rrrrr s $<BJ) __future__rcodecsr1rtypingrrrrframesrr r __all__getincrementaldecoderr:r rrrrs