j2ddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z ddl mZmZddlmZmZmZmZddlmZdd lmZd gZej2d Zeej8j;d d ZGdd Zy)) annotationsN)AnyCallableIterableIteratorLiteraloverload)ConcurrencyErrorTooManyFragments)BINARYCONTTEXTFrame)Data)Deadline Assemblerzutf-8WEBSOCKETS_MAX_FRAGMENTS4096ceZdZdZddddf ddZdddZddZeddZedd Zeddd Zeddd Zeddd Zddd ZeddZ eddZ edd dZ dd dZ d!dZ d"dZ d"dZ d"dZ y)#ra Assemble messages from frames. :class:`Assembler` expects only data frames. The stream of frames must respect the protocol; if it doesn't, the behavior is undefined. Args: pause: Called when the buffer of frames goes above the high water mark; should pause reading from the network. resume: Called when the buffer of frames goes below the low water mark; should resume reading from the network. NcyNre/home/oem/CyberBrain26-unified/dev-env/.venv/lib/python3.12/site-packages/websockets/sync/messages.pyzAssembler.)rcyrrrrrrzAssembler.*rrc8tj|_tj|_|||dz}|||dz}|"| |dkr t d||kr t d||c|_|_||_ ||_ d|_ d|_ d|_ y)Nrz%low must be positive or equal to zeroz)high must be greater than or equal to lowF) threadingLockmutexqueue SimpleQueueframes ValueErrorhighlowpauseresumepausedget_in_progressclosed)selfr)r*r+r,s r__init__zAssembler.__init__%s^^% 8=7H7H7J   !)C ?;;   ' & '  ' & ' ' ' 's'B%->AB%AAB%%B.cyrrr0r6decodes rr8z Assembler.getwsHKrcyrrrDs rr8z Assembler.getzsKNrcyrrrDs rr8z Assembler.get}sRUrcyrrrDs rr8z Assembler.getUXrcyrrrDs rr8z Assembler.getrIrc|j5|jr tdd|_ddd t|}|j |j d}|j5|j ddd|jtus|jtusJ||jtu}|g}|js |j |j d}|j5|j ddd|jtusJt|tk\rt!t|j#||jsd|_dj%d|D}|r|j'S|S#1swYrxYw#1swY-xYw#t$r|j|wxYw#1swYxYw#d|_wxYw)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`. decode: :obj:`False` disables UTF-8 decoding of text frames and returns :class:`bytes`. :obj:`True` forces UTF-8 decoding of binary frames and returns :class:`str`. Raises: EOFError: If the stream of frames has ended. UnicodeDecodeError: If a text frame contains invalid UTF-8. ConcurrencyError: If two coroutines run :meth:`get` or :meth:`get_iter` concurrently. TimeoutError: If a timeout is provided and elapses before a complete message is received. &get() or get_iter() is already runningTNF)raise_if_elapsedrc34K|]}|jywr)data).0r<s r z Assembler.get..s7u 7s)r$r.r rr=r6 maybe_resumeopcoderr finr;rBrlen MAX_FRAGMENTSr r?joinrE)r0r6rEdeadliner<r'rOs rr8z Assembler.gets4ZZ (##&'OPP#'D  ( )(H''(8(8%(8(PQE $!!# $<<4'5<<6+A AA~-WFii // ((%(@EZZ(%%'(||t+++v;-/*=99 e$ii$$)D xx777 ;;= K] ( ( $ $$$$V,  (($)D s`F8G!-F)>AG!!F61 G!=GAG!F&)F3.G!6GG!GG!! G*cyrrr0rEs rget_iterzAssembler.get_iters@CrcyrrrZs rr[zAssembler.get_itersCFrcyrrrZs rr[zAssembler.get_itersFIrc#K|j5|jr tdd|_ddd|j}|j5|j ddd|j t us|j tusJ||j t u}|r3t}|j|j|jnt|j|js|j}|j5|j ddd|j tusJ|r)j|j|jnt|j|jsd|_y#1swYqxYw#1swYIxYw#1swYxYww)a 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:`ConcurrencyError` is raised. This method only makes sense for fragmented messages. If messages aren't fragmented, use :meth:`get` instead. Args: decode: :obj:`False` disables UTF-8 decoding of text frames and returns :class:`bytes`. :obj:`True` forces UTF-8 decoding of binary frames and returns :class:`str`. Raises: EOFError: If the stream of frames has ended. UnicodeDecodeError: If a text frame contains invalid UTF-8. ConcurrencyError: If two coroutines run :meth:`get` or :meth:`get_iter` concurrently. rLTNF)r$r.r r=rRrSrr UTF8DecoderrErOrTbytesr)r0rEr<decoders rr[zAssembler.get_itersn2ZZ (##&'OPP#'D  (##% ZZ     ||t#u||v'=== >\\T)F !mG..UYY7 7 # #))'')E $!!# $<<4' ''nnUZZ;;EJJ'')) %K ( (   $ $sS GF$GF*#B6GF7*A*GGF'"G*F4/G7G<Gc|j5|jr td|jj ||j dddy#1swYyxYw)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r5N)r$r/r:r'r@ maybe_pause)r0r<s rr@z Assembler.putsOZZ {{788 KKOOE "        s AAA"c|jy|jjsJ|jj |jkDr%|j sd|_|j yyy)z7Pause the writer if queue is above the high water mark.NT)r)r$lockedr'qsizer-r+r0s rrczAssembler.maybe_pause2s` 99  zz  """ ;;    *4;;DK JJL4? *rc|jy|jjsJ|jj |jkr%|j rd|_|j yyy)z7Resume the writer if queue is below the low water mark.NF)r*r$rer'rfr-r,rgs rrRzAssembler.maybe_resume?s` 88  zz  """ ;;   $(( *t{{DK KKM0; *rc|j5|jr dddyd|_|jr|jj d|j rd|_|j dddy#1swYyxYw)z End the stream of frames. Calling :meth:`close` concurrently with :meth:`get`, :meth:`get_iter`, or :meth:`put` is safe. They will raise :exc:`EOFError`. NTF)r$r/r.r'r@r-r,rgs rclosezAssembler.closeLsmZZ {{  DK## %{{#     sA>AA>>B) r) int | Noner*rkr+Callable[[], Any]r,rlreturnNoner)r6 float | Nonermr)r'zIterable[Frame]rmrn)r6rorE Literal[True]rmstr)r6rorELiteral[False]rmr`)NN)r6rorE bool | Nonermr)rErprmz Iterator[str])rErrrmzIterator[bytes])rErsrmzIterator[Data])r<rrmrn)rmrn)__name__ __module__ __qualname____doc__r1r=rBr r8r[r@rcrRrjrrrrrs   #/$0 $$$! $ " $  $L,',KK NN UU XX XXHTCC FF II>%@ 2  r) __future__rcodecsosr%r"typingrrrrrr exceptionsr r r'r rrrrutilsr__all__getincrementaldecoderr_intenvironr8rVrrrrrsj" GG;.. -*f**73 BJJNN#=vFG KKr