File: //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/client_ws.cpython-37.opt-1.pyc
B
�k�a,) � @ s� d Z ddlZddlmZmZmZ ddlZddlmZ ddl m
Z
ddl m Z m
Z
ddlmZmZmZmZmZmZ dd lmZ dd
lmZmZ dd lmZm Z m Z m Z G d d
� d
�Z dS )z WebSocket client for asyncio.� N)�Any�Optional�cast� )� ClientError)�ClientResponse)�
call_later�
set_result)�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WebSocketError� WSCloseCode� WSMessage� WSMsgType)�WebSocketWriter)� EofStream�FlowControlDataQueue)�DEFAULT_JSON_DECODER�DEFAULT_JSON_ENCODER� JSONDecoder� JSONEncoderc @ s e Zd Zddddd�deee eeeee j
ee ee e edd�
dd�Z dd �d
d �Z
dd �d d
�Zdd �dd�Zdd �dd�Zeed �dd��Zeee d �dd��Zeee d �dd��Zee d �dd��Zeed �dd��ZdDeeed �d d �Zee d �d d �ZdEedd"�d#d$�Z dFedd"�d%d&�Z dGeee dd'�d(d)�Z dHeee dd'�d*d+�Z dIe d,�eee e!dd-�d.d/�Z"e#j$d!d0�e eed1�d2d3�Z%dJee e&d4�d5d6�Z'dd7�ee ed4�d8d9�Z(dd7�ee ed4�d:d;�Z)e*dd<�e+ee ed=�d>d?�Z,d d �d@dA�Z-e&d �dBdC�Z.dS )K�ClientWebSocketResponseNr F)�receive_timeout� heartbeat�compress�client_notakeoverz FlowControlDataQueue[WSMessage])
�reader�writer�protocol�response�timeout� autoclose�autoping�loopr r r r �returnc
C s� || _ |j| _|| _|| _|| _d| _d| _d | _|| _ | | _
|| _ || _ |
| _
d | _|
d k rh|
d | _d | _|| _d | _d | _| | _| | _| �� d S )NFg @)� _response�
connectionZ_conn�_writer�_reader� _protocol�_closed�_closing� _close_code�_timeout�_receive_timeout�
_autoclose� _autoping�
_heartbeat�
_heartbeat_cb�_pong_heartbeat�_pong_response_cb�_loop�_waiting�
_exception� _compress�_client_notakeover�_reset_heartbeat)
�selfr r r r r r! r" r# r r r r � r< �D/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/client_ws.py�__init__ s.
z ClientWebSocketResponse.__init__)r$ c C s8 | j d k r| j �� d | _ | jd k r4| j�� d | _d S )N)r4 �cancelr2 )r; r< r< r= �_cancel_heartbeatH s
z)ClientWebSocketResponse._cancel_heartbeatc C s* | � � | jd k r&t| j| j| j�| _d S )N)r@ r1 r �_send_heartbeatr5 r2 )r; r< r< r= r: Q s
z(ClientWebSocketResponse._reset_heartbeatc C sN | j d k rJ| jsJ| j�| j�� � | jd k r6| j�� t| j | j
| j�| _d S )N) r1 r* r5 Z create_taskr' �pingr4 r? r �_pong_not_receivedr3 )r; r< r< r= rA Y s
z'ClientWebSocketResponse._send_heartbeatc C s, | j s(d| _ tj| _t�� | _| j�� d S )NT) r* r
�ABNORMAL_CLOSUREr, �asyncio� TimeoutErrorr7 r% �close)r; r< r< r= rC f s
z*ClientWebSocketResponse._pong_not_receivedc C s | j S )N)r* )r; r< r< r= �closedm s z ClientWebSocketResponse.closedc C s | j S )N)r, )r; r< r< r= �
close_codeq s z"ClientWebSocketResponse.close_codec C s | j S )N)r) )r; r< r< r= r u s z ClientWebSocketResponse.protocolc C s | j S )N)r8 )r; r< r< r= r y s z ClientWebSocketResponse.compressc C s | j S )N)r9 )r; r< r< r= r } s z)ClientWebSocketResponse.client_notakeover)�name�defaultr$ c C s2 | j j}|dkr|S |j}|dkr&|S |�||�S )z$extra info from connection transportN)r% r& � transport�get_extra_info)r; rJ rK ZconnrL r< r< r= rM � s z&ClientWebSocketResponse.get_extra_infoc C s | j S )N)r7 )r; r< r< r= � exception� s z!ClientWebSocketResponse.exception� )�messager$ c � s | j �|�I d H d S )N)r' rB )r; rP r< r< r= rB � s z ClientWebSocketResponse.pingc � s | j �|�I d H d S )N)r' �pong)r; rP r< r< r= rQ � s z ClientWebSocketResponse.pong)�datar r$ c � s6 t |t�stdt|� ��| jj|d|d�I d H d S )Nz data argument must be str (%r)F)�binaryr )�
isinstance�str� TypeError�typer' �send)r; rR r r< r< r= �send_str� s
z ClientWebSocketResponse.send_strc � s< t |tttf�s tdt|� ��| jj|d|d�I d H d S )Nz#data argument must be byte-ish (%r)T)rS r )rT �bytes� bytearray�
memoryviewrV rW r' rX )r; rR r r< r< r= �
send_bytes� s z"ClientWebSocketResponse.send_bytes)�dumps)rR r r^ r$ c � s | j ||�|d�I d H d S )N)r )rY )r; rR r r^ r< r< r= � send_json� s z!ClientWebSocketResponse.send_json)�coderP )r` rP r$ c
� s� | j d k r*| js*| j�td� | j I d H | j�s�| �� d| _y| j�||�I d H W nb tj k
r� t
j | _ | j
�� � Y n: tk
r� } z t
j | _ || _| j
�� dS d }~X Y nX | jr�| j
�� dS x�y:t�| j�4 I d H � | j�� I d H }W d Q I d H R X W nf tj k
�r8 t
j | _ | j
�� � Y n< tk
�rr } z t
j | _ || _| j
�� dS d }~X Y nX |jtjkr�|j| _ | j
�� dS q�W ndS d S )Nr TF)r6 r* r( Z feed_datar r@ r' rG rE �CancelledErrorr
rD r, r% � Exceptionr7 r+ �
async_timeoutr r- �readrW r �CLOSErR )r; r` rP �exc�msgr< r< r= rG � sL
$
z ClientWebSocketResponse.close)r r$ c
� s\ �xT| j d k rtd��| jr tS | jr8| �� I d H tS yr| j�� | _ zFt� |pT| j
�4 I d H � | j � � I d H }W d Q I d H R X | �
� W d | j }d | _ t|d� X W �n tjtjfk
r� tj| _� Y n� tk
�r tj| _| �� I d H ttjd d �S tk
�r& d| _tj| _tS tk
�rn } z*|j | _| j|j d�I d H ttj |d �S d }~X Y nP t k
�r� } z0|| _ d| _tj| _| �� I d H ttj |d �S d }~X Y nX |j tj!k�r�d| _|j"| _| j�sT| j#�rT| �� I d H nZ|j tj$k�rd| _nD|j tj%k�r<| j&�r<| �'|j"�I d H qn|j tj(k�rT| j&�rTq|S d S )Nz+Concurrent call to receive() is not allowedT)r` ))r6 � RuntimeErrorr* r
r+ rG r5 Z
create_futurerc r r. r( rd r: r rE ra rF r
rD r, r �OKr r �CLOSEDr r r` �ERRORrb r7 rW re rR r/ �CLOSINGZPINGr0 rQ ZPONG)r; r rg �waiterrf r< r< r= �receive� sf
z ClientWebSocketResponse.receive)r c � sB | � |�I d H }|jtjkr6td|j� d|j�d���tt|j�S )NzReceived message �:z is not str)rn rW r ZTEXTrV rR r rU )r; r rg r< r< r= � receive_str s z#ClientWebSocketResponse.receive_strc � sB | � |�I d H }|jtjkr6td|j� d|j�d���tt|j�S )NzReceived message ro z
is not bytes)rn rW r ZBINARYrV rR r rZ )r; r rg r< r< r= �
receive_bytes s z%ClientWebSocketResponse.receive_bytes)�loadsr )rr r r$ c � s | j |d�I d H }||�S )N)r )rp )r; rr r rR r< r< r= � receive_json s z$ClientWebSocketResponse.receive_jsonc C s | S )Nr< )r; r< r< r= � __aiter__% s z!ClientWebSocketResponse.__aiter__c � s, | � � I d H }|jtjtjtjfkr(t�|S )N)rn rW r re rl rj �StopAsyncIteration)r; rg r< r< r= � __anext__( s z!ClientWebSocketResponse.__anext__)N)rO )rO )N)N)N)N)/�__name__�
__module__� __qualname__r r rU r �float�boolrE ZAbstractEventLoop�intr> r@ r: rA rC �propertyrH rI r r r r rM �
BaseExceptionrN rZ rB rQ rY r] r r r_ r
ri rG r rn rp rq r r rs rt rv r< r< r< r= r sJ ,
/: r ) �__doc__rE �typingr r r rc Zclient_exceptionsr Z
client_reqrepr Zhelpersr r Zhttpr
r r r
r r Zhttp_websocketr Zstreamsr r Ztypedefsr r r r r r< r< r< r= �<module> s