File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/client_proto.cpython-310.pyc
o
���g ' � @ s� d dl Z d dlmZ d dlmZmZmZ ddlmZ ddl m
Z
m Z m Z m
Z
ddlmZmZmZmZ ddlmZmZ dd lmZ dd
lmZmZmZ G d d � d eeeeef �Z dS )
� N)�suppress)�Any�Optional�Tuple� )� BaseProtocol)�
ClientOSError�ClientPayloadError�ServerDisconnectedError�SocketTimeoutError)�
_EXC_SENTINEL�EMPTY_BODY_STATUS_CODES�BaseTimerContext�
set_exception)�HttpResponseParser�RawResponseMessage)�HttpProcessingError)�
EMPTY_PAYLOAD� DataQueue� StreamReaderc s� e Zd ZdZdejddfdd�Zedefdd��Z edefd d
��Z
d@d d �Z d@d
d�Z defdd�Z
dee ddf� fdd� Zd@dd�Zd@� fdd� Zd@� fdd� Zefdededdf� fdd �
Zd ed eddfd d �Zdd!d!d"dd#d$d%d%d&� d'ee d(ed)ed*ed+ee d,ed-ed.ed/eddfd0d1�Zd@d2d3�Z d@d4d5�Z d@d6d7�Z edee fd8d9��Z e j d+ee ddfd:d9��Z d@d;d<�Z!d=e"ddfd>d?�Z#� Z$S )A�ResponseHandlerz8Helper class to adapt between Protocol and StreamReader.�loop�returnNc C s` t j| |d� t�| |� d| _d | _d| _d | _d | _d| _d| _ d | _
d | _ d | _ d| _
d S )N)r F� � )r �__init__r �
_should_close�_payload�
_skip_payload�_payload_parser�_timer�_tail� _upgraded�_parser�
_read_timeout�_read_timeout_handle�_timeout_ceil_threshold)�selfr � r( �?/usr/local/lib/python3.10/dist-packages/aiohttp/client_proto.pyr s
zResponseHandler.__init__c C � | j S �N)r" �r' r( r( r) �upgraded/ � zResponseHandler.upgradedc C sF t | jp!| jd uo| j�� p!| jp!| jd up!| jd up!| jp!| j�S r+ ) �boolr r �is_eofr" �
_exceptionr �_bufferr! r, r( r( r) � should_close3 s ������z ResponseHandler.should_closec C s
d| _ d S �NT)r r, r( r( r) � force_close? s
zResponseHandler.force_closec C s2 | j }|d ur|�� d | _ d | _| �� d S d S r+ )� transport�closer �
_drop_timeout)r' r6 r( r( r) r7 B s �zResponseHandler.closec C s | j d uo
| j �� S r+ )r6 �
is_closingr, r( r( r) � is_connectedJ s z ResponseHandler.is_connected�excc
sT | � � |}|}|d u }| jd ur*tt��
| j�� W d � n1 s%w Y d }| jd urgz| j�� }W n. tyf } z"| jd ur\d|��}|sS|�d|��}t| jt|�|� W Y d }~nd }~ww | � � s�t
|t �rut |j
� }|r{t|�}|rtn|}|d us�J �|d us�J �| �||� d| _d | _d | _d | _d| _t� �|� d S )Nz#Response payload is not completed: z. TF)r8 r r � Exception�feed_eofr# r r r r0 �
isinstance�OSErrorr �argsr
r r �_reading_paused�super�connection_lost) r' r; �original_connection_error� reraised_exc�connection_closed_cleanly� uncompleted�underlying_exc�client_payload_exc_msg�underlying_non_eof_exc�� __class__r( r) rC M s\
�
������
�� z ResponseHandler.connection_lostc C � | � � d S r+ )r8 r, r( r( r) � eof_received� s z ResponseHandler.eof_receivedc � t � �� | �� d S r+ )rB �
pause_readingr8 r, rK r( r) rP � �
z ResponseHandler.pause_readingc rO r+ )rB �resume_reading�_reschedule_timeoutr, rK r( r) rR � rQ z ResponseHandler.resume_reading� exc_causec s d| _ | �� t� �||� d S r4 )r r8 rB r )r' r; rT rK r( r) r � s z ResponseHandler.set_exception�parser�payloadc C s: || _ || _| �� | jr| jd}| _| �|� d S d S )Nr )r r r8 r! �
data_received)r' rU rV �datar( r( r) �
set_parser� s �zResponseHandler.set_parserFTi r i� ) �timer� skip_payload�read_until_eof�auto_decompress� read_timeout� read_bufsize�timeout_ceil_threshold�
max_line_size�max_field_sizerZ r[ r\ r] r^ r_ r` ra rb c C sZ || _ || _|| _t| | j||t| |||| d�
| _| jr+| jd}
| _| �|
� d S d S )N)rZ �payload_exception�response_with_bodyr\ r] ra rb r ) r r$ r& r �_loopr r# r! rW ) r' rZ r[ r\ r] r^ r_ r` ra rb rX r( r( r) �set_response_params� s&
�
�z#ResponseHandler.set_response_paramsc C s" | j d ur| j �� d | _ d S d S r+ )r% �cancelr, r( r( r) r8 � s
�z ResponseHandler._drop_timeoutc C s> | j }| jd ur
| j�� |r| j�|| j�| _d S d | _d S r+ )r$ r% rg re �
call_later�_on_read_timeout)r' �timeoutr( r( r) rS � s
�
z#ResponseHandler._reschedule_timeoutc C rM r+ )rS r, r( r( r) �
start_timeout� s z ResponseHandler.start_timeoutc C r* r+ �r$ r, r( r( r) r^ � r. z ResponseHandler.read_timeoutc C s
|| _ d S r+ rl )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' r; r( r( r) ri � s
�z ResponseHandler._on_read_timeoutrX c
C s� | � � |sd S | jd ur&| j�|�\}}|r$d | _d | _|r$| �|� d S | js.| jd u r7| j|7 _d S z | j�|�\}}}W n6 tyx } z*| j d urT| j �
� t |t �rdt |j
|j|jd�}nt � }| �||� W Y d }~d S d }~ww || _d }|D ]&\} }| jr�d| _|| _| js�| j
tv r�| �| tfd� q�| �| |fd� q�|d ur�|tur�|�| j� n| �� |r�|r�| �|� d S d S d S )N)�code�message�headersTr )rS r � feed_datar rW r" r# r! �
BaseExceptionr6 r7 r> r rm rn ro r r3 r r r
r �on_eofr8 )
r' rX �eof�tail�messagesr- rH r; rV rn r( r( r) rW � s\
� �� �z ResponseHandler.data_received)r N)%�__name__�
__module__� __qualname__�__doc__�asyncio�AbstractEventLoopr �propertyr/ r- r3 r5 r7 r: r rq rC rN rP rR r r r rY r �float�intrf r8 rS rk r^ �setterri �bytesrW �
__classcell__r( r( rK r) r s|
8���� ������� �
� �
�
$
r ) rz �
contextlibr �typingr r r �
base_protocolr �client_exceptionsr r r
r �helpersr r
r r �httpr r �http_exceptionsr �streamsr r r r r( r( r( r) �<module> s "