U
'7`/( @ s d Z ddlZddlmZmZ ddlZddlmZ ddlm Z ddl
m Z m Z ddl
mZmZmZmZmZ dd lmZ dd
lmZmZ dd lmZmZmZm Z G d d
d
Z dS )z WebSocket client for asyncio. N)AnyOptional ) ClientError)ClientResponse)
call_later
set_result)WS_CLOSED_MESSAGEWS_CLOSING_MESSAGEWebSocketError WSMessage WSMsgType)WebSocketWriter) EofStreamFlowControlDataQueue)DEFAULT_JSON_DECODERDEFAULT_JSON_ENCODER JSONDecoder JSONEncoderc @ s e Zd Zddddddeee eeeee j
ee ee e edd
ddZ dd d
d Z
dd d d
Zdd ddZdd ddZeed ddZeee d ddZeee d ddZee d ddZeed ddZdEeeed d d Zee d d d ZdFedd"d#d$Z dGedd"d%d&Z dHeee dd'd(d)Z dIeee dd'd*d+Z dJe d,eee e!dd-d.d/Z"d0d!d1e eed2d3d4Z#dKee e$d5d6d7Z%dd8ee ed5d9d:Z&dd8ee ed5d;d<Z'e(dd=e)ee ed>d?d@Z*d d dAdBZ+e$d dCdDZ,dS )LClientWebSocketResponseNr F)receive_timeout heartbeatcompressclient_notakeoverz FlowControlDataQueue[WSMessage])
readerwriterprotocolresponsetimeout autocloseautopingloopr 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/python38/lib64/python3.8/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)r2 cancelr0 r9 r: r: r; _cancel_heartbeatG s
z)ClientWebSocketResponse._cancel_heartbeatc C s* | | jd k r&t| j| j| j| _d S r= )r@ r/ r _send_heartbeatr3 r0 r? r: r: r; r8 P s
z(ClientWebSocketResponse._reset_heartbeatc C sN | j d k rJ| jsJ| j| j | jd k r6| j t| j | j
| j| _d S r= ) r/ r( r3 Z create_taskr% pingr2 r> r _pong_not_receivedr1 r? r: r: r; rA X s
z'ClientWebSocketResponse._send_heartbeatc C s* | j s&d| _ d| _t | _| j d S )NT )r( r* asyncio TimeoutErrorr5 r# closer? r: r: r; rC e s
z*ClientWebSocketResponse._pong_not_receivedc C s | j S r= )r( r? r: r: r; closedl s z ClientWebSocketResponse.closedc C s | j S r= )r* r? r: r: r;
close_codep s z"ClientWebSocketResponse.close_codec C s | j S r= )r' r? r: r: r; r t s z ClientWebSocketResponse.protocolc C s | j S r= )r6 r? r: r: r; r x s z ClientWebSocketResponse.compressc C s | j S r= )r7 r? r: r: r; r | s z)ClientWebSocketResponse.client_notakeover)namedefaultr" c C s2 | j j}|dkr|S |j}|dkr&|S |||S )z$extra info from connection transportN)r# r$ transportget_extra_info)r9 rJ rK ZconnrL r: r: r; rM s z&ClientWebSocketResponse.get_extra_infoc C s | j S r= )r5 r? r: r: r; exception s z!ClientWebSocketResponse.exception )messager" c s | j |I d H d S r= )r% rB r9 rP r: r: r; rB s z ClientWebSocketResponse.pingc s | j |I d H d S r= )r% pongrQ r: r: r; rR s z ClientWebSocketResponse.pong)datar r" c s6 t |tstdt| | jj|d|dI d H d S )Nz data argument must be str (%r)FZbinaryr )
isinstancestr TypeErrortyper% sendr9 rS r r: r: r; send_str s
z ClientWebSocketResponse.send_strc s< t |tttfs tdt| | jj|d|dI d H d S )Nz#data argument must be byte-ish (%r)TrT )rU bytes bytearray
memoryviewrW rX r% rY rZ r: r: r;
send_bytes s z"ClientWebSocketResponse.send_bytes)dumps)rS r r` r" c s | j |||dI d H d S )N)r )r[ )r9 rS r r` r: r: r; send_json s z!ClientWebSocketResponse.send_json )coderP )rc rP r" c
s | j d k r*| js*| jtd | j I d H | js| d| _z| j||I d H W nd tj k
r d| _
| j Y n> t k
r } z d| _
|| _
| j W Y dS d }~X Y nX | jr| j dS z2tj| j| jd | j I d H }W 5 Q R X W nh tj k
r. d| _
| j Y n@ t k
rl } z d| _
|| _
| j W Y dS d }~X Y nX |jtjkr|j| _
| j dS qndS d S )Nr TrD r! F)r4 r( r& Z feed_datar
r@ r% rG rE CancelledErrorr* r# Exceptionr5 r)
async_timeoutr r+ r3 readrX r
CLOSErS )r9 rc rP excmsgr: r: r; rG sJ
z ClientWebSocketResponse.close)r r" c
s\ | j d k rtd| jr tS | jr4| I d H tS zj| j | _ z>t j
|pP| j | jd | j
I d H }W 5 Q R X | W 5 | j }d | _ t|d X W n" tjtjfk
r d| _ Y n tk
r d| _| I d H ttjd d Y S tk
r d| _d| _t Y S tk
rl } z2|j| _| j|jdI d H ttj|d W Y S d }~X Y nV tk
r } z6|| _ d| _d| _| I d H ttj|d W Y S d }~X Y nX |j tj krd| _|j | _| jsX| j rX| I d H nZ|j tj!krd| _nD|j tj"kr@| j#r@| $|j I d H q n|j tj%krX| j#rXq |S )Nz+Concurrent call to receive() is not allowedTrd rD rb )rc )&r4 RuntimeErrorr( r r) rG r3 Z
create_futurer rg r r, r& rh r8 rE re rF r* r r r
CLOSEDr r rc ZERRORrf r5 rX ri rS r- CLOSINGZPINGr. rR ZPONG)r9 r Zwaiterrk rj r: r: r; receive sj
"$z ClientWebSocketResponse.receiver c s< | |I d H }|jtjkr6td|j d|jd|jS )NReceived message :z is not str)ro rX r
ZTEXTrW rS r9 r rk r: r: r; receive_str s z#ClientWebSocketResponse.receive_strc s< | |I d H }|jtjkr6td|j d|jd|jS )Nrq rr z
is not bytes)ro rX r
ZBINARYrW rS rs r: r: r;
receive_bytes s z%ClientWebSocketResponse.receive_bytes)loadsr )rv r r" c s | j |dI d H }||S )Nrp )rt )r9 rv r rS r: r: r; receive_json s z$ClientWebSocketResponse.receive_jsonc C s | S r= r: r? r: r: r; __aiter__& s z!ClientWebSocketResponse.__aiter__c s, | I d H }|jtjtjtjfkr(t|S r= )ro rX r
ri rn rm StopAsyncIteration)r9 rk r: r: r; __anext__) s z!ClientWebSocketResponse.__anext__)N)rO )rO )N)N)N)N)-__name__
__module__ __qualname__r r rV r floatboolrE ZAbstractEventLoopintr< r@ r8 rA rC propertyrH rI r r r r rM
BaseExceptionrN r\ rB rR r[ r_ r r ra rG r ro rt ru r r rw rx rz r: r: r: r; r sz *
/< r ) __doc__rE typingr r rg Zclient_exceptionsr Z
client_reqrepr Zhelpersr r Zhttpr r
r r r
Zhttp_websocketr Zstreamsr r Ztypedefsr r r r r r: r: r: r; <module> s