o `^hQ&@sddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZddlmZdgZed ZGd ddZdS) ) annotationsN)Iteratorcast)ConcurrencyError) OP_BINARYOP_CONTOP_TEXTFrame)Data Assemblerzutf-8c@sDeZdZdZdddZddd d Zdd dZdddZdddZdS)r z) Assemble messages from frames. returnNonecCsFt|_t|_t|_d|_d|_d|_g|_ d|_ d|_ dS)NF) threadingLockmutexEventmessage_completemessage_fetchedget_in_progressput_in_progressdecoderchunks chunks_queueclosedselfrV/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/websockets/sync/messages.py__init__s    zAssembler.__init__Ntimeout float | Noner cCs|j|jr td|jrtdd|_Wdn1swY|j|}|jPd|_|ss0    $z Assembler.getIterator[Data]ccs:|j2|jr td|jrtd|j}g|_tdt|_ |j r,|j dd|_Wdn1s9wY|D]}|Vq@|j }dur[|V|j }dusO|j5d|_|jritd|j spJ|j |jgks|Jd|_ |j rJ|jWddS1swYdS)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. Raises: EOFError: If the stream of frames has ended. ConcurrencyError: If two threads run :meth:`get` or :meth:`get_iter` concurrently. r"r#zqueue.SimpleQueue[Data | None]NTF)rrr'rrrrqueue SimpleQueuerrr*putr1r+rr-)rrchunkrrrget_iter|s@   "zAssembler.get_iterframer cCs|j|jr td|jrtd|jturtdd|_n|jt ur'd|_n|jt us.J|jdur=|j |j |j }n|j }|jdurL|j|n|j||j s^ WddS|jreJ|j|jduru|jd|jr|Jd|_Wdn1swY|j|j"d|_|jrtd|jsJ|jd|_WddS1swYdS)a, Add ``frame`` to the next message. When ``frame`` is the final frame in a message, :meth:`put` waits until the message is fetched, which can be achieved by calling :meth:`get` or by fully consuming the return value of :meth:`get_iter`. :meth:`put` assumes that the stream of frames respects the protocol. If it doesn't, the behavior is undefined. Raises: EOFError: If the stream of frames has ended. ConcurrencyError: If two threads run :meth:`put` concurrently. r"zput is already runningstrict)errorsNTF)rrr'rropcoder UTF8Decoderrrrdecodedatafinrrappendr5rr*r-rr(r+)rr8r>rrrr5sF         * "z Assembler.putcCs|j;|jr WddSd|_|jr&|j|jdur&|jd|jr6|jWddSWddS1sAwYdS)z End the stream of frames. Callling :meth:`close` concurrently with :meth:`get`, :meth:`get_iter`, or :meth:`put` is safe. They will raise :exc:`EOFError`. NT) rrrrr-rr5rrrrrrcloses     "zAssembler.close)r r)N)r r!r r )r r2)r8r r r) __name__ __module__ __qualname____doc__rr1r7r5rArrrrr s  & > AH) __future__rcodecsr3rtypingrr exceptionsrframesrrr r r __all__getincrementaldecoderr<r rrrrs