← Back
Editing: streams.cpython-310.opt-2.pyc
o 0�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� NZAF_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 �6/opt/alt/python310/lib64/python3.10/asyncio/streams.py�<lambda>1 � z!open_connection.<locals>.<lambda>)r �get_running_loopr r Zcreate_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 Z 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* r- 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)r3 r2 � get_debugr �debug�r8 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)r3 r2 r: r r; r6 �done� set_result�r8 �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)r7 r3 r6 r> r? � set_exception�r8 �excrA 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)r7 �ConnectionResetErrorr3 r2 � create_futurer6 �append�remover@ r r r � _drain_helper� s � z FlowControlMixin._drain_helperc C s t �r )�NotImplementedError�r8 �streamr r r �_get_close_waiter� � z"FlowControlMixin._get_close_waiterr ) �__name__� __module__� __qualname__r9 r= rB rG rL rP r r r r r. u s r. 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)�superr9 �weakref�ref�_stream_reader_wr�_source_traceback�_strong_reader�_reject_connection�_stream_writer�_task� _transport�_client_connected_cb� _over_sslr2 rI �_closed)r8 Z stream_readerr* r �� __class__r r r9 � s z StreamReaderProtocol.__init__c C s | j d u rd S | � � S r )rX 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[ rY r2 Zcall_exception_handler�abortr^ rd � set_transport�get_extra_infor` r_ r r\ r Z iscoroutineZ create_taskr] rZ )r8 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 ) rd �feed_eofrD ra r> r? rU rG rX r\ r] r^ )r8 rF r! rb r r rG � s z$StreamReaderProtocol.connection_lostc C s | j }|d ur|�|� d S d S r )rd � feed_data)r8 �datar! r r r � data_received s �z"StreamReaderProtocol.data_receivedc C s$ | j }|d ur |�� | jrdS dS )NFT)rd rl r` )r8 r! r r r � eof_received s z!StreamReaderProtocol.eof_receivedc C � | j S r )ra rN r r r rP � 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 )ra �AttributeErrorr> � cancelled� exception)r8 �closedr r r �__del__ s � �z StreamReaderProtocol.__del__�NN)rR rS rT rY r9 �propertyrd rk rG ro rp rP rw � __classcell__r r rb 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 )r^ � _protocol�_readerr2 rI Z _complete_futr? )r8 r" r r! r r r r r9 0 s zStreamWriter.__init__c C s@ | j jd| j��g}| jd ur|�d| j��� d�d�|��S )N� transport=zreader=�<{}>� )rc rR r^ r| rJ �format�join�r8 �infor r r �__repr__: s zStreamWriter.__repr__c C rq r �r^ r< r r r r" @ s zStreamWriter.transportc C � | j �|� d S r )r^ �write�r8 rn r r r r� D � zStreamWriter.writec C r� r )r^ � writelinesr� r r r r� G r� zStreamWriter.writelinesc C � | j �� S r )r^ � write_eofr< r r r r� J � zStreamWriter.write_eofc C r� r )r^ � can_write_eofr< r r r r� M r� zStreamWriter.can_write_eofc C r� r )r^ �closer< r r r r� P r� zStreamWriter.closec C r� r )r^ � is_closingr< r r r r� S r� zStreamWriter.is_closingc � s �| j �| �I d H d S r )r{ rP r< r r r � wait_closedV s �zStreamWriter.wait_closedNc C s | j �||�S r )r^ rh )r8 �name�defaultr r r rh 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| ru r^ r� r r{ rL )r8 rF r r r �drain\ s � zStreamWriter.drainr )rR rS rT r9 r� ry r" r� r� r� r� r� r� r� rh 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 r1 r2 � bytearray�_buffer�_eof�_waiter� _exceptionr^ r3 r: r � extract_stack�sys� _getframerY )r8 r r r r r r9 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� rJ �lenr� r� �_DEFAULT_LIMITr� r� r^ r3 r� r� r� r r r r� � s zStreamReader.__repr__c C rq r )r� r< r r r ru � rr zStreamReader.exceptionc C s8 || _ | j}|d urd | _|�� s|�|� d S d S d S r )r� r� rt rD rE r r r rD � s �zStreamReader.set_exceptionc C s4 | j }|d urd | _ |�� s|�d � d S d S d S r )r� rt r? r@ r r r �_wakeup_waiter� s �zStreamReader._wakeup_waiterc C s || _ d S r r� )r8 r" r r r rg � s zStreamReader.set_transportc C s2 | j rt| j�| jkrd| _ | j�� d S d S d S )NF)r3 r� r� r� r^ �resume_readingr< r r r �_maybe_resume_transport� s �z$StreamReader._maybe_resume_transportc C s d| _ | �� d S rC )r� r� r< r r r rl � 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� r^ r3 r� r� Z pause_readingrM r� r r r rm � 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� � RuntimeErrorr3 r^ r� r2 rI )r8 Z 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)r8 �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� )r8 Z separatorr� �offsetZbuflenZisep�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� rJ r� r� r� r� r� r� )r8 �nZblocks�blockrn 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� )r8 r� Z incompletern r r r r� � s, � � zStreamReader.readexactlyc C s | S r r r< r r r � __aiter__� rQ zStreamReader.__aiter__c � s �| � � I d H }|dkrt�|S )Nr� )r� �StopAsyncIteration)r8 �valr r r � __anext__� s �zStreamReader.__anext__)r� )r� )rR rS rT rY r� r9 r� ru rD r� rg r� rl r� rm r� r� r� r� r� r� r� r r r r r v s& [3) r rx r )�__all__r4 Zsocketr� �warningsrV �hasattr� r r r r r �logr Ztasksr r� r r r r ZProtocolr. r r r r r r r �<module> sB � � " � �@qP
Save File
Cancel