RPf %ddlmZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z ddlm Z dgZejdZGd dZy) ) annotationsN)IteratorListOptionalcast)FrameOpcode)Data Assemblerzutf-8c:eZdZdZddZd d dZd dZd dZddZy) r z) Assemble messages from frames. ctj|_tj|_tj|_d|_d|_d|_g|_ d|_ d|_ y)NF) threadingLockmutexEventmessage_completemessage_fetchedget_in_progressput_in_progressdecoderchunks chunks_queueclosedselfs R/var/www/fastapitest/venv/lib/python3.12/site-packages/websockets/sync/messages.py__init__zAssembler.__init__sl^^% !* 1(0 %$=A #% LP Nc|j5|jr td|jr t dd|_ddd|j j |}|j5d|_|std|dd|jr td|j jsJ|j j|jd nd }|j|j}|jjrJ|jjg|_ |jJ|cdddS#1swYxYw#1swYyxYw) 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.1fsr)rrEOFErrorr RuntimeErrorrwait TimeoutErroris_setclearrjoinrrsetr)rtimeout completedjoinermessages rgetz Assembler.get;sS(ZZ ({{788##"#GHH#'D  ())..w7 ZZ #(D "]73-q#ABB{{788((//1 11  ! ! ' ' )"&,,"63BF"KK 4G++224 44  $ $ &DK$$, ,,1   ( (  s6E2CE'E$'E0c#rK|j5|jr td|jr t d|j }g|_t dtj|_ |jjr|jjdd|_dddEd{ |jj}|n|"|j5d|_|jjsJ|jj|jr td|jjrJ|jj!|j gk(sJd|_ dddy#1swYxYw7#1swYyxYww)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)putr1r*rr,)rrchunks rget_iterzAssembler.get_iterws&ZZ ({{788##"#GHH[[FDK $3!!#!D $$++-!!%%d+#'D + (0%%))+E}K  ZZ %#(D ((//1 11  ! ! ' ' ){{788++224 44  $ $ &;;"$ $$ $D  % %? ( (0  % %sB F7B!F0 F7=F)>3F71B#F+ F7F&"F7+F40F7c|j5|jr td|jr t d|j t jurtd|_ nK|j t jurd|_ n'|j t jurn dddy|j1|jj|j|j}n |j}|j|j j#|n|jj%||js dddy|j&j)rJ|j&j+|j|jj%d|j,j)rJd|_ddd|j,j/|j5d|_|j,j)sJ|j,j1|jr tdd|_ dddy#1swYxYw#1swYyxYw)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 TEXT UTF8DecoderrBINARYCONTdecodedatafinrrappendr5rr)r,rr'r*)rframerAs rr5z Assembler.puts  ZZ) ({{788##"#;<<||v{{**(; .# ,) () ($||'||**5::uyyAzz  ( ""4(!!%%d+99;) () (B,,335 55  ! ! % % '  ,!!%%d+++224 44#'D S) (X !!# ZZ #(D ''..0 00  & & ({{788DL ]) () (\ s'BI*5BI*BI*AI6*I36I?cf|j5|jr dddyd|_|jrA|jj |j |j j d|jr|jj dddy#1swYyxYw)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,rr5rrrs rclosezAssembler.closesZZ +{{ + +DK##%%))+$$0%%))$/##$$((* + + +sB'A:B''B0)returnNone)N)r-zOptional[float]rGr )rGzIterator[Data])rDr rGrH) __name__ __module__ __qualname____doc__rr1r7r5rFrrr r s& "H:x@%DH T+r) __future__rcodecsr3rtypingrrrrframesr r r __all__getincrementaldecoderr=r rMrrrTsD" 11" -*f**73 H+H+r