g~&ddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZdgZej*d Ze d ZGd d e eZGd dZy)) annotationsN)Any AsyncIteratorCallableGenericIterableTypeVar)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data Assemblerzutf-8Tc@eZdZdZd dZd dZd dZd dZd dZd 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 P/var/www/openai/venv/lib/python3.12/site-packages/websockets/asyncio/messages.py__init__zSimpleQueue.__init__#s),,. 7;+6+<+<+> c,t|jSr)lenrrs r __len__zSimpleQueue.__len__(s4::r"c|jj||j7|jjs|jj dyyy)z+Put an item into the queue without waiting.N)rappendrdone set_result)ritems r putzSimpleQueue.put+sH $ ?? &t/C/C/E OO & &t ,0F &r"c~K|jsl|j td|jj |_ |jd{|jj d|_|jj S7?#|jj d|_wxYww)z?Remove and return an item from the queue, waiting if necessary.Nzget is already running)rrr r create_futurecancelpopleftrs r getzSimpleQueue.get1szz*&'?@@"ii557DO 'oo%%&&("&zz!!## &&&("&s0AB=BBB;B=B#B::B=c|jJd|jrJd|jj|y)z)Put back items into an empty, idle queue.Nz%cannot reset() while get() is runningz&cannot reset() while queue isn't empty)rrextend)ritemss r resetzSimpleQueue.reset>s<&O(OO&::GGG~ % r"c|j>|jjs$|jjtd|jj y)Nstream of frames ended)rr( set_exceptionEOFErrorrclearrs r abortzSimpleQueue.abortDsC ?? &t/C/C/E OO ) )(3K*L M r"NreturnNone)r<int)r*rr<r=)r<r)r3z Iterable[T]r<r=) __name__ __module__ __qualname____doc__r!r%r+r0r4r:r"r rrs%? - $! r"rcleZdZdZddddf d dZdddZdddZdd Zdd Zdd Z dd Z 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. NcyrrCrCr"r zAssembler._s4r"cyrrCrCr"r rGzAssembler.`sDr"ct|_||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)rrMrNrOrPs r r!zAssembler.__init__[s{+6- ;!)C 7DE E #:HI I"C 48    % r"cVK|jr td|jr tdd|_ |jj d{}|j|jtus|jtusJ||jtu}|g}|jse |jj d{}|j|jtusJ|j||jsed|_dj!d|D}|r|j#S|S7#t j$r d|_wxYw7#t j$r$|jj|d|_wxYww)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: 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. ConcurrencyError: If two coroutines run :meth:`get` or :meth:`get_iter` concurrently. r6&get() or get_iter() is already runningTNFr"c34K|]}|jywr)data).0frames r z Assembler.get..s7u s)rSr8rRr rKr0rCancelledError maybe_resumeopcoderr finr4r r'joindecode)rr`rYrKrWs r r0z Assembler.getzs|* ;;34 4   "#KL L $ ++//++E ||w&%,,)*CCC >\\W,F)) "kkoo//    <<7* ** MM% )) %xx777 ;;= K=,%% #(D  0))  !!&)',$  s`6F)EE EAF)5E/E-E/AF)3F) EE**F)-E//7F&&F)cdK|jr td|jr tdd|_ |jj d{}|j|jtus|jtusJ||jtu}|r4t}|j|j|jn|j|js|jj d{}|j|jt usJ|r*j|j|jn|j|jsd|_y76#t j$r d|_wxYw7w)ap 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. ConcurrencyError: If two coroutines run :meth:`get` or :meth:`get_iter` concurrently. r6rUTNF)rSr8rRr rKr0rr[r\r]rr UTF8Decoderr`rWr^r )rr`rYdecoders r get_iterzAssembler.get_iters\0 ;;34 4   "#KL L $ ++//++E ||w&%,,)*CCC >\\W,F !mG..UYY7 7** )) ++//++E    <<7* **nnUZZ;;jj )) %9,%% #(D  &,sB6F0FF FB8F0F.A/F0F0 FF++F0c|jr td|jj||j y)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r6N)rSr8rKr+ maybe_pause)rrYs r r+z Assembler.puts3 ;;34 4  r"ct|j|jkDr%|jsd|_|j yyy)z7Pause the writer if queue is above the high water mark.TN)r$rKrMrQrOrs r rfzAssembler.maybe_pause s7 t{{ dii ' DK JJL1< 'r"ct|j|jkr%|jrd|_|j yyy)z7Resume the writer if queue is below the low water mark.FN)r$rKrNrQrPrs r r\zAssembler.maybe_resumes7 t{{ txx 'DKKDK KKM-8 'r"c`|jryd|_|jjy)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)rSrKr:rs r closezAssembler.closes' ;;   r") rMr>rNz int | NonerOCallable[[], Any]rPrkr<r=r)r` bool | Noner<r)r`rlr<zAsyncIterator[Data])rYrr<r=r;) r?r@rArBr!r0rdr+rfr\rjrCr"r rrKsj "#/$0 !  "   >>@?%B r") __future__rrcodecsrtypingrrrrrr exceptionsr rKr r rrr__all__getincrementaldecoderrbrrrrCr"r rssj" *77 -*f**73  CL-'!*-`ZZr"