File: //opt/saltstack/salt/lib/python3.10/site-packages/cheroot/__pycache__/makefile.cpython-310.pyc
o
;j � @ s` d Z ddlZddlZdZG dd� dej�ZG dd� dej�ZG dd � d e�Zd
ej fd d �Z
dS )
zSocket file object.� Ni @ c @ s e Zd ZdZdd� Zdd� ZdS )�BufferedWriterz-Faux file object attached to a socket object.c C s` | � � t|t�r
td��| j� | j�|� | �� t|�W d � S 1 s)w Y dS )zWrite bytes to buffer.z can't write str to binary streamN) � _checkClosed�
isinstance�str� TypeErrorZ _write_lock�
_write_buf�extend�_flush_unlocked�len)�self�b� r
�D/opt/saltstack/salt/lib/python3.10/site-packages/cheroot/makefile.py�write s
$�zBufferedWriter.writec
C sn | � d� | jr5z | j�t| j��}W n tjy) } z |j}W Y d }~nd }~ww | jd |�= | jsd S d S )Nzflush of closed file)r r �rawr �bytes�io�BlockingIOError�characters_written)r �n�er
r
r r s
�� �z BufferedWriter._flush_unlockedN)�__name__�
__module__� __qualname__�__doc__r r r
r
r
r r s r c s< e Zd ZdZdejf� fdd� Z� fdd�Zdd� Z� Z S ) � StreamReaderzSocket stream reader.�rc � t � �t�||�|� d| _dS )z Initialize socket stream reader.r N)�super�__init__�socket�SocketIO�
bytes_read�r �sock�mode�bufsize�� __class__r
r r ) �
zStreamReader.__init__c s( t � j|i |��}| jt|�7 _|S )zCapture bytes read.)r �readr" r
)r �args�kwargs�valr' r
r r* . s zStreamReader.readc C s t | j�| jkS )z.Return true if there is buffered data to read.)r
Z _read_bufZ _read_pos)r r
r
r �has_data4 s zStreamReader.has_data)
r r r r r �DEFAULT_BUFFER_SIZEr r* r. �
__classcell__r
r
r' r r & s
r c s4 e Zd ZdZdejf� fdd� Z� fdd�Z� ZS )� StreamWriterzSocket stream writer.�wc r )z Initialize socket stream writer.r N)r r r r! �
bytes_writtenr# r' r
r r < r) zStreamWriter.__init__c s0 t � j|g|�R i |��}| jt|�7 _|S )zCapture bytes written.)r r r3 r
)r r- r+ r, �resr' r
r r A s zStreamWriter.write) r r r r r r/ r r r0 r
r
r' r r1 9 s r1 r c C s d|v rt nt}|| ||�S )z(File object attached to a socket object.r )r r1 )r$ r% r& �clsr
r
r �MakeFileH s r6 ) r Z_pyior r ZSOCK_WRITE_BLOCKSIZEr �BufferedReaderr r1 r/ r6 r
r
r
r �<module> s