File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/client_ws.cpython-310.pyc
Back
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 (
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings