| Current File : //proc/self/root/lib/python3/dist-packages/twisted/trial/_dist/__pycache__/stream.cpython-312.pyc |
�
Ϫ�f� � �� � d Z ddlmZ ddlmZmZmZmZ ddlm Z m
Z
ddl m Z m
Z
mZmZ ed� Z G d� de
� Z G d � d
e
� Ze
G d � d � � Zd
ededee fd�Zde dee defd�Zy)z
Buffer byte streams.
� )�count)�Dict�Iterator�List�TypeVar)�Factory�define)�AMP�Command�Integer�String�Tc �$ � e Zd ZdZd e� fgZy)�
StreamOpenz
Open a new stream.
� streamIdN)�__name__�
__module__� __qualname__�__doc__r �response� � �</usr/lib/python3/dist-packages/twisted/trial/_dist/stream.pyr r s � �� �g�i�(�)�Hr r c �4 � e Zd ZdZd e� fd e� fgZy)� StreamWritez,
Write a chunk of data to a stream.
r s dataN)r r r r r �Bytes� argumentsr r r r r s$ � ��
�g�i� � �%�'���Ir r c � � e Zd ZU dZ e� Zee ed< e e
� Z e ee
e f ed< defd�Zdededdfd �Zdede
e fd
�Zy) �StreamReceiverz9
Buffering de-multiplexing byte stream receiver.
�_counter�_streams�returnc �N � t | j � }g | j |< |S )zE
Open a new stream and return its unique identifier.
)�nextr r! )�self�newIds r �openzStreamReceiver.open+ s% � � �T�]�]�#�� !��
�
�e� �� r �streamId�chunkNc �@ � | j | j |� y)z�
Write to an open stream using its unique identifier.
@raise KeyError: If there is no such open stream.
N)r! �append)r% r( r) s r �writezStreamReceiver.write3 s � �
�
�
�h� �&�&�u�-r c �8 � | j 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.
)r! �pop)r% r( s r �finishzStreamReceiver.finish; s � � �}�}� � ��*�*r )r r r r r r r �int�__annotations__r �dictr! r r �bytesr' r, r/ r r r r r " sr � �� $�g�H�h�s�m�%�'.�t�}�H�d�3��U� �#�$�4��c� �.�c� .�%� .�D� .�+�s� +�t�E�{� +r r �data� chunkSizer" c # �j K � d}|t | � k r | |||z �� ||z
}|t | � k r� yy�w)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.
r N)�len)r4 r5 �poss r r) r) E s@ � �� � �C�
��D� �/��3��y� �)�)� �y��� ��D� �/�s �.3�3�amp�chunksc � �� K � | j t � � d{ ��� d }t |t � sJ �|D ]" }| j t ||�� � d{ ��� �$ |S 7 �B7 �
�w)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( r4 )�
callRemoter �
isinstancer0 r )r9 r: r( �oneChunks r �streamr? V sh � �� � �n�n�Z�0�0�*�=�H� �h�� $�$� $�� L���n�n�[�8�(�n�K�K�K�L� �O� 1�� L�s! �A"�A �9A"�A � A"� A"N)r � itertoolsr �typingr r r r �attrsr r �twisted.protocols.ampr
r r r
r r r r r r3 r0 r) r? r r r �<module>rD s� ��� � 0� 0� !� H� H� �C�L��*�� *��'� � � +� +� � +�D�� �#� �(�5�/� �"�c� �8�E�?� �s� r