ZZ9 @ sm d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m 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 m Z m Z m Z m Z m Z d d l m Z d d l m Z d d l m Z m Z m Z d d
l! m" Z" d Z# d Z$ e j% d d d d Gd d d Z& Gd d d e" Z' d S) N) CIMultiDict )hdrs)
call_later
set_result)
WS_CLOSED_MESSAGEWS_CLOSING_MESSAGEWS_KEYWebSocketErrorWebSocketReaderWebSocketWriter WSMessage WSMsgType
ws_ext_gen ws_ext_parse) ws_logger)FlowControlDataQueue)HTTPBadRequest
HTTPExceptionHTTPMethodNotAllowed)StreamResponseWebSocketResponseWebSocketReadyr frozenTslotsc @ s@ e Z d Z e j d e Z e j d e Z d d Z d S)r typec C s | j S)N)ok)self r
/web_ws.py__bool__ s zWebSocketReady.__bool__N)
__name__
__module__ __qualname__attrZibboolr strprotocolr! r r r r r s c s e Z d Z d d d d d d d d d d d f d
d f d d Z d
d Z d d Z d d Z d d Z f 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 d' d( Z d) d* d+ Z d) d, d- Z d d. d/ Z d d0 d1 Z d d2 e j d3 d4 Z d5 d6 Z d7 d8 d9 d: d; d< Z d d= d> Z d d d? d@ Z d d dA dB Z dC e j d d dD dE Z dF dG Z! dH dI Z" dJ dK Z# S)Lr timeoutg $@receive_timeoutN autocloseTautoping heartbeat protocolscompressc s t j d d | | _ d | _ d | _ d | _ d | _ d | _ d | _ d | _ d | _
d | _ d | _ | | _
| | _ | | _ | | _ | | _ d | _ | d k r | d | _ d | _ | | _ d S)Nstatuse Fr 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)r r) r* r+ r, r- r. r/ ) __class__r r r3 & s,
zWebSocketResponse.__init__c C sN | j d k r% | j j d | _ | j d k rJ | j j d | _ d S)N)rF cancelrD )r r r r _cancel_heartbeatA s
z#WebSocketResponse._cancel_heartbeatc C s; | j | j d k r7 t | j | j | j | _ d S)N)rJ rC r _send_heartbeatr< rD )r r r r _reset_heartbeatJ s
z"WebSocketResponse._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)
rC r8 r6 pingrF rI r _pong_not_receivedrE r< )r r r r rK Q s
z!WebSocketResponse._send_heartbeatc C sV | j d k rR | j j d k rR d | _ d | _ t j | _ | j j j d S)NTi )Z_req transportr8 r; asyncio TimeoutErrorr> close)r r r r rN Z s
! z$WebSocketResponse._pong_not_receivedc sh | j d k r | j S| j | \ } } t j | Id H} | j | | | | j Id H| S)N)_payload_writer
_pre_startr2 prepare _post_startdrain)r requestr( writerZpayload_writer)rH r r rU a s zWebSocketResponse.preparec C s | j } | j t j k r3 t | j t j g d | j t j d j j k r~ t d d j
| j t j d | j t j d j k r t d d j
| j t j d } t j | k r8d d | t j j
d D } x: | D] } | | j k r| } PqWt j d
| | j | j t j d } | d k rqt d d j
| | j t j } y> | st t j | d k rt d d j
| Wn0 t j k
rt d d j
| d Yn Xt j t j | j t j j } t t j d t j d t j d t j! | i } d }
| j" } | r| j t j# } t$ | d d \ } }
| rt% d | d d d |
}
|
| t j# <| r| | t j <| | | |
f S)NZ websocket textz@No WebSocket UPGRADE hdr: {}
Can "Upgrade" only to "WebSocket".Zupgradez No CONNECTION upgrade hdr: {}c S s" g | ] } t | j q Sr )r' strip).0protor r r
<listcomp>~ s z0WebSocketResponse._handshake.<locals>.<listcomp>,u8 Client protocols %r don’t overlap server-known ones %r1387zUnsupported version: {} zHandshake error: {!r}ZchunkedFZisserverTr/ Zserver_notakeover)ra rb rc )&headersmethodr ZMETH_GETr getZUPGRADElowerr\ r formatZ
CONNECTIONZSEC_WEBSOCKET_PROTOCOLsplitr4 r warningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEYlenbase64Z b64decodebinasciiErrorZ b64encodehashlibsha1encoder digestdecoder ZTRANSFER_ENCODINGZSEC_WEBSOCKET_ACCEPTrG ZSEC_WEBSOCKET_EXTENSIONSr r )r rX re r( Z
req_protocolsr^ versionkeyZ
accept_valZresponse_headers
notakeoverr/
extensionsZ
enabledextr r r
_handshakel sh '!
"(
z WebSocketResponse._handshakec C s | j | _ | j | \ } } } } | j | j d | j j | | j | | _ t | j
| j
j d | d | } | | f S)Nr1 r/ rw ) loopr< ry rL Z
set_statusre updateZ force_closerG r _protocolrO )r rX re r( r/ rw rY r r r rT s
z WebSocketResponse._pre_startc C si | | _ | | _ t | j d d d | j | _ | j j t | j d | j | j j
d d S)Nlimit rd rz r/ Fi ) r5 r6 r r| r< r7 r( Z
set_parserr rG Z
keep_alive)r rX r( rY r r r rV s z WebSocketResponse._post_startc C sm | j d k r t d y | j | \ } } } } Wn t k
r[ t d d SYn Xt d | Sd S)NzAlready startedFT)r6 RuntimeErrorry r r )r rX _r( r r r can_prepare s
z WebSocketResponse.can_preparec C s | j S)N)r8 )r r r r closed s zWebSocketResponse.closedc C s | j S)N)r; )r r r r
close_code s z WebSocketResponse.close_codec C s | j S)N)r5 )r r r r ws_protocol s z WebSocketResponse.ws_protocolc C s | j S)N)rG )r r r r r/ s zWebSocketResponse.compressc C s | j S)N)r> )r r r r exception s zWebSocketResponse.exceptionbc s4 | j d k r t d | j j | Id Hd S)NzCall .prepare() first)r6 r rM )r messager r r rM s zWebSocketResponse.pingc s4 | j d k r t d | j j | Id Hd S)NzCall .prepare() first)r6 r pong)r r r r r r s zWebSocketResponse.pongc se | j d k r t d t | t s@ t d t | | j j | d d d | Id Hd S)NzCall .prepare() firstz data argument must be str (%r)binaryFr/ )r6 r
isinstancer' TypeErrorr send)r datar/ r r r send_str s
zWebSocketResponse.send_strc sn | j d k r t d t | t t t f sI t d t | | j j | d d d | Id Hd S)NzCall .prepare() firstz#data argument must be byte-ish (%r)r Tr/ ) r6 r r bytes bytearray
memoryviewr r r )r r r/ r r r
send_bytes s z WebSocketResponse.send_bytesdumpsc s" | j | | d | Id Hd S)Nr/ )r )r r r/ r r r r send_json s zWebSocketResponse.send_jsonc sD | j r
d S| j d k r( t d | j Id Hd | _ d S)Nz Response has not been startedT)Z _eof_sentrS r rR )r r r r write_eof s zWebSocketResponse.write_eofcodei r c s | j d k r t d | j | j d k r] | j r] | j j t d | j Id H| j sd | _ y) | j j | | | j j
Id HWnd t j t j
f k
r d | _ Yn; t k
r } z d | _ | | _ d SWYd d } ~ Xn X| j r d Sy; t j | j d | j | j j Id H} Wd QRXWn[ t j k
rjd | _ Yn; t k
r} z d | _ | | _ d SWYd d } ~ Xn X| j t j k r| j | _ d Sd | _ t j
| _ d Sd Sd S)NzCall .prepare() firstr Ti rz F)r6 r rJ r= r8 r7 Z feed_datar rR rS rW rP CancelledErrorrQ r; Exceptionr> r9
async_timeoutr) r? r< readr r CLOSEr )r r r excmsgr r r rR sJ
zWebSocketResponse.closec s | j d k r t d x| j d k r9 t d | j rp | j d 7_ | j t k rl t d t S| j r} t Sy | j j
| _ zK t j | p | j
d | j | j j Id H} Wd QRX| j Wd | j } t | d d | _ XWn t j t j f k
r-d | _ Yn t k
r} 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 r6d | _ | j | _ | j r| j r| j Id Hnr | j t j! k rTd | _ nT | j t j" k r| j# r| j$ | j Id Hq n | j t j% k r| j# rq | SWd S) NzCall .prepare() firstz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.rz Ti r )&r7 r r= r8 r: THRESHOLD_CONNLOST_ACCESSr r9 r r< Z
create_futurer r) r@ r rL r rP r rQ r; r
r rR r
r ERRORr r> r r r rA CLOSINGZPINGrB r ZPONG)r r) r waiterr r r r receiveC sb
% & zWebSocketResponse.receivec sK | j | Id H} | j t j k rD t d j | j | j | j S)Nz.Received message {}:{!r} is not WSMsgType.TEXT)r r r ZTEXTr ri r )r r) r r r r receive_str| s z WebSocketResponse.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 r r ZBINARYr ri r )r r) r r r r
receive_bytes s z WebSocketResponse.receive_bytesloadsc s! | j d | Id H} | | S)Nr) )r )r r r) r r r r receive_json s z WebSocketResponse.receive_jsonc s t d d S)Nz"Cannot call .write() for websocket)r )r r r r r write s zWebSocketResponse.writec C s | S)Nr )r r r r __aiter__ s zWebSocketResponse.__aiter__c s< | j Id H} | j t j t j t j f k r8 t | S)N)r r r r r ZCLOSEDStopAsyncIteration)r r r r r __anext__ s zWebSocketResponse.__anext__)$r" r# r$ r3 rJ rL rK rN rU ry rT rV r propertyr r r r/ r rM r r r jsonr r r rR r r r r r r r r r r )rH r r $ s@ F
19 )r r r )(rP rm rn rp r r r% Z multidictr rZ r Zhelpersr r Zhttpr r r r
r r r
r r r logr Zstreamsr Zweb_exceptionsr r r Z web_responser __all__r sr r r r r r <module> s$ F +