Jtj*ddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZddlmZmZmZmZddlmZd gZejd Ze d ZGd d e eZGdd ZdS)) annotationsN) AsyncIteratorIterable)AnyCallableGenericLiteralTypeVaroverload)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 Y/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/websockets/asyncio/messages.py__init__zSimpleQueue.__init__s-,.. 7;+6+<+>+> intc*t|jSr)lenr!r"s r$__len__zSimpleQueue.__len__"s4:r&itemrc|j||j5|js|jddSdSdS)z+Put an item into the queue without waiting.N)r!appendrdone 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)r!EOFErrorrr create_futurecancelpopleft)r#r1s 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)rr!extend)r#r:s r$resetzSimpleQueue.reset9sK&&(O&&&:GGGGG~ %     r&c|jB|js+|jtddSdSdS)z8Close the queue, raising EOFError in get() if necessary.Nr4)rr. set_exceptionr5r"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)r1r2rr)r:r;rr) __name__ __module__ __qualname____doc__r%r*r0r9r>rAr&r$rrs???? ---- $ $ $ $ $!!!! NNNNNNr&rceZdZdZddddfd&d Zed'dZed(dZed)d*dZd)d*dZed+dZed,dZed)d-dZd)d-dZd.d"Zd/d#Z d/d$Z d/d%Z dS)0ra 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. NcdSrrGrGr&r$zAssembler.Ys4r&cdSrrGrGr&r$rJzAssembler.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 ValueErrorrLrNrOrQpausedget_in_progressclosed)r#rLrNrOrQs r$r%zAssembler.__init__Us+6--   !)C z Assembler.get..s$77u 777777r&)rWr rTr9rX maybe_resumeopcoderrfinrCancelledErrorr>rr-joinrY)r#rYrhrTrfs r$r9z Assembler.gets,   M"#KLL L#  )+//dk/::::::::E      <7**eli.G.G.GG~0WFi %"&+//dk/"B"BBBBBBBEE-K%%f---  !!###|w.... e$$$i %$)D 5D ( ( ( (xx7777777  ;;== Ks%A2D1&B;:D1;+C&&AD11 D:AsyncIterator[str]cdSrrGr]s r$get_iterzAssembler.get_itersEHSr&AsyncIterator[bytes]cdSrrGr]s r$rqzAssembler.get_itersHKr&AsyncIterator[Data]cdSrrGr]s r$rqzAssembler.get_itersKN3r&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. rdTNF)rWr rTr9rXrrmrjrkrr UTF8DecoderrYrfrlr)r#rYrhdecoders r$rqzAssembler.get_iters2   M"#KLL L# +//dk/::::::::EE%   #(D   |w&&%,)*C*C*CC >\W,F  !mmG..UY77 7 7 7 7 7*    ) ! +//dk/::::::::E      <7**** !nnUZ;;;;;;;j    ) ! %s &AA rhrc|jrtd|j||dS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r4N)rXr5rTr0 maybe_pause)r#rhs r$r0z Assembler.putsI ; 5344 4  r&c|jdSt|j|jkr$|jsd|_|dSdSdS)z7Pause the writer if queue is above the high water mark.NT)rLr)rTrVrOr"s r$rzzAssembler.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)rNr)rTrVrQr"s r$rjzAssembler.maybe_resume!sW 8  F t{  tx ' 'DK 'DK KKMMMMM ( ' ' 'r&cX|jrdSd|_|jdS)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)rXrTrAr"s r$closezAssembler.close,s6 ;  F  r&) rLrMrNrMrOrPrQrPrr)rYrZrr[)rYr^rr_r)rYrarr)rYrZrro)rYr^rrr)rYrarrt)rhrrrrB) rCrDrErFr%r r9rqr0rzrjr~rGr&r$rrEsa  " #/<$0L     D:::X: ===X= @@@@X@:::::xHHHXH KKKXK NNNNXNA%A%A%A%A%F            r&) __future__rrcodecsrcollections.abcrrtypingrrrr r r exceptionsr rTrrrrr__all__getincrementaldecoderrwrrrrGr&r$rs`"""""" 33333333EEEEEEEEEEEEEEEE))))))777777777777 -*f*733  GCLL-N-N-N-N-N'!*-N-N-N`uuuuuuuuuur&