vg(ddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z ddl mZddlmZmZmZmZddlmZd gZejd Ze d ZGd d e eZGdd ZdS)) annotationsN) AsyncIteratorIterable)AnyCallableGenericTypeVar)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data Assemblerzutf-8TcDeZdZdZddZddZdd Zddd ZddZddZ dS) SimpleQueuez Simplified version of :class:`asyncio.Queue`. Provides only the subset of functionality needed by :class:`Assembler`. returnNonecttj|_d|_t j|_dSN)asyncioget_running_looploop get_waiter collectionsdequequeueselfs X/var/www/pixelcanvas.ch/venv/lib/python3.11/site-packages/websockets/asyncio/messages.py__init__zSimpleQueue.__init__s-,.. 7;+6+<+>+> intc*t|jSr)lenrr s r"__len__zSimpleQueue.__len__"s4:r$itemrc|j||j5|js|jddSdSdS)z+Put an item into the queue without waiting.N)rappendrdone set_result)r!r)s r"putzSimpleQueue.put%s] $ ? &t/C/C/E/E & O & &t , , , , , ' & & &r$TblockboolcnK|js|std|j Jd|j|_ |jd{V|jd|_n%#|jd|_wxYw|jS)z?Remove and return an item from the queue, waiting if necessary.stream of frames endedNzcannot call get() concurrently)rEOFErrorrr create_futurecancelpopleft)r!r/s r"getzSimpleQueue.get+sz ' 97888?**,L***"i5577DO 'o%%%%%%%&&((("&&&((("&&&&&z!!###s A99"Bitems Iterable[T]c~|j Jd|jr Jd|j|dS)z)Put back items into an empty, idle queue.Nz%cannot reset() while get() is runningz&cannot reset() while queue isn't empty)rrextend)r!r8s r"resetzSimpleQueue.reset9sL&&(O&&&:GGGGGG %     r$c|jB|js+|jtddSdSdS)z8Close the queue, raising EOFError in get() if necessary.Nr2)rr, set_exceptionr3r s r"abortzSimpleQueue.abort?sP ? &t/C/C/E/E & O ) )(3K*L*L M M M M M ' & & &r$Nrr)rr%)r)rrr)T)r/r0rr)r8r9rr) __name__ __module__ __qualname____doc__r#r(r.r7r<r?r$r"rrs???? ---- $ $ $ $ $!!!! NNNNNNr$rc\eZdZdZddddfdd ZdddZdddZddZddZddZ ddZ dS) 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. NcdSrrErEr$r"zAssembler.Ys4r$cdSrrErEr$r"rHzAssembler.ZsDr$high int | NonelowpauseCallable[[], Any]resumerrct|_|||dz}|||dz}|,|*|dkrtd||krtd||c|_|_||_||_d|_d|_d|_ dS)Nrz%low must be positive or equal to zeroz)high must be greater than or equal to lowF) rframes ValueErrorrJrLrMrOpausedget_in_progressclosed)r!rJrLrMrOs r"r#zAssembler.__init__Us+6--   !)C z Assembler.get..s$77u 777777r$)rUr rRr7rV maybe_resumeopcoderr finrCancelledErrorr<r r+joinrW)r!rWr^rRr\s r"r7z Assembler.getws,   M"#KLL L#  )+//dk/::::::::E      <7**eli.G.G.G.G~0WFi %"&+//dk/"B"BBBBBBBEE-K%%f---  !!###|w.... e$$$i %$)D 5D ( ( ( (xx7777777  ;;== Ks%A2D1&B;:D1;+C&&AD11 D:AsyncIterator[Data]cK|jrtdd|_ |j|j d{V}n#t j$r d|_wxYw||jtus|jtusJ||jtu}|r2t}| |j |jWVn |j WV|js|j|j d{V}||jtusJ|r$| |j |jWVn |j WV|jd|_dS)a Stream the next message. Iterating the return value of :meth:`get_iter` asynchronously 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. rZTNF)rUr rRr7rVrrcr`rarr UTF8DecoderrWr\rbr )r!rWr^decoders r"get_iterzAssembler.get_iters2   M"#KLL L# +//dk/::::::::EE%   #(D   |w&&%,)*C*C*C*C >\W,F  !mmG..UY77 7 7 7 7 7*    ) ! +//dk/::::::::E      <7**** !nnUZ;;;;;;;j    ) ! %s &AA r^rc|jrtd|j||dS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r2N)rVr3rRr. maybe_pause)r!r^s r"r.z Assembler.putsI ; 5344 4  r$c|jdSt|j|jkr$|jsd|_|dSdSdS)z7Pause the writer if queue is above the high water mark.NT)rJr'rRrTrMr s r"rkzAssembler.maybe_pausesW 9  F t{  di ' ' 'DK JJLLLLL ( ' ' 'r$c|jdSt|j|jkr$|jrd|_|dSdSdS)z7Resume the writer if queue is below the low water mark.NF)rLr'rRrTrOr s r"r`zAssembler.maybe_resumesW 8  F t{  tx ' 'DK 'DK KKMMMMM ( ' ' 'r$cX|jrdSd|_|jdS)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)rVrRr?r s r"closezAssembler.closes6 ;  F  r$) rJrKrLrKrMrNrOrNrrr)rWrXrr)rWrXrre)r^rrrr@) rArBrCrDr#r7rir.rkr`rorEr$r"rrEs  " #/<$0L     D:::::xA%A%A%A%A%F            r$) __future__rrcodecsrcollections.abcrrtypingrrrr exceptionsr rRr r rrr__all__getincrementaldecoderrgrrrrEr$r"rwsT"""""" 33333333222222222222))))))777777777777 -*f*733  GCLL-N-N-N-N-N'!*-N-N-N`ccccccccccr$