���Z�" � @ s� d Z d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l m
Z
m Z m Z m
Z
m Z Gd d � d � Z d S) z WebSocket client for asyncio.� N� )� ClientError)�
call_later�
set_result)�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WebSocketError� WSMessage� WSMsgTypec
@ s� e Z d Z d d d d d d d d d d � Z d
d � Z d d
� Z d d � Z d d � Z e d d � � Z e d d � � Z
e d d � � Z e d d � � Z e d d � � Z
d d d � Z d d � Z d d! d" � Z d d# d$ � Z d d% d& � Z d d' d( � Z d d) e j d* d+ �Z d, d- d. d/ d0 d1 � Z d d2 d3 � Z d4 d d5 d6 � Z d4 d d7 d8 � Z d9 e j d4 d d: d; � Z d< d= � Z d> d? � Z d S)@�ClientWebSocketResponse�receive_timeoutN� heartbeat�compressr �client_notakeoverFc
C s� | | _ | j | _ | | _ | | _ | | _ d | _ d | _ d | _ | | _ | | _
| | _ | | _ |
| _
d | _ |
d k r� |
d | _ d | _ | | _ d | _ d | _ | | _ | | _ | j � d S)NFg @)� _response�
connection�_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)
�self�reader�writer�protocol�response�timeoutZ autocloseZautoping�loopr r
r r � r. �
/client_ws.py�__init__ s.
z ClientWebSocketResponse.__init__c C sN | j d k r% | j j � d | _ | j d k rJ | j j � d | _ d S)N)r �cancelr )r' r. r. r/ �_cancel_heartbeat. s
z)ClientWebSocketResponse._cancel_heartbeatc C s; | j � | j d k r7 t | j | j | j � | _ d S)N)r2 r r �_send_heartbeatr! r )r' r. r. r/ r& 7 s
z(ClientWebSocketResponse._reset_heartbeatc C sd | j d k r` | j r` | j j � | j d k rB | j j � t | j | j | j � | _ d S)N)
r r r �pingr r1 r �_pong_not_receivedr r! )r' r. r. r/ r3 > s
z'ClientWebSocketResponse._send_heartbeatc C s; | j s7 d | _ d | _ t j � | _ | j j � d S)NTi� )r r �asyncio� TimeoutErrorr# r �close)r' r. r. r/ r5 G s
z*ClientWebSocketResponse._pong_not_receivedc C s | j S)N)r )r' r. r. r/ �closedN s z ClientWebSocketResponse.closedc C s | j S)N)r )r' r. r. r/ �
close_codeR s z"ClientWebSocketResponse.close_codec C s | j S)N)r )r' r. r. r/ r* V s z ClientWebSocketResponse.protocolc C s | j S)N)r$ )r' r. r. r/ r Z s z ClientWebSocketResponse.compressc C s | j S)N)r% )r' r. r. r/ r ^ s z)ClientWebSocketResponse.client_notakeoverc C s: y | j j j j | | � SWn t k
r5 | SYn Xd S)z$extra info from connection transportN)r r Z transport�get_extra_info� Exception)r' �name�defaultr. r. r/ r; b s
z&ClientWebSocketResponse.get_extra_infoc C s | j S)N)r# )r' r. r. r/ � exceptionj s z!ClientWebSocketResponse.exception�bc � s | j j | � Id Hd S)N)r r4 )r' �messager. r. r/ r4 m s z ClientWebSocketResponse.pingc � s | j j | � Id Hd S)N)r �pong)r' rA r. r. r/ rB p s z ClientWebSocketResponse.pongc � sJ t | t � s% t d t | � � � | j j | d d d | �Id Hd S)Nz data argument must be str (%r)�binaryFr )�
isinstance�str� TypeError�typer �send)r' �datar r. r. r/ �send_strs s z ClientWebSocketResponse.send_strc � sS t | t t t f � s. t d t | � � � | j j | d d d | �Id Hd S)Nz#data argument must be byte-ish (%r)rC Tr )rD �bytes� bytearray�
memoryviewrF rG r rH )r' rI r r. r. r/ �
send_bytesx s z"ClientWebSocketResponse.send_bytes�dumpsc � s" | j | | � d | �Id Hd S)Nr )rJ )r' rI r rO r. r. r/ � send_json~ s z!ClientWebSocketResponse.send_json�codei� rA � c � s� | j d k r8 | j r8 | j j t d � | j Id H| j s�| j � d | _ y | j j | | � Wnu t j k
r� d | _
| j j � � YnH t k
r� } z( d | _
| | _
| j j � d SWYd d } ~ Xn X| j r� | j j � d Sx� y; t j | j d | j �� | j j � Id H} Wd QRXWnu t j k
rkd | _
| j j � � YnH t k
r�} z( d | _
| | _
| j j � d SWYd d } ~ Xn X| j t j k r | j | _
| j j � d Sq Wn d Sd S)Nr Ti� r- F)r" r r Z feed_datar r2 r r8 r6 �CancelledErrorr r r< r# r �
async_timeoutr, r r! �readrG r
�CLOSErI )r' rQ rA �exc�msgr. r. r/ r8 � sL
z ClientWebSocketResponse.closec � s� x�| j d k r t d � � | j r+ t S| j rG | j � Id Ht Sy� | j j � | _ zK t j | pq | j
d | j �� | j j � Id H} Wd QRX| j
� Wd | j } d | _ t | d � XWn
t j t j f k
r� d | _ � Yn� t k
r d | _ d | _ t SYn� t k
ry} z< | j | _ | j d | j � Id Ht t j | d � SWYd d } ~ Xnb t k
r�} zB | | _ d | _ d | _ | j � Id Ht t j | d � SWYd d } ~ Xn X| j t j k r'd | _ | j | _ | j r�| j r�| j � Id Hnr | j t j k rEd | _ nT | j t j k r{| j! r{| j" | j � Id Hq n | j t j# k r�| j! r�q | SWd S)Nz+Concurrent call to receive() is not allowedr- Ti� rQ )$r" � RuntimeErrorr r r r8 r! Z
create_futurerT r, r r rU r& r r6 rS r7 r r r rQ r r
�ERRORr< r# rG rV rI r �CLOSINGZPINGr rB ZPONG)r' r, rX �waiterrW r. r. r/ �receive� sd
% & z ClientWebSocketResponse.receiver, c � sK | j | � Id H} | j t j k rD t d j | j | j � � � | j S)Nz#Received message {}:{!r} is not str)r] rG r
ZTEXTrF �formatrI )r' r, rX r. r. r/ � receive_str� s z#ClientWebSocketResponse.receive_strc � sK | j | � Id H} | j t j k rD t d j | j | j � � � | j S)Nz%Received message {}:{!r} is not bytes)r] rG r
ZBINARYrF r^ rI )r' r, rX r. r. r/ �
receive_bytes� s z%ClientWebSocketResponse.receive_bytes�loadsc � s! | j d | � Id H} | | � S)Nr, )r_ )r' ra r, rI 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< | j � Id H} | j t j t j t j f k r8 t � | S)N)r] rG r
rV r[ ZCLOSED�StopAsyncIteration)r' rX r. r. r/ � __anext__ s z!ClientWebSocketResponse.__anext__) �__name__�
__module__� __qualname__r0 r2 r& r3 r5 �propertyr9 r: r* r r r; r? r4 rB rJ rN �jsonrO rP r8 r] r_ r` ra rb rc re r. r. r. r/ r s4 /9 r )�__doc__r6 rj rT Zclient_exceptionsr Zhelpersr r Zhttpr r r r r
r r. r. r. r/ �<module> s (