File: //usr/local/lib/python3.10/asyncio/__pycache__/streams.cpython-310.opt-2.pyc
o
�i�d � @ s& d Z ddlZddlZddlZddlZddlZeed�r e d7 Z ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd l
mZ dd lmZ d
Zd ed�dd�Zd ed�dd�Zeed�rpd ed�dd�Zd ed�dd�ZG dd� de j�ZG dd� dee j�ZG dd � d �ZG d d � d �ZdS )!)� StreamReader� StreamWriter�StreamReaderProtocol�open_connection� start_server� N�AF_UNIX)�open_unix_connection�start_unix_server� )�
coroutines)�events)�
exceptions)�format_helpers)� protocols)�logger)�sleepi )�limitc � sb � t �� }t||d�}t||d�� |j� fdd�| |fi |��I d H \}}t|� ||�}||fS )N�r �loop�r c � � S �N� r ��protocolr �,/usr/local/lib/python3.10/asyncio/streams.py�<lambda>1 � z!open_connection.<locals>.<lambda>)r �get_running_loopr r �create_connectionr ) �host�portr �kwdsr �reader� transport�_�writerr r r r s � ��r c � s8 � t �� �� ��fdd�}�j|||fi |��I d H S )Nc � t ��d�} t| � �d�}|S �Nr r �r r �r# r ��client_connected_cbr r r r �factoryO �
�z start_server.<locals>.factory)r r �
create_server)r, r r! r r" r- r r+ r r 6 s
� r c � s` � t �� }t||d�}t||d�� |j� fdd�| fi |��I d H \}}t|� ||�}||fS )Nr r c r r r r r r r r c r z&open_unix_connection.<locals>.<lambda>)r r r r Zcreate_unix_connectionr )�pathr r" r r# r$ r% r& r r r r [ s � ��r c � s6 � t �� �� ��fdd�}�j||fi |��I d H S )Nc r' r( r) r* r+ r r r- l r. z"start_unix_server.<locals>.factory)r r Zcreate_unix_server)r, r0 r r" r- r r+ r r g s
�r c @ s@ e Zd Z ddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�FlowControlMixinNc C s8 |d u r t jdd�| _n|| _d| _t�� | _d| _d S )N� )�
stacklevelF)r �_get_event_loop�_loop�_paused� collections�deque�_drain_waiters�_connection_lost)�selfr r r r �__init__ s
zFlowControlMixin.__init__c C s$ d| _ | j�� rt�d| � d S d S )NTz%r pauses writing)r6 r5 � get_debugr �debug�r; r r r �
pause_writing� s
�z FlowControlMixin.pause_writingc C s> d| _ | j�� rt�d| � | jD ] }|�� s |�d � qd S )NFz%r resumes writing)r6 r5 r= r r> r9 �done�
set_result�r; �waiterr r r �resume_writing� s
��z FlowControlMixin.resume_writingc C sF d| _ | jsd S | jD ]}|�� s |d u r|�d � q |�|� q d S �NT)r: r6 r9 rA rB �
set_exception�r; �excrD r r r �connection_lost� s
��z FlowControlMixin.connection_lostc � s\ �| j rtd��| js
d S | j�� }| j�|� z|I d H W | j�|� d S | j�|� w )NzConnection lost)r: �ConnectionResetErrorr6 r5 �
create_futurer9 �append�removerC r r r �
_drain_helper� s �
z FlowControlMixin._drain_helperc C s t �r )�NotImplementedError�r; �streamr r r �_get_close_waiter� � z"FlowControlMixin._get_close_waiterr ) �__name__�
__module__� __qualname__r< r@ rE rJ rO rS r r r r r1 u s
r1 c sd e Zd Z dZd� fdd� Zedd� �Zdd� Z� fdd �Zd
d � Z d d
� Z
dd� Z dd� Z � Z
S )r Nc st t � j|d� |d urt�|�| _|j| _nd | _|d ur || _d| _d | _d | _ d | _
|| _ d| _ | j
�� | _d S )Nr F)�superr< �weakref�ref�_stream_reader_wr�_source_traceback�_strong_reader�_reject_connection�_stream_writer�_task�
_transport�_client_connected_cb� _over_sslr5 rL �_closed)r; Z
stream_readerr, r �� __class__r r r< � s
z StreamReaderProtocol.__init__c C s | j d u rd S | � � S r )r[ r? r r r �_stream_reader� s
z#StreamReaderProtocol._stream_readerc C s� | j rddi}| jr| j|d< | j�|� |�� d S || _| j}|d ur*|�|� |�d�d u| _ | j
d urXt || || j�| _ | �
|| j �}t
�|�rS| j�|�| _d | _d S d S )N�messagezpAn open stream was garbage collected prior to establishing network connection; call "stream.close()" explicitly.Zsource_tracebackZ
sslcontext)r^ r\ r5 Zcall_exception_handler�abortra rg �
set_transport�get_extra_inforc rb r r_ r Z iscoroutineZ create_taskr` r] )r; r$ �contextr# �resr r r �connection_made� s2 �
��
�z$StreamReaderProtocol.connection_madec s~ | j }|d ur|d u r|�� n|�|� | j�� s+|d u r%| j�d � n| j�|� t� �|� d | _d | _ d | _
d | _ d S r ) rg �feed_eofrG rd rA rB rX rJ r[ r_ r` ra )r; rI r# re r r rJ � s
z$StreamReaderProtocol.connection_lostc C s | j }|d ur|�|� d S d S r )rg � feed_data)r; �datar# r r r �
data_received s �z"StreamReaderProtocol.data_receivedc C s$ | j }|d ur |�� | jrdS dS )NFT)rg ro rc )r; r# r r r � eof_received s z!StreamReaderProtocol.eof_receivedc C � | j S r )rd rQ r r r rS � z&StreamReaderProtocol._get_close_waiterc C sD z| j }W n
ty Y d S w |�� r |�� s |�� d S d S d S r )rd �AttributeErrorrA � cancelled� exception)r; �closedr r r �__del__ s
� �z StreamReaderProtocol.__del__�NN)rU rV rW r\ r< �propertyrg rn rJ rr rs rS rz �
__classcell__r r re r r � s
r c @ st e Zd Z dd� Zdd� Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
ddd�Zdd� ZdS )r c C s4 || _ || _|| _|| _| j�� | _| j�d � d S r )ra � _protocol�_readerr5 rL Z
_complete_futrB )r; r$ r r# r r r r r< 0 s zStreamWriter.__init__c C s@ | j jd| j��g}| jd ur|�d| j��� d�d�|��S )N�
transport=zreader=�<{}>� )rf rU ra r rM �format�join�r; �infor r r �__repr__: s
zStreamWriter.__repr__c C rt r �ra r? r r r r$ @ s zStreamWriter.transportc C � | j �|� d S r )ra �write�r; rq r r r r� D � zStreamWriter.writec C r� r )ra �
writelinesr� r r r r� G r� zStreamWriter.writelinesc C �
| j �� S r )ra � write_eofr? r r r r� J �
zStreamWriter.write_eofc C r� r )ra �
can_write_eofr? r r r r� M r� zStreamWriter.can_write_eofc C r� r )ra �closer? r r r r� P r� zStreamWriter.closec C r� r )ra �
is_closingr? r r r r� S r� zStreamWriter.is_closingc � s �| j �| �I d H d S r )r~ rS r? r r r � wait_closedV s �zStreamWriter.wait_closedNc C s | j �||�S r )ra rk )r; �name�defaultr r r rk Y s zStreamWriter.get_extra_infoc � sP � | j d ur| j �� }|d ur|�| j�� r td�I d H | j�� I d H d S )Nr )r rx ra r� r r~ rO )r; rI r r r �drain\ s �
zStreamWriter.drainr )rU rV rW r< r� r| r$ r� r� r� r� r� r� r� rk r� r r r r r & s
r c @ s� e Zd ZdZedfdd�Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd&dd �Zd'd d �Zd d!� Zd"d#� Zd$d%� ZdS )(r Nc C sz |dkrt d��|| _|d u rt�� | _n|| _t� | _d| _d | _d | _ d | _
d| _ | j� � r;t
�t�d��| _d S d S )Nr zLimit cannot be <= 0Fr
)�
ValueError�_limitr r4 r5 � bytearray�_buffer�_eof�_waiter�
_exceptionra r6 r= r �
extract_stack�sys� _getframer\ )r; r r r r r r< z s"
��zStreamReader.__init__c C s� dg}| j r|�t| j �� d�� | jr|�d� | jtkr'|�d| j� �� | jr3|�d| j��� | jr?|�d| j��� | jrK|�d| j��� | j rS|�d� d �
d
� |��S ) Nr z bytes�eofzlimit=zwaiter=z
exception=r� Zpausedr� r� ) r� rM �lenr� r� �_DEFAULT_LIMITr� r� ra r6 r� r� r� r r r r� � s
zStreamReader.__repr__c C rt r )r� r? r r r rx � ru zStreamReader.exceptionc C s8 || _ | j}|d urd | _|�� s|�|� d S d S d S r )r� r� rw rG rH r r r rG � s �zStreamReader.set_exceptionc C s4 | j }|d urd | _ |�� s|�d � d S d S d S r )r� rw rB rC r r r �_wakeup_waiter� s �zStreamReader._wakeup_waiterc C s
|| _ d S r r� )r; r$ r r r rj � s
zStreamReader.set_transportc C s2 | j rt| j�| jkrd| _ | j�� d S d S d S )NF)r6 r� r� r� ra �resume_readingr? r r r �_maybe_resume_transport� s �z$StreamReader._maybe_resume_transportc C s d| _ | �� d S rF )r� r� r? r r r ro � s zStreamReader.feed_eofc C s | j o| j S r )r� r� r? r r r �at_eof� s zStreamReader.at_eofc C s� |sd S | j �|� | �� | jd ur:| js<t| j �d| j kr>z| j�� W n
ty4 d | _Y d S w d| _d S d S d S d S )N� T) r� �extendr� ra r6 r� r� Z
pause_readingrP r� r r r rp � s
� �
�zStreamReader.feed_datac � s` � | j d urt|� d���| jrd| _| j�� | j�� | _ z | j I d H W d | _ d S d | _ w )NzF() called while another coroutine is already waiting for incoming dataF)r� � RuntimeErrorr6 ra r� r5 rL )r; � func_namer r r �_wait_for_data� s �
�
zStreamReader._wait_for_datac
� s� � d}t |�}z | �|�I d H }W |S tjy) } z |jW Y d }~S d }~w tjyW } z"| j�||j�rC| jd |j| �= n| j� � | �
� t |j d ��d }~ww )N�
r )
r� � readuntilr
�IncompleteReadError�partial�LimitOverrunErrorr� �
startswith�consumed�clearr� r� �args)r; �sep�seplen�line�er r r �readline� s$ �
��
��zStreamReader.readliner� c � s � t |�}|dkrtd��| jd ur| j�d} t | j�}|| |krA| j�||�}|dkr0n-|d | }|| jkrAt�d|��| jrTt | j�}| j�
� t� |d ��| � d�I d H q|| jkrht�d|��| jd || � }| jd || �= | �
� t |�S ) Nr z,Separator should be at least one-byte stringT���r
z2Separator is not found, and chunk exceed the limitr� z2Separator is found, but chunk is longer than limit)r� r� r� r� �findr� r
r� r� �bytesr� r� r� r� )r; � separatorr� �offset�buflenZisep�chunkr r r r� sD �
�
�
�zStreamReader.readuntilr� c � s� � | j d ur
| j �|dkrdS |dk r.g } | �| j�I d H }|s#n|�|� qd�|�S | js<| js<| �d�I d H t| jd |� �}| jd |�= | � � |S )Nr � T�read)
r� r� r� rM r� r� r� r� r� r� )r; �nZblocks�blockrq r r r r� s s* �
�
zStreamReader.readc � s� � |dk r
t d��| jd ur| j�|dkrdS t| j�|k rA| jr2t| j�}| j�� t�||��| � d�I d H t| j�|k s t| j�|krSt| j�}| j�� nt| jd |� �}| jd |�= | �
� |S )Nr z*readexactly size can not be less than zeror� � readexactly) r� r� r� r� r� r� r� r
r� r� r� )r; r� Z
incompleterq r r r r� � s, �
�
zStreamReader.readexactlyc C s | S r r r? r r r � __aiter__� rT zStreamReader.__aiter__c � s �| � � I d H }|dkrt�|S )Nr� )r� �StopAsyncIteration)r; �valr r r � __anext__� s
�zStreamReader.__anext__)r� )r� )rU rV rW r\ r� r< r� rx rG r� rj r� ro r� rp r� r� r� r� r� r� r� r r r r r v s&
[3) r r{ r )�__all__r7 �socketr� �warningsrY �hasattr� r r r
r r �logr Ztasksr r� r r r r ZProtocolr1 r r r r r r r �<module> sB
� �
" � �@qP