A>qj(ddlmZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl m Z ddlmZmZmZmZddlmZd gZej*d ZGd d Zy) ) annotationsN) AsyncIterator)AnyCallableLiteraloverload)ConcurrencyError)BINARYCONTTEXTFrame)Data Assemblerzutf-8ceZdZdZddddf 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. NcyNrW/opt/rentech/trading_bot/.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_resumeopcoder r fin Cancelledr _state receive_tasksr7 send_nowaitr appendjoinr.)r*r.r9framesstater7s rr/z Assembler.getPs,   "#KL L# ! ) 9"..6688    <<4'5<<6+A AA~-WFii ="&"2"2":":"<!'<((44U;<((="#;<<=$)D smG;D8D6D8AG/E<E=EAG/3G;6D88EG/EBG,,G// G88G;cyrrr-s rget_iterzAssembler.get_itersEHrcyrrr-s rrJzAssembler.get_itersHKrcyrrr-s rrJzAssembler.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. r4TNFr5)r(r r!r;rrAr<r=r>r?r r UTF8Decoderr.r7r@bytesr )r*r.r9decoders rrJzAssembler.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. r5N)r)r=r rD maybe_pause)r*r9s 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#lenr rBr7r'r%r*s rrRzAssembler.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$rUr rBr7r'r&rVs rr>zAssembler.maybe_resumesU 88   t&&++ , 8T[[DK KKM>I 8rc`|jryd|_|jjy)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)r)r closerVs rrYzAssembler.closes) ;;    r) r# int | Noner$rZr%Callable[[], Any]r&r[returnNone)r. Literal[True]r\str)r.Literal[False]r\rOr)r. bool | Noner\r)r.r^r\zAsyncIterator[str])r.r`r\zAsyncIterator[bytes])r.rar\zAsyncIterator[Data])r9rr\r])r\r]) __name__ __module__ __qualname____doc__r+rr/rJrSrRr>rYrrrrrs   #/$0 """! " " "  "H:: == @@ENHH KK NNH%T   !r) __future__rcodecsrcollections.abcrtypingrrrrr exceptionsr rGr r r rr__all__getincrementaldecoderrNrrrrrmsM" )33 ).. -*f**73 J!J!r