o >h @sdZddlmZddlmZmZmZmZddlm Z m Z ddl m Z m Z mZmZedZGddde ZGd d d e Ze Gd d d Zd ededeefddZde deedefddZdS)z Buffer byte streams. )count)DictIteratorListTypeVar)Factorydefine)AMPCommandIntegerStringTc@seZdZdZdefgZdS) StreamOpenz Open a new stream. streamIdN)__name__ __module__ __qualname____doc__r responserrz/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/_dist/stream.pyrsrc@s$eZdZdZdefdefgZdS) StreamWritez, Write a chunk of data to a stream. rsdataN)rrrrr Bytes argumentsrrrrrs rc@szeZdZUdZeZeeed<e e Z e ee efed<defddZdededd fd d Zdede efd d Zd S)StreamReceiverz9 Buffering de-multiplexing byte stream receiver. _counter_streamsreturncCst|j}g|j|<|S)zE Open a new stream and return its unique identifier. )nextrr)selfnewIdrrropen+s  zStreamReceiver.openstreamIdchunkNcCs|j||dS)z Write to an open stream using its unique identifier. @raise KeyError: If there is no such open stream. N)rappend)rr"r#rrrwrite3szStreamReceiver.writecCs |j|S)z Indicate an open stream may receive no further data and return all of its current contents. @raise KeyError: If there is no such open stream. )rpop)rr"rrrfinish;s zStreamReceiver.finish)rrrrrrrint__annotations__rdictrrrbytesr!r%r'rrrrr"s rdata chunkSizerccs@d}|t|kr||||V||7}|t|ks dSdS)a Break a byte string into pieces of no more than ``chunkSize`` length. @param data: The byte string. @param chunkSize: The maximum length of the resulting pieces. All pieces except possibly the last will be this length. @return: The pieces. rN)len)r,r-posrrrr#Es  r#ampchunkscsH|tIdHd}t|tsJ|D] }|jt||dIdHq|S)z Send the given stream chunks, one by one, over the given connection. The chunks are sent using L{StreamWrite} over a stream opened using L{StreamOpen}. @return: The identifier of the stream over which the chunks were sent. Nr")r"r,) callRemoter isinstancer(r)r0r1r"oneChunkrrrstreamVs  r5N)r itertoolsrtypingrrrrattrsrrtwisted.protocols.ampr r r r rr rrrr+r(r#r5rrrrs  "