File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/client_ws.cpython-310.pyc
o
;j L � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZm Z m
Z
m Z m Z m
Z
ddlZddlmZ ddlmZmZmZ dd lmZ dd
lmZmZ dd lmZm Z m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$m%Z% dd
l&m'Z' ddl(m)Z)m*Z*m+Z+m,Z,m-Z- ej.dkr�ddlm/Z/ nddl0m/Z/ ej.dkr�ddlZ1ddlm2Z2 n
ddl1Z1ddl0m2Z2 e/de3de
d d�Z4ej5ddd�G dd� d��Z6e6ddd�Z7G dd � d e e4 �Z8dS ) z WebSocket client for asyncio.� N)�Callable)�
TracebackType)�Any�Generic�Literal�Optional�cast�overload� )�WebSocketDataQueue)� ClientError�ServerTimeoutError�WSMessageTypeError)�ClientResponse)�calculate_timeout_when�
set_result)�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WebSocketError� WSCloseCode� WSMessage�WSMessageDecodeText�WSMessageNoDecodeText� WSMsgType)�_INTERNAL_RECEIVE_TYPES�WebSocketWriter)� EofStream)�DEFAULT_JSON_DECODER�DEFAULT_JSON_ENCODER�JSONBytesEncoder� JSONDecoder� JSONEncoder)� �
)�TypeVar)r" � )�Self� _DecodeTextT)�bound� covariant�default)�frozen�slotsc @ s0 e Zd Zejee dd�Zejee dd�ZdS )�ClientWSTimeoutN)�typer* ) �__name__�
__module__� __qualname__�attrZibr �float�
ws_receive�ws_close� r6 r6 �E/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/client_ws.pyr- 4 s r- g $@)r4 r5 c @ s� e Zd Zdddd�dedededB ded ed
ed ed e j
d
e dB de deddfdd�Z
d{dd�Zd{dd�Zd{dd�Zd{dd�Zd{dd �Zd{d d �Zd|d!d"�Zd{d#d$�Zd%eddfd&d'�Zd{d(d)�Zd{d*d+�Zedefd,d-��Zede dB fd.d/��Z ededB fd0d1��Z ede fd2d3��Z edefd4d5��Z d}d6ed7e de fd8d9�Z!dedB fd:d;�Z"d~d=e#ddfd>d?�Z$d~d=e#ddfd@dA�Z% d}d=e#dBe&de dB ddfdCdD�Z'd}dEede dB ddfdFdG�Z(d}dEe#de dB ddfdHdI�Z) d}e*dJ�dEe de dB dKe+ddfdLdM�Z, d}dEe de dB dKe-ddfdNdO�Z.e/j0d<dP�dQe d=e#defdRdS�Z1e2 d}dTdUd e dB de3fdVdW��Z4e2 d}dTdXd e dB de5fdYdW��Z4e2 d}dTdZd e dB de3e5B fd[dW��Z4 d}d e dB de3e5B fd\dW�Z4e2dd]�dTdUd e dB defd^d_��Z6e2dd]�dTdXd e dB de#fd`d_��Z6e2dd]�dTdZd e dB dee#B fdad_��Z6dd]�d e dB dee#B fdbd_�Z6dd]�d e dB de#fdcdd�Z7e2deddf�dTdUdge8d e dB de fdhdi��Z9e2deddf�dTdXdge:e#ge f d e dB de fdjdi��Z9e2deddf�dTdZdge8e:e#ge f B d e dB de fdkdi��Z9e;ddf�dge8e:e#ge f B d e dB de fdldi�Z9de<fdmdn�Z=e2dTdUde3fdodp��Z>e2dTdXde5fdqdp��Z>e2dTdZde3e5B fdrdp��Z>de3e5B fdsdp�Z>de<fdtdu�Z?dve@e dB dwedB dxeAdB ddfdydz�ZBdS )�ClientWebSocketResponseNr F)� heartbeat�compress�client_notakeover�reader�writer�protocol�response�timeout� autoclose�autoping�loopr9 r: r; �returnc C s� || _ |j| _|| _|| _|| _d| _d| _d | _|| _ || _
|| _ | | _ d | _
d| _| d ur4| d | _d | _|| _d| _d | _d | _|
| _| | _d | _d| _d | _| �� d S )NFg g @)� _response�
connection�_conn�_writer�_reader� _protocol�_closed�_closing� _close_code�_timeout�
_autoclose� _autoping�
_heartbeat�
_heartbeat_cb�_heartbeat_when�_pong_heartbeat�_pong_response_cb�_loop�_waiting� _close_wait�
_exception� _compress�_client_notakeover�
_ping_task�_need_heartbeat_reset�_heartbeat_reset_handle�_reset_heartbeat) �selfr< r= r> r? r@ rA rB rC r9 r: r; r6 r6 r7 �__init__> s6
z ClientWebSocketResponse.__init__c C sd | � � | jd ur| j�� d | _d| _| jd ur!| j�� d | _| jd ur0| j�� d | _d S d S �NF)�_cancel_pong_response_cbr^ �cancelr] rR r\ �r` r6 r6 r7 �_cancel_heartbeatk s
�z)ClientWebSocketResponse._cancel_heartbeatc C s" | j d ur| j �� d | _ d S d S �N)rU rd re r6 r6 r7 rc x s
�z0ClientWebSocketResponse._cancel_pong_response_cbc C s> | j d u s| jr
d S | j}|d usJ �d| _|�| j�| _d S )NT)rQ r] rV Z call_soon�_flush_heartbeat_resetr^ )r` rC r6 r6 r7 �_on_data_received} s z)ClientWebSocketResponse._on_data_receivedc C s" d | _ | jsd S | �� d| _d S rb )r^ r] r_ re r6 r6 r7 rh � s
z.ClientWebSocketResponse._flush_heartbeat_resetc C s� | j d u rd S | �� | j}|d usJ �| j}|d ur |jjnd}|�� }t|| j |�}|| _| j d u r>|�
|| j �| _ d S d S )N� ) rQ rc rV rG �
_connector�_timeout_ceil_threshold�timer rS rR �call_at�_send_heartbeat)r` rC �conn�timeout_ceil_threshold�now�whenr6 r6 r7 r_ � s
�
�z(ClientWebSocketResponse._reset_heartbeatc C s� d | _ | jrd S | j}|�� }|| jk r |�| j| j�| _ d S | j}|d ur*|jj nd}t
|| j |�}| � � |�|| j
�| _| j�dtj�}tjdkrUtj||dd�}n|�|�}|�� si|| _|�| j� d S | �|� d S )Nrj � )r" � T)rC Z eager_start) rR r] rV rm rS rn ro rG rk rl r rT rc �_pong_not_receivedrU rH �
send_framer �PING�sys� version_info�asyncioZTaskZ create_taskZdoner\ Zadd_done_callback�_ping_task_done)r` rC rr rp rq rs �coroZ ping_taskr6 r6 r7 ro � s0
��
z'ClientWebSocketResponse._send_heartbeat�task�asyncio.Task[None]c C s( |� � s|�� }r| �|� d| _dS )z*Callback for when the ping task completes.N)Z cancelled� exception�_handle_ping_pong_exceptionr\ )r` r~ �excr6 r6 r7 r| � s
z'ClientWebSocketResponse._ping_task_donec C s | � td| j� d��� d S )NzNo PONG received after z seconds)r� r
rT re r6 r6 r7 rv � s �z*ClientWebSocketResponse._pong_not_receivedr� c C sZ | j rdS | �� tj| _|| _| j�� | jr)| j s+| j
� t t
j|d�d� dS dS dS )z5Handle exceptions raised during ping/pong processing.Nr )rK � _set_closedr �ABNORMAL_CLOSURErM rY rE �closerW rL rI � feed_datar r �ERROR)r` r� r6 r6 r7 r� � s
�z3ClientWebSocketResponse._handle_ping_pong_exceptionc C � d| _ | �� dS )zdSet the connection to closed.
Cancel any heartbeat timers and set the closed flag.
TN)rK rf re r6 r6 r7 r� � � z#ClientWebSocketResponse._set_closedc C r� )zfSet the connection to closing.
Cancel any heartbeat timers and set the closing flag.
TN)rL rf re r6 r6 r7 � _set_closing� r� z$ClientWebSocketResponse._set_closingc C � | j S rg )rK re r6 r6 r7 �closed� � z ClientWebSocketResponse.closedc C r� rg )rM re r6 r6 r7 �
close_code� r� z"ClientWebSocketResponse.close_codec C r� rg )rJ re r6 r6 r7 r> � r� z ClientWebSocketResponse.protocolc C r� rg )rZ re r6 r6 r7 r: � r� z ClientWebSocketResponse.compressc C r� rg )r[ re r6 r6 r7 r; r� z)ClientWebSocketResponse.client_notakeover�namer* c C s2 | j j}|du r
|S |j}|du r|S |�||�S )z$extra info from connection transportN)rE rF � transport�get_extra_info)r` r� r* rp r� r6 r6 r7 r� s z&ClientWebSocketResponse.get_extra_infoc C r� rg )rY re r6 r6 r7 r� s z!ClientWebSocketResponse.exceptionrt �messagec � � �| j �|tj�I d H d S rg )rH rw r rx �r` r� r6 r6 r7 �ping � �z ClientWebSocketResponse.pingc � r� rg )rH rw r �PONGr� r6 r6 r7 �pong r� z ClientWebSocketResponse.pong�opcodec � s �| j �|||�I dH dS )z Send a frame over the websocket.N)rH rw )r` r� r� r: r6 r6 r7 rw s �z"ClientWebSocketResponse.send_frame�datac � s@ �t |t�stdt|� ��| jj|�d�tj|d�I d H d S )Nz data argument must be str (%r)zutf-8�r: ) �
isinstance�str� TypeErrorr. rH rw �encoder �TEXT�r` r� r: r6 r6 r7 �send_str s �
�z ClientWebSocketResponse.send_strc � s@ �t |tttf�stdt|� ��| jj|tj |d�I d H d S )Nz#data argument must be byte-ish (%r)r� )
r� �bytes� bytearray�
memoryviewr� r. rH rw r �BINARYr� r6 r6 r7 �
send_bytes% s � z"ClientWebSocketResponse.send_bytes)�dumpsr� c � s �| j ||�|d�I d H d S )Nr� )r� �r` r� r: r� r6 r6 r7 � send_json* s � z!ClientWebSocketResponse.send_jsonc � s �| j ||�|d�I dH dS )z�Send JSON data using a bytes-returning encoder as a binary frame.
Use this when your JSON encoder (like orjson) returns bytes
instead of str, avoiding the encode/decode overhead.
r� N)r� r� r6 r6 r7 �send_json_bytes3 s � z'ClientWebSocketResponse.send_json_bytes)�coder� r� c
� s� �| j r%| js%| jd usJ �| j�� | _| �� | j�td� | jI d H | j r*dS | �
� z | j � ||�I d H W n0 t
jyL tj| _| j� � � tyj } ztj| _|| _| j� � W Y d }~dS d }~ww | jru| j� � dS z+t�| jj�4 I d H � | j�� I d H }W d �I d H n 1 I d H s�w Y W n0 t
jy� tj| _| j� � � ty� } ztj| _|| _| j� � W Y d }~dS d }~ww |jtj u r�|j | _| j� � dS qv)Nr FT) rW rL rV Z
create_futurerX r� rI r� r rK r� rH r� r{ �CancelledErrorr r� rM rE � ExceptionrY �
async_timeoutr@ rN r5 �readr. r �CLOSEr� )r` r� r� r� �msgr6 r6 r7 r� A s` �
��
(��
��
�z ClientWebSocketResponse.closer` z&ClientWebSocketResponse[Literal[True]]c � � �d S rg r6 �r` r@ r6 r6 r7 �receiver � �z ClientWebSocketResponse.receivez'ClientWebSocketResponse[Literal[False]]c � r� rg r6 r� r6 r6 r7 r� w r� z$ClientWebSocketResponse[_DecodeText]c � r� rg r6 r� r6 r6 r7 r� | r� c
� s� �|p| j j} | jrtd��| jrtS | jr | �� I d H tS zTd| _z@|rOt� |�4 I d H � | j
� � I d H }W d �I d H n 1 I d H sIw Y n| j
� � I d H }W d| _| j rdt
| j d � nd| _| j rrt
| j d � w w W n� tjtjfy� tj| _� ty� tj| _| �� I d H ttjd d � Y S ty� | �� tj| _t Y S ty� } z|j | _| j|j d�I d H ttj |d �W Y d }~S d }~w t �y } z || _ | � � tj| _| �� I d H ttj |d �W Y d }~S d }~ww |j!t"v�r
|S |j!tj#u �r*| � � |j$| _| j�s(| j%�r(| �� I d H |S |j!tj&u �r7| � � |S |j!tj'u �rL| j(�rL| �)|j$�I d H q|j!tj*u �rX| j(�rXq|S )NTz+Concurrent call to receive() is not allowedF)r� )+rN r4 rW � RuntimeErrorrK r rL r� r� r@ rI r� rX r r{ r� � TimeoutErrorr r� rM r �OKr r �CLOSEDr r� r r� r� r� rY r� r. r r� r� rO �CLOSINGrx rP r� r� )r` r@ Zreceive_timeoutr� r� r6 r6 r7 r� � s� � (��� ���� ��� ���r@ c � r� rg r6 r� r6 r6 r7 � receive_str� r� z#ClientWebSocketResponse.receive_strc � r� rg r6 r� r6 r6 r7 r� � r� c � r� rg r6 r� r6 r6 r7 r� � r� c � sD �| � |�I dH }|jtjur td|j� d|j�d���tt|j�S )zrReceive TEXT message.
Returns str when decode_text=True (default), bytes when decode_text=False.
N�Received message �:z is not WSMsgType.TEXT)r� r. r r� r r� r r� �r` r@ r� r6 r6 r7 r� � s � � c � sD �| � |�I d H }|jtjur td|j� d|j�d���tt|j�S )Nr� r� z is not WSMsgType.BINARY)r� r. r r� r r� r r� r� r6 r6 r7 �
receive_bytes� s � � z%ClientWebSocketResponse.receive_bytes.)�loadsr@ r� c � r� rg r6 �r` r� r@ r6 r6 r7 � receive_json� � �z$ClientWebSocketResponse.receive_jsonc � r� rg r6 r� r6 r6 r7 r� � r� c � r� rg r6 r� r6 r6 r7 r� � r� c � s �| j |d�I d H }||�S )Nr� )r� )r` r� r@ r� r6 r6 r7 r� s �c C s | S rg r6 re r6 r6 r7 � __aiter__ s z!ClientWebSocketResponse.__aiter__c � r� rg r6 re r6 r6 r7 � __anext__ r� z!ClientWebSocketResponse.__anext__c � r� rg r6 re r6 r6 r7 r� r� c � r� rg r6 re r6 r6 r7 r� r� c � s. �| � � I d H }|jtjtjtjfv rt�|S rg )r� r. r r� r� r� �StopAsyncIteration)r` r� r6 r6 r7 r� ! s
�c � s �| S rg r6 re r6 r6 r7 �
__aenter__' s �z"ClientWebSocketResponse.__aenter__�exc_type�exc_val�exc_tbc � s �| � � I d H d S rg )r� )r` r� r� r� r6 r6 r7 � __aexit__* s �z!ClientWebSocketResponse.__aexit__)rD N)r~ r rD Nrg )rt )Cr/ r0 r1 r r r� r r- �boolr{ ZAbstractEventLoopr3 �intra rf rc ri rh r_ ro r| rv �
BaseExceptionr� r� r� �propertyr� r� r> r: r; r r� r� r� r� r� r rw r� r� r r! r� r r� r r� r� r r r� r r� r� r r� r r r& r� r� r� r. r r� r6 r6 r6 r7 r8 = s� �������� � � �
�
�
-
(
����
� �����
� ����
� 1��� ���� ���� ���
�J��� ���� ���� � ���� ��� �� ����� �
���
� � �� �� �
����r8 )9�__doc__r{ ry �collections.abcr �typesr �typingr r r r r r r2 Z_websocket.readerr Zclient_exceptionsr r
r Z
client_reqrepr Zhelpersr r �httpr r r r r r r r Zhttp_websocketr r Zstreamsr Ztypedefsr r r r r! rz r$ Ztyping_extensionsr� r& r� r' �sr- ZDEFAULT_WS_CLIENT_TIMEOUTr8 r6 r6 r6 r7 �<module> s8 (