o g-@sddlmZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z mZmZmZddlmZdd lmZd gZed ZGd d d ZdS) ) annotationsN)AnyCallableIterableIterator)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data)Deadline Assemblerzutf-8c@seZdZdZddddddfd*ddZd+d,ddZd-ddZd.d/ddZd+d0dd Zd1d"d#Z d2d$d%Z d2d&d'Z d2d(d)Z dS)3ra 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. NcCdSNrrrO/var/www/visachat/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 | Noner cCs|jrz |jjdd}Wn:tjytddwz|dur+|dkr+|jjdd}n|jjd|d}WntjyFtd|dd dw|durOtd|S) NFblockstream of frames endedrT)r/r,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)NTFr.)r"appendr%r2r#r3put)r*r%queuedr6rrr reset_queue^s"zAssembler.reset_queuedecode bool | Noner cCsn|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).0r6rrr sz Assembler.get..)r"r(rrr7r, maybe_resumeopcoder r finr5r<r r9joinr=)r*r,r=deadliner6r%rBrrrr2qsH       z Assembler.getIterator[Data]ccs<|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. r?TNF) r"r(rr7rErFr r UTF8Decoderr=rBrGr )r*r=r6decoderrrrget_iters8     zAssembler.get_iterr6cCsN|j|jr td|j||WddS1s wYdS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r0N)r"r)r4r%r: maybe_pause)r*r6rrrr: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*rrrrN  zAssembler.maybe_pausecCsL|jdurdS|jsJ|j|jkr"|jr$d|_|dSdSdS)z7Resume the writer if queue is below the low water mark.NF)rr"rOr%rPr'rrQrrrrErRzAssembler.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'rrQrrrclose)s   "zAssembler.close) rrrrrrrrrrr)r,r-rr )r%r8rr)NN)r,r-r=r>rr )r=r>rrJ)r6r rr)rr) __name__ __module__ __qualname____doc__r+r7r<r2rMr:rNrErSrrrrrs &   G >   ) __future__rcodecsr#r typingrrrr exceptionsrr%r r r r r utilsr__all__getincrementaldecoderrKrrrrrs