qj:2ddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZdd lmZd gZej.d ZGd d Zy) ) annotationsN)AnyCallableIterableIteratorLiteraloverload)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data)Deadline Assemblerzutf-8ceZdZdZddddf 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. NcyNrY/var/www/html/kelly-backend/venv/lib/python3.12/site-packages/websockets/sync/messages.pyzAssembler.&s4rcyrrrrrrzAssembler.'sDrc8tj|_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 ?;;   &   & ZZs'B%->AB%AAB%%B.cyrrr,r2decodes rr4z Assembler.gettsHKrcyrrr@s rr4z Assembler.getwsKNrcyrrr@s rr4z Assembler.getzsRUrcyrrr@s rr4z Assembler.get}UXrcyrrr@s rr4z Assembler.getrErc|j5|jr tdd|_ddd t|}|j |j d}|j5|j ddd|jtus|jtusJ||jtu}|g}|jsy |j |j d}|j5|j ddd|jtusJ|j||jsyd|_djd|D}|r|j!S|S#1swYQxYw#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).0r8s r z Assembler.get..s7u s)r r*r rr9r2 maybe_resumeopcoderr finr7r>r r;joinrA)r,r2rAdeadliner8r#rKs rr4z Assembler.gets4ZZ##&'OPP#'D  )(H''(8(8%(8(PQE!!#<<7*elli.G GG~0WFii // ((%(@EZZ%%' ||w... e$ii $)D xx777 ;;= KYZ$$$V,   Z $)D s_E;8G-F>AG!F1 G=F49G;FF GF11G4F=9G G cyrrr,rAs rget_iterzAssembler.get_iters@CrcyrrrTs rrUzAssembler.get_itersCFrcyrrrTs rrUzAssembler.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. rHTNF)r r*r r9rNrOrr UTF8DecoderrArKrPbytesr )r,rAr8decoders rrUzAssembler.get_itersV2ZZ##&'OPP#'D ##% ZZ    ||w&%,,)*CCC >\\W,F !mG..UYY7 7 # #))'')E!!#<<7* **nnUZZ;;EJJ'')) %KZZsS 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. r1N)r r+r6r#r< maybe_pause)r,r8s rr<z Assembler.putsDZZ{{788 KKOOE "     ZZs 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'r,s rr]zAssembler.maybe_pause-s` 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 r_r#r`r)r(ras rrNzAssembler.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(ras rclosezAssembler.closeGs_ZZ{{ZDK## %{{#  ZZsA>AA>>B) r% int | Noner&rer'Callable[[], Any]r(rfreturnNoner)r2 float | Nonergr)r#zIterable[Frame]rgrh)r2rirA Literal[True]rgstr)r2rirALiteral[False]rgrZ)NN)r2rirA bool | Nonergr)rArjrgz Iterator[str])rArlrgzIterator[bytes])rArmrgzIterator[Data])r8rrgrh)rgrh)__name__ __module__ __qualname____doc__r-r9r>r r4rUr<r]rNrdrrrrrs   #/$0 $$$! $ " $  $L,',KK NN UU XX XXFPCC FF II>%@ 2  r) __future__rcodecsr!rtypingrrrrrr exceptionsr r#r r rrrutilsr__all__getincrementaldecoderrYrrrrrysM" GG)77 -*f**73 IIr