a Ñšhñ*ã@sÂddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZddlmZmZmZmZddlmZd gZe d ¡Ze d ƒZGd d „d e eƒZGdd „d ƒZdS)é)Ú annotationsN)Ú AsyncIteratorÚIterable)ÚAnyÚCallableÚGenericÚLiteralÚTypeVarÚoverloadé)ÚConcurrencyError)Ú OP_BINARYÚOP_CONTÚOP_TEXTÚFrame)ÚDataÚ Assemblerzutf-8ÚTc@sleZdZdZddœdd„Zddœdd„Zd dd œd d „Zddd dœdd„Zdddœdd„Zddœdd„Z dS)Ú SimpleQueuez… Simplified version of :class:`asyncio.Queue`. Provides only the subset of functionality needed by :class:`Assembler`. ÚNone©ÚreturncCst ¡|_d|_t ¡|_dS©N)ÚasyncioZget_running_loopÚloopÚ get_waiterÚ collectionsÚdequeÚqueue©Úself©r!úV/wd/webapps/venvs/v2025_4um/lib/python3.9/site-packages/websockets/asyncio/messages.pyÚ__init__s zSimpleQueue.__init__ÚintcCs t|jƒSr)Úlenrrr!r!r"Ú__len__"szSimpleQueue.__len__r)ÚitemrcCs0|j |¡|jdur,|j ¡s,|j d¡dS)z+Put an item into the queue without waiting.N)rÚappendrÚdoneZ set_result)r r'r!r!r"Úput%s zSimpleQueue.putTÚbool)ÚblockrcÃsn|jsd|stdƒ‚|jdus$Jdƒ‚|j ¡|_z |jIdHW|j ¡d|_n|j ¡d|_0|j ¡S)z?Remove and return an item from the queue, waiting if necessary.ústream of frames endedNzcannot call get() concurrently)rÚEOFErrorrrZ create_futureÚcancelÚpopleft)r r,r!r!r"Úget+s  ÿ zSimpleQueue.getz Iterable[T])ÚitemsrcCs0|jdusJdƒ‚|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 r2r!r!r"Úreset9szSimpleQueue.resetcCs(|jdur$|j ¡s$|j tdƒ¡dS)z8Close the queue, raising EOFError in get() if necessary.Nr-)rr)Z set_exceptionr.rr!r!r"Úabort?szSimpleQueue.abortN)T) Ú__name__Ú __module__Ú __qualname__Ú__doc__r#r&r*r1r4r5r!r!r!r"rsrc@seZdZdZdddd„dd„fdddddd œd d „Zed d dœdd„ƒZedddœdd„ƒZed+dddœdd„ƒZd,dddœdd„Zed ddœdd„ƒZedddœdd„ƒZed-dddœdd„ƒZd.dddœdd„Zd dd!œd"d#„Zdd$œd%d&„Z dd$œd'd(„Z dd$œd)d*„Z 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. NcCsdSrr!r!r!r!r"ÚYózAssembler.cCsdSrr!r!r!r!r"r:Zr;z int | NonezCallable[[], Any]r)ÚhighÚlowÚpauseÚresumercCs˜tƒ|_|dur |dur |d}|dur8|dur8|d}|durh|durh|dkrXtdƒ‚||krhtdƒ‚|||_|_||_||_d|_d|_d|_ dS)Nérz%low must be positive or equal to zeroz)high must be greater than or equal to lowF) rÚframesÚ ValueErrorr<r=r>r?ÚpausedÚget_in_progressÚclosed)r r<r=r>r?r!r!r"r#Us zAssembler.__init__z Literal[True]Ústr)ÚdecodercÃsdSrr!©r rGr!r!r"r1wsz Assembler.getzLiteral[False]ÚbytescÃsdSrr!rHr!r!r"r1zsz bool | NonercÃsdSrr!rHr!r!r"r1}scÃs|jrtdƒ‚d|_z¾|j |j ¡IdH}| ¡|jtusL|jtusLJ‚|dur^|jtu}|g}|j sÊz|j |j ¡IdH}Wn"t j y¦|j  |¡‚Yn0| ¡|jt us¾J‚| |¡qdWd|_nd|_0d dd„|Dƒ¡}|rü| ¡S|SdS)a0 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. UnicodeDecodeError: If a text frame contains invalid UTF-8. ConcurrencyError: If two coroutines run :meth:`get` or :meth:`get_iter` concurrently. ú&get() or get_iter() is already runningTNFr;css|] }|jVqdSr)Údata)Ú.0Úframer!r!r"Ú ¶r;z Assembler.get..)rDr rAr1rEÚ maybe_resumeÚopcoderr ÚfinrÚCancelledErrorr4rr(ÚjoinrG)r rGrMrArKr!r!r"r1€s0  zAsyncIterator[str]cCsdSrr!rHr!r!r"Úget_iter¼szAssembler.get_iterzAsyncIterator[bytes]cCsdSrr!rHr!r!r"rT¿szAsyncIterator[Data]cCsdSrr!rHr!r!r"rTÂscCs|jrtdƒ‚d|_z|j |j ¡IdH}WntjyJd|_‚Yn0| ¡|jt usl|jt uslJ‚|dur~|jt u}|rœt ƒ}|  |j |j¡Vn|j V|jsø|j |j ¡IdH}| ¡|jtusÖJ‚|rî|  |j |j¡Vq¤|j Vq¤d|_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. rJTNF)rDr rAr1rErrRrOrPrr Ú UTF8DecoderrGrKrQr)r rGrMÚdecoderr!r!r"rTÅs0   r)rMrcCs&|jrtdƒ‚|j |¡| ¡dS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r-N)rEr.rAr*Ú maybe_pause)r rMr!r!r"r*s z Assembler.putrcCs6|jdurdSt|jƒ|jkr2|js2d|_| ¡dS)z7Pause the writer if queue is above the high water mark.NT)r<r%rArCr>rr!r!r"rWs  zAssembler.maybe_pausecCs6|jdurdSt|jƒ|jkr2|jr2d|_| ¡dS)z7Resume the writer if queue is below the low water mark.NF)r=r%rArCr?rr!r!r"rO!s  zAssembler.maybe_resumecCs|jr dSd|_|j ¡dS)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)rErAr5rr!r!r"Úclose,szAssembler.close)N)N)N)N) r6r7r8r9r#r r1rTr*rWrOrXr!r!r!r"rEs0û"<C  )Ú __future__rrÚcodecsrÚcollections.abcrrÚtypingrrrrr r Ú exceptionsr rAr rrrrÚ__all__ÚgetincrementaldecoderrUrrrr!r!r!r"Ús     0