o KK&h?1@sddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZdd lmZd gZed ZGd d d ZdS) ) annotationsN)AnyCallableIterableIteratorLiteraloverload)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data)Deadline Assemblerzutf-8c@seZdZdZddddddfd8ddZd9d:ddZd;ddZedd?d%dZd>d?d&dZed@d(d)Z edAd+d)Z ed9dBd-d)Z d9dBd.d)Z dCd0d1Z dDd2d3Z dDd4d5Z dDd6d7Z dS)Era 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. NcCdSNrrrK/var/www/html/venv/lib/python3.10/site-packages/websockets/sync/messages.py&zAssembler.cCrrrrrrrr'rhigh int | NonelowpauseCallable[[], Any]resumereturnNonecCst|_t|_|dur|dur|d}|dur"|dur"|d}|dur:|dur:|dkr2td||kr:td|||_|_||_ ||_ d|_ d|_ d|_ dS)Nrz%low must be positive or equal to zeroz)high must be greater than or equal to lowF) threadingLockmutexqueue SimpleQueueframes ValueErrorrrrrpausedget_in_progressclosed)selfrrrrrrr__init__"s"  zAssembler.__init__timeout float | NonercCs|jrz |jjdd}Wn:tjytddwz|dur+|dkr+|jjdd}n|jjd|d}WntjyFtd|dd dw|durOtd|S) NFblockstream of frames endedrT)r1r.z timed out in z.1fs)r+r'getr%EmptyEOFError TimeoutError)r,r.framerrrget_next_frameHs" zAssembler.get_next_framer'Iterable[Frame]c Cs|j8g}z ||jjddqtjyYnw|D]}|j|q|D]}|j|q*WddS1s>wYdS)NTFr0)r$appendr'r4r%r5put)r,r'queuedr8rrr reset_queue^s"zAssembler.reset_queuedecode Literal[True]strcCrrrr,r.r?rrrr4tz Assembler.getLiteral[False]bytescCrrrrBrrrr4wrCcCrrrrBrrrr4zrCcCrrrrBrrrr4}rC bool | NonercCrrrrBrrrr4rCcCsn|j|jr tdd|_Wdn1swYzt|}||jdd}|j |Wdn1s=wY|jtusN|jt usNJ|durW|jtu}|g}|j sz ||jdd}Wn t yu| |w|j |Wdn1swY|jt usJ|||j r]Wd|_nd|_wddd|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: timeout: If a timeout is provided and elapses before a complete message is received, :meth:`get` raises :exc:`TimeoutError`. 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. TimeoutError: If a timeout is provided and elapses before a complete message is received. &get() or get_iter() is already runningTNF)raise_if_elapsedcss|]}|jVqdSr)data).0r8rrr sz Assembler.get..)r$r*r rr9r. maybe_resumeopcoder r finr7r>r r;joinr?)r,r.r?deadliner8r'rJrrrr4sH        Iterator[str]cCrrrr,r?rrrget_iterrCzAssembler.get_iterIterator[bytes]cCrrrrSrrrrTrCIterator[Data]cCrrrrSrrrrTrCccs<|j|jr tdd|_Wdn1swY|}|j |Wdn1s4wY|jtusE|jtusEJ|durN|jtu}|r]t}| |j |j Vn|j V|j s|}|j |Wdn1szwY|jt usJ|r| |j |j Vn|j V|j rdd|_dS)a Stream the next message. Iterating the return value of :meth:`get_iter` 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. rGTNF) r$r*r r9rMrNr r UTF8Decoderr?rJrOr )r,r?r8decoderrrrrTs8     r8cCsN|j|jr td|j||WddS1s wYdS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r2N)r$r+r6r'r< maybe_pause)r,r8rrrr<s   "z Assembler.putcCsL|jdurdS|jsJ|j|jkr"|js$d|_|dSdSdS)z7Pause the writer if queue is above the high water mark.NT)rr$lockedr'qsizer)rr,rrrrY*  zAssembler.maybe_pausecCsL|jdurdS|jsJ|j|jkr"|jr$d|_|dSdSdS)z7Resume the writer if queue is below the low water mark.NF)rr$rZr'r[r)rr\rrrrM7r]zAssembler.maybe_resumecCs|j3|jr WddSd|_|jr|jd|jr.d|_|WddSWddS1s9wYdS)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`. NTF)r$r+r*r'r<r)rr\rrrcloseDs   "zAssembler.close) rrrrrrrrrr r)r.r/rr)r'r:rr )r.r/r?r@rrA)r.r/r?rDrrE)NN)r.r/r?rFrr)r?r@rrR)r?rDrrU)r?rFrrV)r8rrr )rr )__name__ __module__ __qualname____doc__r-r9r>rr4rTr<rYrMr^rrrrrs> &    G    >   ) __future__rcodecsr%r"typingrrrrrr exceptionsr r'r r r rrutilsr__all__getincrementaldecoderrWrrrrrs