�
<��g� � �8 � d dl mZ d dlmZ G d� d� � ZdS )� )� annotations)� Generatorc �R � e Zd ZdZdd�Zdd�Zdd
�Zdd �Zdd
�Zdd�Z dd�Z
dd�Z dS )� StreamReaderz�
Generator-based stream reader.
This class doesn't support concurrent calls to :meth:`read_line`,
:meth:`read_exact`, or :meth:`read_to_eof`. Make sure calls are
serialized.
�return�Nonec �: � t � � | _ d| _ d S )NF)� bytearray�buffer�eof��selfs �b/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/websockets/streams.py�__init__zStreamReader.__init__ s � � �k�k�� ������ �m�int� Generator[None, None, bytes]c # �f K � d}d} | j � d|� � dz }|dk rnOt | j � � }||k rt d|� d|� d�� � �| j rt d|� d �� � �d
V � �t||k rt d|� d|� d�� � �| j d
|� }| j d
|�= |S ) a�
Read a LF-terminated line from the stream.
This is a generator-based coroutine.
The return value includes the LF character.
Args:
m: Maximum number bytes to read; this is a security limit.
Raises:
EOFError: If the stream ends without a LF.
RuntimeError: If the stream ends in more than ``m`` bytes.
r T�
� �read � bytes, expected no more than � bytes�stream ends after z bytes, before end of lineN)r �find�len� RuntimeErrorr �EOFError)r r �n�p�rs r � read_linezStreamReader.read_line s� � � � �
��
�� �� � � ���*�*�Q�.�A��1�u�u���D�K� � �A��1�u�u�"�#U�1�#U�#U�A�#U�#U�#U�V�V�V��x�
S� � Q�A� Q� Q� Q�R�R�R� �E�E�E� �
�q�5�5� � Q�q� Q� Q�� Q� Q� Q�R�R� R� �K����O�� �K����O��r r c # � K � |dk sJ �t | j � � |k rM| j r*t | j � � }t d|� d|� d�� � �dV � t | j � � |k �M| j d|� }| j d|�= |S )z�
Read a given number of bytes from the stream.
This is a generator-based coroutine.
Args:
n: How many bytes to read.
Raises:
EOFError: If the stream ends in less than ``n`` bytes.
r r z bytes, expected r N)r r r r )r r r! r" s r �
read_exactzStreamReader.read_exact6 s� � � � � �A�v�v�v�v��$�+� � ��"�"��x�
S��� �$�$�� � Q�A� Q� Q�� Q� Q� Q�R�R�R� �E�E�E� �$�+� � ��"�"�
�K����O�� �K����O��r c # �� K � | j s;t | j � � }||k rt d|� d|� d�� � �dV � | j �;| j dd� }| j dd�= |S )a
Read all bytes from the stream.
This is a generator-based coroutine.
Args:
m: Maximum number bytes to read; this is a security limit.
Raises:
RuntimeError: If the stream ends in more than ``m`` bytes.
r r r N)r r r r )r r r! r" s r � read_to_eofzStreamReader.read_to_eofM s� � � � � �(� ��D�K� � �A��1�u�u�"�#U�1�#U�#U�A�#U�#U�#U�V�V�V� �E�E�E� �(� �
�K����N�� �K����N��r �Generator[None, None, bool]c # �6 K � | j rdS | j rdS dV � �)zy
Tell whether the stream has ended and all data was read.
This is a generator-based coroutine.
TFN)r r r
s r �at_eofzStreamReader.at_eofc s9 � � � � ��{�
� �u��x�
��t�
�E�E�E� r �data�bytesc �R � | j rt d� � �| xj |z
c_ dS )z�
Write data to the stream.
:meth:`feed_data` cannot be called after :meth:`feed_eof`.
Args:
data: Data to write.
Raises:
EOFError: If the stream has ended.
� stream endedN)r r r )r r+ s r � feed_datazStreamReader.feed_datas s0 � � �8� +��>�*�*� *� � � �t�� � � � r c �@ � | j rt d� � �d| _ dS )z�
End the stream.
:meth:`feed_eof` cannot be called more than once.
Raises:
EOFError: If the stream has ended.
r. TN)r r r
s r �feed_eofzStreamReader.feed_eof� s' � � �8� +��>�*�*� *�����r c � � | j dd�= dS )zG
Discard all buffered data, but don't end the stream.
N)r r
s r �discardzStreamReader.discard� s � �
�K����N�N�Nr N)r r )r r r r )r r r r )r r( )r+ r, r r ) �__name__�
__module__� __qualname__�__doc__r r# r% r' r* r/ r1 r3 � r r r r s� � � � � � �� �� � � � � � � �D� � � �.� � � �,� � � � � � � �" � � � � � � � � � r r N)�
__future__r �collections.abcr r r8 r r �<module>r; se �� "� "� "� "� "� "� %� %� %� %� %� %�Q� Q� Q� Q� Q� Q� Q� Q� Q� Qr