xj*ddlmZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl m Z ddlmZmZmZmZddlmZd gZejd ZGd d ZdS) ) annotationsN) AsyncIterator)AnyCallableLiteraloverload)ConcurrencyError)BINARYCONTTEXTFrame)Data Assemblerzutf-8ceZdZdZddddfd&d Zed'dZed(dZed)d*dZd)d*dZed+dZed,dZed)d-dZd)d-dZd.d"Zd/d#Z d/d$Z d/d%Z dS)0ra 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. NcdSNrIC:\Projects\Historical\venv\Lib\site-packages\websockets/trio/messages.pyzAssembler.'s4rcdSrrrrrrzAssembler.(sDrhigh int | NonelowpauseCallable[[], Any]resumereturnNonecH||tjtj\|_|_|||dz}|||dz}|,|*|dkrt d||krt d||c|_|_||_ ||_ d|_ d|_ d|_ dS)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 ValueErrorrrrrpausedget_in_progressclosed)selfrrrrs r__init__zAssembler.__init__#s 7 :-1-Edh-O-O*$*   !)C z Assembler.get..s$77u 777777r)r+r r(receiver# EndOfChannelEOFError maybe_resumeopcoder r fin Cancelledr'_state receive_tasksr> send_nowaitr appendjoinr/)r-r/r@framesstater>s rr4z Assembler.getPsU,   M"#KLL L# ! ) 9".6688888888$ 9 9 97888 9      <4''5<6+A+A+AA~-WFi % ="&"2":":"<"<<<<<<!'<<(44U;;;;(==="#;<<<=!!###|t++++ e$$$#i %($)D 5D ( ( ( (xx7777777  ;;== Ks8AFA!!AF1CFA;E  AF F AsyncIterator[str]cdSrrr3s rget_iterzAssembler.get_itersEHSrAsyncIterator[bytes]cdSrrr3s rrRzAssembler.get_itersHKrAsyncIterator[Data]cdSrrr3s rrRzAssembler.get_itersKN3rcK|jrtdd|_ |jd{V}n8#tj$r d|_tj$rtdwxYw||j tus|j tusJ||j tu}|r2t}| |j|jWVnt!|jWV|js |jd{V}n"#tj$rtdwxYw||j t"usJ|r$| |j|jWVnt!|jWV|jd|_dS)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. r;TNFr<)r+r r(rBr#rHrCrDrErFr r UTF8Decoderr/r>rGr6r )r-r/r@decoders rrRzAssembler.get_iters$2   M"#KLL L# 5*2244444444EE~   #(D   5 5 5344 4 5 |t##u|v'='='== >\T)F  $!mmG..UY77 7 7 7 7 7 ## # # # #) (  9".6688888888$ 9 9 97888 9      <4'''' (nnUZ;;;;;;;EJ'''''') (" %sA5A6D..E r@rc|jrtd|j||dS)z Add ``frame`` to the next message. Raises: EOFError: If the stream of frames has ended. r<N)r,rDr'rK maybe_pause)r-r@s rputz Assembler.putsL ; 5344 4 $$U+++ rc|jdSt|jjj|jkr$|jsd|_|dSdSdS)z7Pause the writer if queue is above the high water mark.NT)rlenr'rIr>r*rr-s rr[zAssembler.maybe_pauses^ 9  F t&+ , ,ty 8 8 8DK JJLLLLL 9 8 8 8rc|jdSt|jjj|jkr$|jrd|_|dSdSdS)z7Resume the writer if queue is below the low water mark.NF)rr^r'rIr>r*rr_s rrEzAssembler.maybe_resumes^ 8  F t&+ , , 8 8T[ 8DK KKMMMMM 9 8 8 8rcx|jrdSd|_|jd|_d|_dS)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`. NTcdSrrrrrrz!Assembler.close.."sTrcdSrrrrrrz!Assembler.close..#sdr)r,r'closerrr_s rrdzAssembler.closesI ;  F     "\ "l r) rrrrrrrrrr )r/r0rr1)r/r5rr6r)r/r8rr)r/r0rrP)r/r5rrS)r/r8rrU)r@rrr )rr ) __name__ __module__ __qualname____doc__r.rr4rRr\r[rErdrrrrrsf    #/<$0L """""H:::X: ===X= @@@@X@EEEEENHHHXH KKKXK NNNNXNH%H%H%H%H%T            ######r) __future__rcodecsr%collections.abcrtypingrrrrr# exceptionsr rNr r r rr__all__getincrementaldecoderrXrrrrrps"""""" ))))))333333333333 ))))))............ -*f*733 O#O#O#O#O#O#O#O#O#O#r