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_MESSAGEWS_CLOSING_MESSAGEWebSocketError 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)@ClientWebSocketResponsereceive_timeoutN heartbeatcompressr 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)
selfreaderwriterprotocolresponsetimeoutZ autocloseZautopingloopr 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 transportget_extra_info Exception)r' namedefaultr. r. r/ r; b s
z&ClientWebSocketResponse.get_extra_infoc C s | j S)N)r# )r' r. r. r/ exceptionj s z!ClientWebSocketResponse.exceptionbc 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 )
isinstancestr TypeErrortyper 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_bytesdumpsc s" | j | | d | Id Hd S)Nr )rJ )r' rI r rO r. r. r/ send_json~ s z!ClientWebSocketResponse.send_jsoncodei 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 excmsgr. 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! rq | 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_bytesloadsc 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[ ZCLOSEDStopAsyncIteration)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 (