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 j1dddd�G dd� d��Z2G dd� de.�Z3dS )� N)�Any�Iterable�Optional�Tuple)� CIMultiDict� )�hdrs)�AbstractStreamWriter)�
call_later�
set_result)
�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSMessage� WSMsgType�
ws_ext_gen� ws_ext_parse)� ws_logger)� EofStream�FlowControlDataQueue)� JSONDecoder� JSONEncoder)�HTTPBadRequest�
HTTPException)� BaseRequest)�StreamResponse)�WebSocketResponse�WebSocketReadyr � T)Z auto_attribs�frozen�slotsc @ s0 e Zd ZU eed< ee ed< ed�dd�ZdS )r �ok�protocol��returnc 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 ZdZddddddddd�eee eeee ee ee dd� � fd d
�Z
dd �d d
�Z dd �dd�Z dd �dd�Z
dd �dd�Zeed�� fdd� Zeedeeef d�dd�Zeeeef d�dd�Zeeedd �d d �Zeed�d 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/d0�Z"dYeee dd1�d2d3�Z#dZe ee dd1�d4d5�Z$d[e%j&d6�e'ee e(dd7�d8d9�Z)dd �d:d;�Z*d<d+d=�e e ed>�d?d@�Z+d\ee e,dA�dBdC�Z-ddD�ee edA�dEdF�Z.ddD�ee e dA�dGdH�Z/e%j0ddI�e1ee e'dJ�dKdL�Z2e ddM�dNdO�Z3d d �dPdQ�Z4e,d �dRdS�Z5e ddT�dUdV�Z6� Z7S )]r Fg $@NTr+ i @ )�timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� 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 )N�e )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 � sP | j d k r| j S | �|�\}}t� �|�I d H }| �|||� |�� I d H |S r( )�_payload_writer�
_pre_startr= �prepare� _post_start�drain)r* ra r% �writerZpayload_writerrT r+ r, rd � s
zWebSocketResponse.preparezCIMultiDict[str]c C s� |j }d|�tjd��� �� kr8td�|�tj��d��d|�tjd��� krftd�|�tj��d��d }tj |kr�dd� |tj �
d �D �}|D ]}|| j kr�|} q�q�t �
d
|| j � |�tjd�}|d kr�td |� �d��|�tj�}z.|�rtt�|��d
k�rtd|��d��W n* tjk
�rD td|��d�d �Y nX t�t�|�� t ��� ��� }t tjdtjdtj |i�} d}
d} | j �r�|�tj �} t | dd�\} }
| �r�t!| 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)�.0�protor+ r+ r, �
<listcomp>� s �z0WebSocketResponse._handshake.<locals>.<listcomp>�,u8 Client protocols %r don’t overlap server-known ones %r)Z13�8�7zUnsupported version: � zHandshake error: Fr T)�isserver)r: rr Zserver_notakeover)"�headers�getr ZUPGRADE�lowerrj r �formatZ
CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitr? r ZwarningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64Z b64decode�binascii�ErrorZ b64encode�hashlibZsha1�encoder Zdigest�decoder ZSEC_WEBSOCKET_ACCEPTrR ZSEC_WEBSOCKET_EXTENSIONSr r )r* ra rs r% Z
req_protocolsrl �version�keyZ
accept_valZresponse_headers�
notakeoverr: �
extensionsZ
enabledextr+ r+ r, �
_handshake� s� �
��
��
�
� � �� �
z WebSocketResponse._handshakec C s` |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}t |j|||d�}||fS )Nr< )r: r� )
rG r� Z
set_statusrs �updateZ force_closerR � _protocolr] r )r* ra rs r% r: r� r] rg r+ r+ r, rc � s
�z WebSocketResponse._pre_start)ra r% rg r' c C sX || _ || _| �� | j}t|jd|d�| _|j�t | j| j
| j d�� |j� d� d S )Ni ��loop�r: F)
r@ rA rY rG r r� rB r% Z
set_parserr rS rR Z
keep_alive)r* ra r% rg r� r+ r+ r, re � s �z WebSocketResponse._post_startc C sV | j d k rtd��z| �|�\}}}}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 �N�Call .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 krtd��t|t�s,tdt|� ��| j j|d|d�I d H d S )Nr� z data argument must be str (%r)F�Zbinaryr: )rA r� �
isinstancer3 � TypeError�type�send�r* r� r: r+ r+ r, �send_str' s
zWebSocketResponse.send_strc � sN | j d krtd��t|tttf�s2tdt|� ��| j j|d|d�I 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 ||�|d�I 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}| jd k rH| jsH|�td� | jI d H | j�s�d| _z,| j �||�I d H | j }|�
� I d H W nV t j t j
fk
r� d| _� Y n4 tk
r� } zd| _|| _W Y �dS d }~X Y nX | jr�dS | j}z0tj| j| jd�� |�� I d H }W 5 Q R X W nT t j k
�r: d| _� Y n6 tk
�rn } zd| _|| _W Y �dS d }~X Y nX |jtjk�r�|j| _dS d| _t �
� | _dS dS d S )Nr� r Tr\ r� F)rA r� rW rB rH rC Z feed_datar
r` rb rf r^ �CancelledErrorr_ rF � ExceptionrI rD �
async_timeoutr4 rJ rG �readr� r �CLOSEr� )r* r� r� �readerrg �exc�msgr+ r+ r, r` G sP
zWebSocketResponse.close)r4 r' c
� s` | j d krtd��| j}| jd k r*td��| jrT| jd7 _| jtkrPtd��tS | jr^t S zh|�
� | _z>t j
|px| 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
�rp } z2|j| _| j|jd �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 k�rd| _|j!| _| j�s\| j"�r\| �� I d H nZ|j tj#k�rd| _nD|j tj$k�rD| j%�rD| �&|j!�I d H qn|j tj'k�r\| j%�r\q|S )
Nr� z+Concurrent call to receive() is not allowedr z WebSocket connection is closed.Tr� r\ r� )r� )(rB r� rG 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~ sl
�
"$zWebSocketResponse.receive�r4 c � s6 | � |�I d H }|jtjkr0td�|j|j���|jS )Nz.Received message {}:{!r} is not WSMsgType.TEXT)r� r� r ZTEXTr� rv r� �r* r4 r� r+ r+ r, � receive_str� s ��z WebSocketResponse.receive_strc � s< | � |�I d H }|jtjkr6td|j� d|j�d���|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 |d�I d H }||�S )Nr� )r� )r* r� r4 r� r+ r+ r, � receive_json� s z WebSocketResponse.receive_json)r� r' c � s t d��d 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_check�floatr r1 r r3 �intr> rW rY rX r[ r r rd r r� r rc re 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^ ry rz r| r� �typingr r r r r� �attrZ multidictr rh 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