U
'7`A @ sB d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z d dl
Z
d dl Z d dl m
Z
ddlmZ ddlmZ ddlmZmZ ddlmZmZmZmZmZmZm Z m Z m Z m Z dd l m!Z! dd
l"m#Z#m$Z$ dd l%m&Z&m'Z' dd l(m)Z)m*Z* dd
l+m,Z, ddl-m.Z. dZ/dZ0e j1ddddG dd dZ2G dd de.Z3dS ) N)AnyIterableOptionalTuple) CIMultiDict )hdrs)AbstractStreamWriter)
call_later
set_result)
WS_CLOSED_MESSAGEWS_CLOSING_MESSAGEWS_KEYWebSocketErrorWebSocketReaderWebSocketWriter WSMessage WSMsgType
ws_ext_gen ws_ext_parse) ws_logger) EofStreamFlowControlDataQueue) JSONDecoder JSONEncoder)HTTPBadRequest
HTTPException) BaseRequest)StreamResponse)WebSocketResponseWebSocketReadyr T)Z auto_attribsfrozenslotsc @ s0 e Zd ZU eed< ee ed< edddZdS )r okprotocolreturnc C s | j S N)r$ self r+ A/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/web_ws.py__bool__0 s zWebSocketReady.__bool__N)__name__
__module__ __qualname__bool__annotations__r strr- r+ r+ r+ r, r + s
r c s e Zd ZdZdddddddddeee eeee ee ee dd fd d
Z
dd d d
Z dd ddZ dd ddZ
dd ddZeed fdd Zeedeeef dddZeeeef dddZeeedd d d Zeedd d Zeed d!d"Zeee d d#d$Zeee d d%d&Z eed d'd(Z ee d d)d*Z dWe dd,d-d.Z!dXe dd,d/d0Z"dYeee dd1d2d3Z#dZe ee dd1d4d5Z$d[e%j&d6e'ee e(dd7d8d9Z)dd d:d;Z*d<d+d=e e ed>d?d@Z+d\ee e,dAdBdCZ-ddDee edAdEdFZ.ddDee e dAdGdHZ/e%j0ddIe1ee e'dJdKdLZ2e ddMdNdOZ3d d dPdQZ4e,d dRdSZ5e ddTdUdVZ6 Z7S )]r Fg $@NTr+ i @ )timeoutreceive_timeout autocloseautoping heartbeat protocolscompress max_msg_size) r4 r5 r6 r7 r8 r9 r: r; r' c s t jdd || _d | _d | _d | _d| _d| _d| _d | _ d | _
d | _ d | _ || _
|| _|| _|| _|| _d | _|d k r|d | _d | _|| _|| _d S )Ne )ZstatusFr g @)super__init__
_protocols _ws_protocol_writer_reader_closed_closing
_conn_lost _close_code_loop_waiting
_exception_timeout_receive_timeout
_autoclose _autoping
_heartbeat
_heartbeat_cb_pong_heartbeat_pong_response_cb _compress
_max_msg_size) r* r4 r5 r6 r7 r8 r9 r: r; __class__r+ r, r> 8 s.
zWebSocketResponse.__init__r& c C s8 | j d k r| j d | _ | jd k r4| j d | _d S r( )rQ cancelrO r) r+ r+ r, _cancel_heartbeat\ s
z#WebSocketResponse._cancel_heartbeatc C s* | | jd k r&t| j| j| j| _d S r( )rW rN r
_send_heartbeatrG rO r) r+ r+ r, _reset_heartbeate s
z"WebSocketResponse._reset_heartbeatc C sN | j d k rJ| jsJ| j| j | jd k r6| j t| j | j
| j| _d S r( ) rN rC rG Z create_taskrA pingrQ rV r
_pong_not_receivedrP r) r+ r+ r, rX m s
z!WebSocketResponse._send_heartbeatc C s< | j d k r8| j jd k r8d| _d| _t | _| j j d S )NT )Z_req transportrC rF asyncio TimeoutErrorrI closer) r+ r+ r, r[ z s
z$WebSocketResponse._pong_not_received)requestr' c s\ | j d k r| j S | |\}}t |I d H }|d k s<t| ||| | I d H |S r( )_payload_writer
_pre_startr= prepareAssertionError _post_startdrain)r* ra r% writerZpayload_writerrT r+ r, rd s
zWebSocketResponse.preparezCIMultiDict[str]c C s |j }d|tjd kr8td|tjdd|tjd krftd|tjdd }tj |krdd |tj
d D }|D ]}|| j kr|} qqt
d
|| j |tjd}|d krtd | d|tj}z.|rtt|d
krtd|dW n* tjk
rD td|dd Y nX tt| t }t tjdtjdtj |i} d}
d} | j r|tj } t | dd\} }
| rt!| d|
d}
|
| tj < |r|| tj < | || |
fS )NZ websocket z@No WebSocket UPGRADE hdr: {}
Can "Upgrade" only to "WebSocket".)textZupgradez No CONNECTION upgrade hdr: {}c S s g | ]}t | qS r+ )r3 strip).0protor+ r+ r,
<listcomp> s z0WebSocketResponse._handshake.<locals>.<listcomp>,u8 Client protocols %r don’t overlap server-known ones %r)Z1387zUnsupported version: zHandshake error: Fr T)isserver)r: rs Zserver_notakeover)"headersgetr ZUPGRADElowerrk r formatZ
CONNECTIONZSEC_WEBSOCKET_PROTOCOLsplitr? r ZwarningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEYlenbase64Z b64decodebinasciiErrorZ b64encodehashlibZsha1encoder Zdigestdecoder ZSEC_WEBSOCKET_ACCEPTrR ZSEC_WEBSOCKET_EXTENSIONSr r )r* ra rt r% Z
req_protocolsrm versionkeyZ
accept_valZresponse_headers
notakeoverr:
extensionsZ
enabledextr+ r+ r,
_handshake s
z WebSocketResponse._handshakec C sl |j | _ | |\}}}}| d | j| | || _|jj}|d k sRt t
|j|||d}||fS )Nr< )r: r ) rG r Z
set_statusrt updateZ force_closerR _protocolr] re r )r* ra rt r% r: r r] rh r+ r+ r, rc s
z WebSocketResponse._pre_start)ra r% rh r' c C sd || _ || _| | j}|d k s&tt|jd|d| _|j t
| j| j | j d |j
d d S )Ni loopr: F)r@ rA rY rG re r r rB r% Z
set_parserr rS rR Z
keep_alive)r* ra r% rh r r+ r+ r, rf s z WebSocketResponse._post_startc C sV | j d k rtdz| |\}}}}W n tk
rF tdd Y S X td|S d S )NzAlready startedFT)rA RuntimeErrorr r r )r* ra _r% r+ r+ r, can_prepare s
z WebSocketResponse.can_preparec C s | j S r( )rC r) r+ r+ r, closed s zWebSocketResponse.closedc C s | j S r( )rF r) r+ r+ r,
close_code
s z WebSocketResponse.close_codec C s | j S r( )r@ r) r+ r+ r, ws_protocol s z WebSocketResponse.ws_protocolc C s | j S r( )rR r) r+ r+ r, r: s zWebSocketResponse.compressc C s | j S r( )rI r) r+ r+ r, exception s zWebSocketResponse.exception )messager' c s( | j d krtd| j |I d H d S NCall .prepare() first)rA r rZ r* r r+ r+ r, rZ s
zWebSocketResponse.pingc s( | j d krtd| j |I d H d S r )rA r pongr r+ r+ r, r ! s
zWebSocketResponse.pong)datar: r' c sH | j d krtdt|ts,tdt| | j j|d|dI d H d S )Nr z data argument must be str (%r)FZbinaryr: )rA r
isinstancer3 TypeErrortypesendr* r r: r+ r+ r, send_str' s
zWebSocketResponse.send_strc sN | j d krtdt|tttfs2tdt| | j j|d|dI d H d S )Nr z#data argument must be byte-ish (%r)Tr ) rA r r bytes bytearray
memoryviewr r r r r+ r+ r,
send_bytes. s
z WebSocketResponse.send_bytes)dumps)r r: r r' c s | j |||dI d H d S )Nr )r )r* r r: r r+ r+ r, send_json5 s zWebSocketResponse.send_jsonc s4 | j r
d S | jd kr td| I d H d| _ d S )Nz Response has not been startedT)Z _eof_sentrb r r` r) r+ r+ r, write_eof> s
zWebSocketResponse.write_eof )coder )r r r' c
s | j d krtd| | j}|d k s,t| jd k rT| jsT|td | jI d H | jsd| _z8| j ||I d H | j
}|d k st| I d H W nV t j
t jfk
r d| _ Y n4 tk
r } zd| _|| _W Y dS d }~X Y nX | j rdS | j}|d k stz0tj| j| jd | I d H }W 5 Q R X W nT t j
k
rb d| _ Y n6 tk
r } zd| _|| _W Y dS d }~X Y nX |jtjkr|j| _dS d| _t | _dS dS d S )Nr r Tr\ r F) rA r rW rB re rH rC Z feed_datar
r` rb rg r^ CancelledErrorr_ rF ExceptionrI rD
async_timeoutr4 rJ rG readr r CLOSEr )r* r r readerrh excmsgr+ r+ r, r` G sV
zWebSocketResponse.close)r4 r' c
sl | j d krtd| j}|d k s$t| jd k r6td| jr`| jd7 _| jtkr\tdtS | j rjt
S zh| | _z>t
j|p| j| jd | j I d H }W 5 Q R X | W 5 | j}t |d d | _X W n tjtjfk
r d| _ Y n tk
r, d| _| I d H ttjd d Y S tk
r| } z2|j | _| j|j d I d H ttj |d W Y S d }~X Y nV t k
r } z6|| _ d| _ d| _| I d H ttj |d W Y S d }~X Y nX |j tj!krd| _ |j"| _| jsh| j#rh| I d H nZ|j tj$kr$d| _ nD|j tj%krP| j&rP| '|j"I d H q$n|j tj(krh| j&rhq$|S )
Nr z+Concurrent call to receive() is not allowedr z WebSocket connection is closed.Tr r\ r )r ))rB r rG re rH rC rE THRESHOLD_CONNLOST_ACCESSr rD r
Z
create_futurer r r4 rK r rY r^ r r_ rF r r` r r CLOSEDr r ZERRORr rI r r r rL CLOSINGZPINGrM r ZPONG)r* r4 r Zwaiterr r r+ r+ r, receive~ sn
"$zWebSocketResponse.receiver4 c s6 | |I d H }|jtjkr0td|j|j|jS )Nz.Received message {}:{!r} is not WSMsgType.TEXT)r r r ZTEXTr rw r r* r4 r r+ r+ r, receive_str s z WebSocketResponse.receive_strc s< | |I d H }|jtjkr6td|j d|jd|jS )NzReceived message :z
is not bytes)r r r ZBINARYr r r r+ r+ r,
receive_bytes s z WebSocketResponse.receive_bytes)loadsr4 )r r4 r' c s | j |dI d H }||S )Nr )r )r* r r4 r r+ r+ r, receive_json s z WebSocketResponse.receive_json)r r' c s t dd S )Nz"Cannot call .write() for websocket)r )r* r r+ r+ r, write s zWebSocketResponse.writec C s | S r( r+ r) r+ r+ r, __aiter__ s zWebSocketResponse.__aiter__c s, | I d H }|jtjtjtjfkr(t|S r( )r r r r r r StopAsyncIteration)r* r r+ r+ r, __anext__ s zWebSocketResponse.__anext__)r r' c C s | j d k r| j | d S r( )rB Z
set_exception)r* r r+ r+ r, _cancel s
zWebSocketResponse._cancel)r )r )N)N)N)N)8r. r/ r0 Z
_length_checkfloatr r1 r r3 intr> rW rY rX r[ r r rd r r r rc rf r r propertyr r r r:
BaseExceptionr r rZ r r r jsonr r r r r r` r r r r r r r r r r r
__classcell__r+ r+ rT r, r 4 s $
P
7?
r )4r^ rz r{ r} r typingr r r r r attrZ multidictr ri r abcr Zhelpersr
r Zhttpr r
r r r r r r r r logr Zstreamsr r Ztypedefsr r Zweb_exceptionsr r Z web_requestr Z web_responser __all__r sr r r+ r+ r+ r, <module> s. 0