o i @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 responserr`C:\Users\psant\Downloads\PAF backend\school\venv\Lib\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. N)nextrr)selfZnewIdrrropen+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. N)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+)Z callRemoter isinstancer'r)r/r0r!ZoneChunkrrrstreamVs  r2N)r itertoolsrtypingrrrrattrsrrZtwisted.protocols.ampr r r r rr rrrr*r'r"r2rrrrs  "