o DRh~&@sddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZdgZed Ze d ZGd d d e eZGd ddZdS)) annotationsN)Any AsyncIteratorCallableGenericIterableTypeVar)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data Assemblerzutf-8Tc@sLeZdZdZdddZdddZdd d Zdd dZdddZdddZ dS) SimpleQueuez Simplified version of :class:`asyncio.Queue`. Provides only the subset of functionality needed by :class:`Assembler`. returnNonecCst|_d|_t|_dSN)asyncioget_running_looploop get_waiter collectionsdequequeueselfr]/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/websockets/asyncio/messages.py__init__#s zSimpleQueue.__init__intcCs t|jSr)lenrrrrr __len__(s zSimpleQueue.__len__itemrcCs8|j||jdur|js|jddSdSdS)z+Put an item into the queue without waiting.N)rappendrdone set_result)rr%rrr put+s zSimpleQueue.putcsd|js-|jdur td|j|_z|jIdHW|jd|_n |jd|_w|jS)z?Remove and return an item from the queue, waiting if necessary.Nzget is already running)rrr r create_futurecancelpopleftrrrr get1s     zSimpleQueue.getitems Iterable[T]cCs0|jdus Jd|jrJd|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)rr.rrr reset>szSimpleQueue.resetcCs2|jdur|js|jtd|jdS)Nstream of frames ended)rr' set_exceptionEOFErrorrclearrrrr abortDszSimpleQueue.abortNrr)rr")r%rrr)rr)r.r/rr) __name__ __module__ __qualname____doc__r!r$r)r-r1r6rrrr rs     rc@sleZdZdZddddddfd$ddZd%d&ddZd%d'ddZd(ddZd)ddZd)d d!Z d)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. NcCdSrrrrrr _zAssembler.cCr=rrrrrr r>`r?highr"low int | NonepauseCallable[[], Any]resumerrcCsht|_|dur |d}|dkrtd||krtd|||_|_||_||_d|_d|_d|_ dS)Nrz%low must be positive or equal to zeroz)high must be greater than or equal to lowF) rframes ValueErrorr@rArCrEpausedget_in_progressclosed)rr@rArCrErrr r![s zAssembler.__init__decode bool | Nonercs|jrtd|jrtdd|_z |jIdH}Wn tjy(d|_w||j t us9|j t us9J|durB|j t u}|g}|j sxz |jIdH}Wntjyd|j |d|_w||j tuspJ|||j rHd|_ddd|D}|r|S|S) 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. r2&get() or get_iter() is already runningTNFcss|]}|jVqdSr)data).0framerrr sz Assembler.get..)rKr4rJr rGr-rCancelledError maybe_resumeopcoder r finr1r r&joinrL)rrLrRrGrPrrr r-zsD    z Assembler.getAsyncIterator[Data]cCs|jrtd|jrtdd|_z |jIdH}Wn tjy(d|_w||j t us9|j t us9J|durB|j t u}|rQt }| |j|jVn|jV|js~|jIdH}||j tuskJ|rw| |j|jVn|jV|jrXd|_dS)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. r2rNTNF)rKr4rJr rGr-rrTrUrVr r UTF8DecoderrLrPrWr )rrLrRdecoderrrr get_iters:  zAssembler.get_iterrRrcCs&|jrtd|j||dS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r2N)rKr4rGr) maybe_pause)rrRrrr r)s  z Assembler.putcCs0t|j|jkr|jsd|_|dSdSdS)z7Pause the writer if queue is above the high water mark.TN)r#rGr@rIrCrrrr r]  zAssembler.maybe_pausecCs0t|j|jkr|jrd|_|dSdSdS)z7Resume the writer if queue is below the low water mark.FN)r#rGrArIrErrrr rUr^zAssembler.maybe_resumecCs|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)rKrGr6rrrr closeszAssembler.close) r@r"rArBrCrDrErDrrr)rLrMrr)rLrMrrY)rRrrrr7) r8r9r:r;r!r-r\r)r]rUr_rrrr rKs  @ A  ) __future__rrcodecsrtypingrrrrrr exceptionsr rGr r r rr__all__getincrementaldecoderrZrrrrrrr s     0