^Qj+ ddlmZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl Z ddl mZmZddlmZmZmZmZddlmZd gZej.d Zeej4j7d d ZGd d Zy)) annotationsN) AsyncIterator)AnyCallableLiteraloverload)ConcurrencyErrorTooManyFragments)BINARYCONTTEXTFrame)Data Assemblerzutf-8WEBSOCKETS_MAX_FRAGMENTS4096ceZdZdZddddf ddZeddZeddZedddZddd Zedd Zedd Zeddd Zddd ZddZddZ ddZ ddZ 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. NcyNr]/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/websockets/trio/messages.pyzAssembler.*rcyrrrrrrzAssembler.+rrc<||tjtj\|_|_|||dz}|||dz}|"| |dkr t d||kr t d||c|_|_||_ ||_ d|_ d|_ d|_ y)Nrz%low must be positive or equal to zeroz)high must be greater than or equal to lowF)trioopen_memory_channelmathinf send_frames recv_frames ValueErrorhighlowpauseresumepausedget_in_progressclosed)selfr&r'r(r)s r__init__zAssembler.__init__&s  -1-E-Edhh-O*$*   !)C z Assembler.get..s7u 7s)r+r r$receiver EndOfChannelEOFError maybe_resumeopcoderr fin Cancelledr#_state receive_tasksr: extendleftreversedr len MAX_FRAGMENTSr appendjoinr1)r-r1r<framesstater:s rr2z Assembler.getSs,   "#KL L# ' ) 9"..6688    <<4'5<<6+A AA~-WFii="&"2"2":":"<<E<=E>A"H$3H EE99H<E>>A?G==H H  H cyrrr0s rget_iterzAssembler.get_itersEHrcyrrr0s rrPzAssembler.get_itersHKrcyrrr0s rrPzAssembler.get_itersKNrcK|jr tdd|_ |jjd{}|j|jtus|jtusJ||jtu}|r4t}|j|j|jnt!|j|js |jjd{}|j|jt"usJ|r*j|j|jnt!|j|jsd|_y7I#tj $r d|_tj $r tdwxYw7#tj $r tdwxYww)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. r7TNFr8)r+r r$r>rrDr?r@rArBrr UTF8Decoderr1r:rCbytesr )r-r1r<decoders rrPzAssembler.get_iters2   "#KL L# 5**2244E ||t#u||v'=== >\\T)F !mG..UYY7 7 # #))  9"..6688    <<4' ''nnUZZ;;EJJ''))" %G5~~ #(D    534 4 5(9$$ 9788 9s_G+F FF B$G+)G GG A4G+G+F 9GG+G G((G+c|jr td|jj||j y)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r8N)r,r@r# send_nowait maybe_pause)r-r<s rputz Assembler.puts7 ;;34 4 $$U+ rc|jyt|jjj|jkDr%|j sd|_|j yyy)z7Pause the writer if queue is above the high water mark.NT)r&rIr#rEr:r*r(r-s rrYzAssembler.maybe_pausesV 99   t&&++ ,tyy 8DK JJLBM 8rc|jyt|jjj|jkr%|j rd|_|j yyy)z7Resume the writer if queue is below the low water mark.NF)r'rIr#rEr:r*r)r\s rrAzAssembler.maybe_resume sU 88   t&&++ , 8T[[DK KKM>I 8rc|jryd|_|jjd|_d|_y)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`. NTcyrrrrrrz!Assembler.close..+rrcyrrrrrrz!Assembler.close..,rr)r,r#closer(r)r\s rrazAssembler.closes9 ;;    " " r) r& int | Noner'rbr(Callable[[], Any]r)rcreturnNone)r1 Literal[True]rdstr)r1Literal[False]rdrUr)r1 bool | Nonerdr)r1rfrdzAsyncIterator[str])r1rhrdzAsyncIterator[bytes])r1rirdzAsyncIterator[Data])r<rrdre)rdre) __name__ __module__ __qualname____doc__r.rr2rPrZrYrArarrrrrs   #/$0 """! " " "  "H:: == @@KZHH KK NNH%T   #r) __future__rcodecsr!oscollections.abcrtypingrrrrr exceptionsr r rMr r rrr__all__getincrementaldecoderrTintenvironr2rJrrrrrxsj" )33 ;.. -*f**73 BJJNN#=vFG U#U#r