Edit File: web_ws.cpython-37.pyc
B �k�akC � @ s\ U d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z m Z d dl Z d dl Z d dl mZ ddlmZ ddlmZ ddlmZmZ ddlmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! dd l"m#Z# dd l$m%Z%m&Z& dd l'm(Z(m)Z)m*Z* dd l+m,Z,m-Z- dd l.m/Z/ ddl0m1Z1 dZ2dZ3e(e4 e5d< e j6dddd�G dd� d��Z7G dd� de1�Z8dS )� N)�Any�Iterable�Optional�Tuple�cast)� CIMultiDict� )�hdrs)�AbstractStreamWriter)� call_later� set_result) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSCloseCode� WSMessage� WSMsgType� ws_ext_gen� ws_ext_parse)� ws_logger)� EofStream�FlowControlDataQueue)�Final� JSONDecoder� JSONEncoder)�HTTPBadRequest� HTTPException)� BaseRequest)�StreamResponse)�WebSocketResponse�WebSocketReadyr � �THRESHOLD_CONNLOST_ACCESST)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/python37/lib64/python3.7/site-packages/aiohttp/web_ws.py�__bool__1 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 dVe dd,�d-d.�Z!dWe dd,�d/d0�Z"dXeee dd1�d2d3�Z#dYe ee dd1�d4d5�Z$dZe%j&d6�e'ee e(dd7�d8d9�Z)dd �d:d;�Z*e+j,d+d<�e e ed=�d>d?�Z-d[ee e.d@�dAdB�Z/ddC�ee ed@�dDdE�Z0ddC�ee e d@�dFdG�Z1e%j2ddH�e3ee e'dI�dJdK�Z4e ddL�dMdN�Z5d d �dOdP�Z6e.d �dQdR�Z7e ddS�dTdU�Z8� Z9S )\r" Fg $@NTr, i @ )�timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� max_msg_size) r5 r6 r7 r8 r9 r: 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 )�statusFr 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+ r5 r6 r7 r8 r9 r: r; r< )� __class__r, r- r@ 9 s. zWebSocketResponse.__init__)r* c C s8 | j d k r| j �� d | _ | jd k r4| j�� d | _d S )N)rS �cancelrQ )r+ r, r, r- �_cancel_heartbeat] s z#WebSocketResponse._cancel_heartbeatc C s8 | � � | jd k r4| jd k s t�t| j| j| j�| _d S )N)rX rP rI �AssertionErrorr �_send_heartbeatrQ )r+ r, r, r- �_reset_heartbeatf s z"WebSocketResponse._reset_heartbeatc C s\ | j d k rX| jsX| jd k s t�| j�| j�� � | jd k rD| j�� t | j | j | j�| _d S )N) rP rE rI rY Z create_taskrC �pingrS rW r �_pong_not_receivedrR )r+ r, r, r- rZ o s z!WebSocketResponse._send_heartbeatc C s> | j d k r:| j jd k r:d| _tj| _t�� | _| j j� � d S )NT) Z_req� transportrE r �ABNORMAL_CLOSURErH �asyncio� TimeoutErrorrK �close)r+ r, r, r- r] } 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 )N)�_payload_writer� _pre_startr? �preparerY � _post_start�drain)r+ rc r) �writerZpayload_writer)rV r, r- rf � 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 �}x,|D ]}|| j kr�|}P q�W t � d || j � |�tjd�}|d kr�td |� �d��|�tj�}y.|�r tt�|��d k�rtd|��d��W n* tjk �rF 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, )r4 �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; rt Zserver_notakeover)"�headers�getr ZUPGRADE�lowerrl r �formatZ CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitrA r �warningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder ZSEC_WEBSOCKET_ACCEPTrT ZSEC_WEBSOCKET_EXTENSIONSr r )r+ rc ru r) Z req_protocolsrn �version�keyZ accept_valZresponse_headers� notakeoverr; � extensionsZ enabledextr, r, r- � _handshake� sf z WebSocketResponse._handshakec C sl |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}|d k sRt �t |j|||d�}||fS )Nr= )r; r� ) rI r� Z set_statusru �updateZ force_closerT � _protocolr^ rY r )r+ rc ru r) r; r� r^ ri r, r, r- re � s z WebSocketResponse._pre_start)rc r) ri r* c C sd || _ || _| �� | j}|d k s&t�t|jd|d�| _|j� t | j| j | j d�� |j� d� d S )Ni )�loop)r; F)rB rC r[ rI rY r r� rD r) Z set_parserr rU rT Z keep_alive)r+ rc r) ri r� r, r, r- rg � s z WebSocketResponse._post_startc C sR | j d k rtd��y| �|�\}}}}W n tk rB tdd �S X td|�S d S )NzAlready startedFT)rC � RuntimeErrorr� r r# )r+ rc �_r) r, r, r- � can_prepare s z WebSocketResponse.can_preparec C s | j S )N)rE )r+ r, r, r- �closed s zWebSocketResponse.closedc C s | j S )N)rH )r+ r, r, r- � close_code s z WebSocketResponse.close_codec C s | j S )N)rB )r+ r, r, r- � ws_protocol s z WebSocketResponse.ws_protocolc C s | j S )N)rT )r+ r, r, r- r; s zWebSocketResponse.compressc C s | j S )N)rK )r+ r, r, r- � exception! s zWebSocketResponse.exception� )�messager* c � s( | j d krtd��| j �|�I d H d S )NzCall .prepare() first)rC r� r\ )r+ r� r, r, r- r\ $ s zWebSocketResponse.pingc � s( | j d krtd��| j �|�I d H d S )NzCall .prepare() first)rC r� �pong)r+ r� 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 )NzCall .prepare() firstz data argument must be str (%r)F)�binaryr; )rC r� � isinstancer4 � 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 )NzCall .prepare() firstz#data argument must be byte-ish (%r)T)r� r; ) rC r� r� �bytes� bytearray� memoryviewr� r� r� )r+ r� r; r, r, r- � send_bytes6 s z WebSocketResponse.send_bytes)�dumps)r� r; r� r* c � s | j ||�|d�I d H d S )N)r; )r� )r+ r� r; r� r, r, r- � send_json= 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_sentrd r� rb )r+ r, r, r- � write_eofF 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 | j�s�d| _y8| j � ||�I d H | j }|d k s�t�|� � I d H W nT t j t jfk r� tj| _� Y n0 tk r� } ztj| _|| _dS d }~X Y nX | jr�dS | j}|d k �st�y8t�| j�4 I d H � |�� I d H }W d Q I d H R X W nR t j k �rh tj| _� Y n2 tk �r� } ztj| _|| _dS d }~X Y nX |jtjk�r�|j | _dS tj| _t �� | _dS dS d S )NzCall .prepare() firstr TF) rC r� rX rD rY rJ rE Z feed_datar rb rd rh r` �CancelledErrorra r r_ rH � ExceptionrK rF � async_timeoutr5 rL �readr� r �CLOSEr� )r+ r� r� �readerri �exc�msgr, r, r- rb O sV " zWebSocketResponse.close)r5 r* c � sn | j d krtd��| j}|d k s$t��xB| jd k r:td��| jrd| jd7 _| jtkr`td��tS | j rnt S yn|� � | _zFt � |p�| j�4 I d H � | j �� I d H }W d Q I d H R X | �� W d | j}t|d� d | _X W n� tjtjfk �r tj| _� Y n� tk �r8 tj| _| �� I d H tt j d d �S t k �r� } z*|j | _| j|j d�I d H tt j |d �S d }~X Y nP t!k �r� } z0|| _"d| _ tj| _| �� I d H tt j |d �S d }~X Y nX |j#t j$k�r d| _ |j%| _| j�sf| j&�rf| �� I d H nZ|j#t j'k�r"d| _ nD|j#t j(k�rN| j)�rN| �*|j%�I d H q(n|j#t j+k�rf| j)�rfq(|S d S )NzCall .prepare() firstz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.T)r� ),rD r� rI rY rJ rE rG r% r rF r Z create_futurer� r5 rM r� r[ r r` r� ra r r_ rH r �OKrb r r �CLOSEDr r� �ERRORr� rK r� r� r� rN �CLOSINGZPINGrO r� ZPONG)r+ r5 r� r� �waiterr� r, r, r- �receive� sj zWebSocketResponse.receive)r5 c � s< | � |�I d H }|jtjkr0td�|j|j���tt|j�S )Nz.Received message {}:{!r} is not WSMsgType.TEXT) r� r� r ZTEXTr� rx r� r r4 )r+ r5 r� r, r, r- � receive_str� s z WebSocketResponse.receive_strc � sB | � |�I d H }|jtjkr6td|j� d|j�d���tt|j�S )NzReceived message �:z is not bytes)r� r� r ZBINARYr� r� r r� )r+ r5 r� r, r, r- � receive_bytes� s z WebSocketResponse.receive_bytes)�loadsr5 )r� r5 r* c � s | j |d�I d H }||�S )N)r5 )r� )r+ r� r5 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 )Nr, )r+ r, r, r- � __aiter__� s zWebSocketResponse.__aiter__c � s, | � � I d H }|jtjtjtjfkr(t�|S )N)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 )N)rD � set_exception)r+ r� r, r, r- �_cancel� s zWebSocketResponse._cancel)r� )r� )N)N)N)N):r/ r0 r1 Z _length_check�floatr r2 r r4 �intr@ rX r[ rZ r] r r rf r r� r re rg r# r� �propertyr� r� r� r; � BaseExceptionr� r� r\ r� r� r� �jsonr� r r r� r� r r� rb r r� r� r� r� r r� r� r� r� r� � __classcell__r, r, )rV r- r" 5 s` " S 7= r" )9r` r| r~ r� r� �typingr r r r r r� �attrZ multidictr rj r �abcr Zhelpersr r �httpr r r r r r r r r r r �logr Zstreamsr r Ztypedefsr r r Zweb_exceptionsr r Z web_requestr Z web_responser! �__all__r% r� r3 �sr# r" r, r, r, r- �<module> s. 4