File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/client_proto.cpython-310.pyc
o
;jY1 � @ s� d dl Z d dlmZ d dlmZmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZmZmZmZmZmZ ddlmZmZ dd lmZ dd
lmZm Z m Z G d d � d ee e ee f �Z dS )
� N)�suppress)�Any�Callable� )� BaseProtocol)�ClientConnectionError�
ClientOSError�ClientPayloadError�ServerDisconnectedError�SocketTimeoutError)�
_EXC_SENTINEL�DEFAULT_CHUNK_SIZE�EMPTY_BODY_STATUS_CODES�BaseTimerContext�
set_exception�
set_result)�HttpResponseParser�RawResponseMessage)�HttpProcessingError)�
EMPTY_PAYLOAD� DataQueue� StreamReaderc s� e Zd ZdZdejddfdd�Zeddejd B fdd��Z ede
fd d
��Z ede
fd d ��Z dGd
d�Z
dGdd�ZdGdd�Zde
fdd�ZdedB ddf� fdd� ZdGdd�ZdG� fdd� ZdHd e
ddf� fd d �
Zefded eddf� fd!d"�
Z dId#ed$ed%eg df dB ddfd&d'�Zdd(d(d ded)d*d*d+d,�
d-e dB d.e
d/e
d0e
d1e dB d2e d3e d4e d5e d6e ddfd7d8�Z dGd9d:�Z dGd;d<�Z!dGd=d>�Z"ede dB fd?d@��Z#e#j$d1e dB ddfdAd@��Z#dGdBdC�Z%dDe&ddfdEdF�Z'� Z(S )J�ResponseHandlerz8Helper class to adapt between Protocol and StreamReader.�loop�returnNc C sh t j| |d d� t�| |� d| _d | _d| _d | _d | _d | _d| _ d | _
d | _ d| _ d | _
d| _d S )N)r �parserF� � )r �__init__r �
_should_close�_payload�
_skip_payload�_payload_parser�_data_received_cbZ_timer�_tail�
_read_timeout�_read_timeout_handle�_timeout_ceil_threshold�_closed�_connection_lost_called)�selfr � r+ �H/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/client_proto.pyr s
zResponseHandler.__init__c C s" | j du r| js| j�� | _ | j S )a6 Future that is set when the connection is closed.
This property returns a Future that will be completed when the connection
is closed. The Future is created lazily on first access to avoid creating
futures that will never be awaited.
Returns:
- A Future[None] if the connection is still open or was closed after
this property was accessed
- None if connection_lost() was already called before this property
was ever accessed (indicating no one is waiting for the closure)
N)r( r) �_loopZ
create_future�r* r+ r+ r, �closed3 s zResponseHandler.closedc C � | j S �N)� _upgradedr. r+ r+ r, �upgradedE � zResponseHandler.upgradedc C sF t | jp!| jd uo| j�� p!| jp!| jd up!| jd up!| jp!| j�S r1 ) �boolr r �is_eofr2 �
_exceptionr" �_bufferr$ r. r+ r+ r, � should_closeI s ������z ResponseHandler.should_closec C s
d| _ d S �NT)r r. r+ r+ r, � force_closeU s
zResponseHandler.force_closec C �8 d | _ | j}|d ur|�� d | _d | _| �� d S d S r1 )r7 � transport�closer �
_drop_timeout�r* r= r+ r+ r, r> X � �zResponseHandler.closec C r< r1 )r7 r= �abortr r? r@ r+ r+ r, rB a rA zResponseHandler.abortc C s | j d uo
| j �� S r1 )r= Z
is_closingr. r+ r+ r, � is_connectedj s z ResponseHandler.is_connected�excc
s� d| _ | �� |}|}|d u }| jd ur/|r t| jd � n|d us#J �t| jtd|���|� | jd urMtt��
| j� � W d � n1 sHw Y d }| j
d ur�z| j
� � }W n. ty� } z"| j d urd|��}|sv|�d|��}t| j t |�|� W Y d }~nd }~ww | �
� s�t|t�r�t|j� }|r�t|�}|r�tn|}|d us�J �|d us�J �| �||� d| _d | _
d | _ d | _d| _t� �|� d S )NTzConnection lost: z#Response payload is not completed: z. F)r) r? r( r r r r" r � ExceptionZfeed_eof�_parserr r r6 �
isinstance�OSErrorr �argsr
r r Z_reading_paused�super�connection_lost) r* rD Zoriginal_connection_errorZ reraised_excZconnection_closed_cleanlyZ uncompleted�underlying_excZclient_payload_exc_msgZunderlying_non_eof_exc�� __class__r+ r, rK m st
��
�
������
�� z ResponseHandler.connection_lostc C � | � � d S r1 )r? r. r+ r+ r, � eof_received� s z ResponseHandler.eof_receivedc s t � �� | �� d S r1 )rJ �
pause_readingr? r. rM r+ r, rQ � s
z ResponseHandler.pause_readingT�
resume_parserc s t � �|� | �� d S r1 )rJ �resume_reading�_reschedule_timeout)r* rR rM r+ r, rS � s z ResponseHandler.resume_reading� exc_causec s d| _ | �� t� �||� d S r: )r r? rJ r )r* rD rU rM r+ r, r � s z ResponseHandler.set_exceptionr �payload�data_received_cbc C s@ || _ || _|| _| �� | jr | jd}| _| �|� d S d S )Nr )r r" r# r? r$ �
data_received)r* r rV rW �datar+ r+ r, �
set_parser� s �zResponseHandler.set_parserFr i� � )
�timer� skip_payload�read_until_eof�auto_decompress� read_timeout� read_bufsize�timeout_ceil_threshold�
max_line_size�max_field_size� max_headersr\ r] r^ r_ r` ra rb rc rd re c
C s\ || _ || _|| _t| | j||t| |||| |
d� | _| jr,| jd} | _| �| � d S d S )N)r\ Zpayload_exceptionZresponse_with_bodyr^ r_ rc rd re r ) r! r% r' r r- r rF r$ rX ) r* r\ r] r^ r_ r` ra rb rc rd re rY r+ r+ r, �set_response_params� s( ��z#ResponseHandler.set_response_paramsc C s" | j d ur| j �� d | _ d S d S r1 )r& �cancelr. r+ r+ r, r? s
�z ResponseHandler._drop_timeoutc C s> | j }| jd ur
| j�� |r| j�|| j�| _d S d | _d S r1 )r% r& rg r- Z
call_later�_on_read_timeout)r* �timeoutr+ r+ r, rT
s
�
z#ResponseHandler._reschedule_timeoutc C rO r1 )rT r. r+ r+ r, �
start_timeout s z ResponseHandler.start_timeoutc C r0 r1 �r% r. r+ r+ r, r` r4 z ResponseHandler.read_timeoutc C s
|| _ d S r1 rk )r* r` r+ r+ r, r` s
c C s0 t d�}| �|� | jd urt| j|� d S d S )Nz#Timeout on reading data from socket)r r r )r* rD r+ r+ r, rh $ s
�z ResponseHandler._on_read_timeoutrY c
C s� |r| � � | jd ur-| jd ur| �� | j�|�\}}|r+d | _d | _|r+| �|� d S | js5| jd u r>| j|7 _d S z | j�|�\}}}W n< t y� } z0| j
d ur[| j
� � t |t
�sa� t |t�rqt|j|j|jd�}nt� }| �||� W Y d }~d S d }~ww || _d }|D ]&\} }| jr�d| _|| _| js�| jtv r�| �| tfd� q�| �| |fd� q�|d ur�|tur�|�| j� n| �� |r�|r�| �|� d S d S d S )N)�code�message�headersTr )rT r" r# Z feed_datar rX r2 rF r$ �
BaseExceptionr= r> rG rE r rl rm rn r r9 r r! r r Zon_eofr? )
r* rY �eof�tail�messagesr3 rL rD rV rm r+ r+ r, rX * sb
� �� �z ResponseHandler.data_received)r N)Tr1 ))�__name__�
__module__� __qualname__�__doc__�asyncioZAbstractEventLoopr �propertyZFuturer/ r5 r3 r9 r; r> rB rC ro rK rP rQ rS r r r r rZ r
r �float�intrf r? rT rj r` �setterrh �bytesrX �
__classcell__r+ r+ rM r, r s�
K����
����
�������� �
� � �
�
&
r ) rw �
contextlibr �typingr r Z
base_protocolr Zclient_exceptionsr r r r
r Zhelpersr r
r r r r �httpr r Zhttp_exceptionsr Zstreamsr r r �tupler r+ r+ r+ r, �<module> s "