j,BddlmZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZddlmZmZddlmZmZmZmZddl mZd gZej4d Zeej:j=d d Zed Z Gdde e Z!Gdd Z"y)) annotationsN) AsyncIteratorSequence)AnyCallableGenericLiteralTypeVaroverload)ConcurrencyErrorTooManyFragments)BINARYCONTTEXTFrame)Data Assemblerzutf-8WEBSOCKETS_MAX_FRAGMENTS4096TcBeZdZdZd dZd dZd dZd d dZddZd dZ y) SimpleQueuez Simplified version of :class:`asyncio.Queue`. Provides only the subset of functionality needed by :class:`Assembler`. cvtj|_d|_t j |_yN)asyncioget_running_looploop get_waiter collectionsdequequeueselfs h/home/oem/CyberBrain26-unified/dev-env/.venv/lib/python3.12/site-packages/websockets/asyncio/messages.py__init__zSimpleQueue.__init__ s),,. 7;+6+<+<+> c,t|jSr)lenr"r#s r%__len__zSimpleQueue.__len__%s4::r'c|jj||j7|jjs|jj dyyy)zPut an item into the queue.N)r"appendrdone set_result)r$items r%putzSimpleQueue.put(sH $ ?? &t/C/C/E OO & &t ,0F &r'cK|jsu|s td|jJd|jj |_ |jd{|jj d|_|jj S7?#|jj d|_wxYww)z?Remove and return an item from the queue, waiting if necessary.stream of frames endedNzcannot call get() concurrently)r"EOFErrorrr create_futurecancelpopleft)r$blocks r%getzSimpleQueue.get.szz788??* L,L L*"ii557DO 'oo%%&&("&zz!!## &&&("&s0A CB BB #;CB #CCcr|jJd|jjt|y)z)Put back items at the front of the queue.Nz(cannot put_back() while get() is running)rr" extendleftreversed)r$itemss r%put_backzSimpleQueue.put_back<s.&R(RR& huo.r'c|j@|jjs%|jjtdyyy)z8Close the queue, raising EOFError in get() if necessary.Nr2)rr- set_exceptionr3r#s r%abortzSimpleQueue.abortAs< ?? &t/C/C/E OO ) )(3K*L M0F &r'NreturnNone)rBint)r/rrBrC)T)r7boolrBr)r<z Sequence[T]rBrC) __name__ __module__ __qualname____doc__r&r*r0r8r=r@r'r%rrs&? - $/ Nr'rceZdZdZddddf ddZeddZeddZedddZddd Zedd Zedd Zeddd Zddd ZddZddZ ddZ ddZ 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. NcyrrJrJr'r%zAssembler.Zr'cyrrJrJr'r%rMzAssembler.[rNr'ct|_|||dz}|||dz}|"| |dkr td||kr td||c|_|_||_||_d|_d|_d|_ y)Nrz%low must be positive or equal to zeroz)high must be greater than or equal to lowF) rframes ValueErrorhighlowpauseresumepausedget_in_progressclosed)r$rTrUrVrWs r%r&zAssembler.__init__Vs+6-   !)C z Assembler.get..s7u 7s)rYr rRr8rZ maybe_resumeopcoderrfinrCancelledErrorr=rr) MAX_FRAGMENTSrr,joinr])r$r]rfrRrds r%r8z Assembler.getst,   "#KL L#  )++//dkk/::E    <<4'5<<6+A AA~-WFii"&++//dkk/"BBE!!#||t+++v;-/*=99 e$ii"$)D xx777 ;;= KA;C-- KK((0 $)D sZF2)F& E. AF&*)E3E1E3A"F&;3F2.F&1E330F##F&& F//F2cyrrJr\s r%get_iterzAssembler.get_itersEHr'cyrrJr\s r%rozAssembler.get_itersHKr'cyrrJr\s r%rozAssembler.get_itersKNr'cK|jr tdd|_ |jj|j d{}|j|jtus|jtusJ||jtu}|r4t}|j|j|jnt|j|js|jj|j d{}|j|jt usJ|r*j|j|jnt|j|jsd|_y7T#t j $r d|_wxYw7w)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. rbTNF)rYr rRr8rZrrkrhrirr UTF8Decoderr]rdrjbytesr)r$r]rfdecoders r%rozAssembler.get_iterse2   "#KL L# ++//dkk/::E ||t#u||v'=== >\\T)F !mG..UYY7 7 # #)) ++//dkk/::E    <<4' ''nnUZZ;;EJJ'')) %=;%% #(D  (;sBG)F" F F"C GGA8GGF""F>>Gc|jr td|jj||j y)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r2N)rZr3rRr0 maybe_pause)r$rfs r%r0z Assembler.puts3 ;;34 4  r'c|jyt|j|jkDr%|jsd|_|j yyy)z7Pause the writer if queue is above the high water mark.NT)rTr)rRrXrVr#s r%rwzAssembler.maybe_pausesF 99   t{{ dii ' DK JJL1< 'r'c|jyt|j|jkr%|jrd|_|j yyy)z7Resume the writer if queue is below the low water mark.NF)rUr)rRrXrWr#s r%rhzAssembler.maybe_resume)sF 88   t{{ txx 'DKKDK KKM-8 'r'c`|jryd|_|jjy)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`. NT)rZrRr@r#s r%closezAssembler.close4s' ;;   r') rT int | NonerUr|rVCallable[[], Any]rWr}rBrC)r] Literal[True]rBstr)r]Literal[False]rBrtr)r] bool | NonerBr)r]r~rBzAsyncIterator[str])r]rrBzAsyncIterator[bytes])r]rrBzAsyncIterator[Data])rfrrBrCrA) rFrGrHrIr&r r8ror0rwrhr{rJr'r%rrGs   #/$0   !  "   D:: == @@?BHH KK NNC%J   r')# __future__rrcodecsr oscollections.abcrrtypingrrrr r r exceptionsr rrRrrrrr__all__getincrementaldecoderrsrDenvironr8rlrrrrJr'r%rs"  3EE;.. -*f**73 BJJNN#=vFG  CL,N'!*,N^{{r'