File: //home/arjun/projects/env/lib64/python3.10/site-packages/aiohttp/__pycache__/web_ws.cpython-310.pyc
o
+wewF � @ sx U d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
m Z m Z d dl
Z
d dlmZ ddlmZ ddlmZ ddlmZmZ ddlmZmZmZmZm Z m 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* dd l+m,Z,m-Z- dd
l.m/Z/ ddl0m1Z1 ej2dkr�d dl Z3nd dl3Z3dZ4dZ5ee6 e7d< e
j8dddd�G dd� d��Z9G dd� de1�Z:dS )� N)�Any�Final�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)� JSONDecoder� JSONEncoder)�HTTPBadRequest�
HTTPException)� BaseRequest)�StreamResponse)� � )�WebSocketResponse�WebSocketReadyr � �THRESHOLD_CONNLOST_ACCESST)� auto_attribs�frozen�slotsc @ s0 e Zd ZU eed< ee ed< defdd�ZdS )r% �ok�protocol�returnc C � | j S �N)r+ ��self� r2 �G/home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/web_ws.py�__bool__6 � zWebSocketReady.__bool__N)�__name__�
__module__� __qualname__�bool�__annotations__r �strr4 r2 r2 r2 r3 r% 1 s
r% c s� e Zd ZdZddddddddd�ded ee d
ed ed ee d
ee dede ddf� fdd�Z
d`dd�Z d`dd�Z d`dd�Z
d`dd�Zdedef� fd d � Zdeded eeef fd d �Zdedeeef fd!d"�Zded#ed$eddfd%d&�Zdedefd'd(�Zedefd)d*��Zedee fd+d,��Zedee fd-d.��Z edefd/d0��Z dad1ed2e de fd3d4�Z dee fd5d6�Z!dbd8e"ddfd9d:�Z#dbd8e"ddfd;d<�Z$dad=edee ddfd>d?�Z%dad=e"dee ddfd@dA�Z& dae'j(dB�d=e dee dCe)ddfdDdE�Z*d`dFdG�Z+e,j-d7dH�dIe d8e"defdJdK�Z.dadee de/fdLdM�Z0ddN�dee defdOdP�Z1ddN�dee de"fdQdR�Z2e'j3ddS�dTe4dee de fdUdV�Z5d=e"ddfdWdX�Z6dcdYdZ�Z7de/fd[d\�Z8d]e ddfd^d_�Z9� Z:S )dr$ Fg $@NTr2 i @ )�timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� max_msg_sizer< r= r> r? r@ rA rB rC r- c s� t � jdd� || _d | _d | _d | _d| _d| _d| _d | _ d | _
d | _ d | _ || _
|| _|| _|| _|| _d | _|d urC|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) r1 r< r= r> r? r@ rA rB rC �� __class__r2 r3 rG > s.
zWebSocketResponse.__init__c C s< | j d ur
| j �� d | _ | jd ur | j�� d | _d S d S r/ )rZ �cancelrX r0 r2 r2 r3 �_cancel_heartbeatb s
�z#WebSocketResponse._cancel_heartbeatc C sT | � � | jd ur(| jd usJ �t| j| j| j| jd ur!| jjjndd�| _d S d S �Nr& )�timeout_ceil_threshold) r` rW rP r �_send_heartbeat�_req� _protocol�_timeout_ceil_thresholdrX r0 r2 r2 r3 �_reset_heartbeatk s
� ��z"WebSocketResponse._reset_heartbeatc C s| | j d ur:| js<| jd usJ �| j�| j�� � | jd ur"| j�� t| j | j
| j| j d ur3| j j j
ndd�| _d S d S d S ra )rW rL rP � create_taskrJ �pingrZ r_ r �_pong_not_receivedrY rd re rf r0 r2 r2 r3 rc y s
� ��z!WebSocketResponse._send_heartbeatc C sF | j d ur | j jd ur!d| _tj| _t�� | _| j j� � d S d S d S )NT)
rd � transportrL r �ABNORMAL_CLOSURErO �asyncio� TimeoutErrorrR �closer0 r2 r2 r3 rj � s
�z$WebSocketResponse._pong_not_received�requestc � s^ �| j d ur | j S | �|�\}}t� �|�I d H }|d us J �| �|||� |�� I d H |S r/ )�_payload_writer�
_pre_startrF �prepare� _post_start�drain)r1 rp r, �writer�payload_writerr] r2 r3 rs � s �
zWebSocketResponse.preparezCIMultiDict[str]c C s� |j }d|�tjd��� �� kr td�|�tj��d��d|�tjd��� vr3td�|�tj��d��d }tj |v r]dd� |tj �
d �D �}|D ] }|| j v rT|} n qIt �
d
|| j � |�tjd�}|d vrptd |� �d��|�tj�}z|r�tt�|��d
kr�td|��d��W n tjy� td|��d�d �w t�t�|�� t ��� ��� }t tjdtjdtj |i�} d}
d} | j r�|�tj �} t | dd�\} }
| r�t!| d|
d�}
|
| tj < |r�|| tj < | || |
fS )N� websocket� z@No WebSocket UPGRADE hdr: {}
Can "Upgrade" only to "WebSocket".)�text�upgradez No CONNECTION upgrade hdr: {}c S s g | ]}t |�� ��qS r2 )r; �strip)�.0�protor2 r2 r3 �
<listcomp>� s
��z0WebSocketResponse._handshake.<locals>.<listcomp>�,u8 Client protocols %r don’t overlap server-known ones %r)�13�8�7zUnsupported version: � zHandshake error: Fr T)�isserver)rB r� �server_notakeover)"�headers�getr
�UPGRADE�lowerr| r �format�
CONNECTION�SEC_WEBSOCKET_PROTOCOL�splitrH r �warning�SEC_WEBSOCKET_VERSION�SEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder �SEC_WEBSOCKET_ACCEPTr[ �SEC_WEBSOCKET_EXTENSIONSr r )r1 rp r� r, �
req_protocolsr~ �version�key�
accept_val�response_headers�
notakeoverrB �
extensions�
enabledextr2 r2 r3 �
_handshake� s� �
��
�
�� ������ �
�z WebSocketResponse._handshakec C sl |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}|d us)J �t |j|||d�}||fS )NrD )rB r� )
rP r� �
set_statusr� �update� force_closer[ re rk r )r1 rp r� r, rB r� rk rv r2 r2 r3 rr � s
�z WebSocketResponse._pre_startr, rv c C sd || _ || _| �� | j}|d usJ �t|jd|d�| _|j�t | j| j
| j d�� |j� d� d S )Ni )�loop�rB F)
rI rJ rg rP r re rK r, �
set_parserr r\ r[ �
keep_alive)r1 rp r, rv r� r2 r2 r3 rt s �z WebSocketResponse._post_startc C sP | j d ur td��z | �|�\}}}}W n ty" tdd � Y S w td|�S )NzAlready startedFT)rJ � RuntimeErrorr� r r% )r1 rp �_r, r2 r2 r3 � can_prepare s
�
z WebSocketResponse.can_preparec C r. r/ )rL r0 r2 r2 r3 �closed � zWebSocketResponse.closedc C r. r/ )rO r0 r2 r2 r3 �
close_code$ r� z WebSocketResponse.close_codec C r. r/ )rI r0 r2 r2 r3 � ws_protocol( r� z WebSocketResponse.ws_protocolc C r. r/ )r[ r0 r2 r2 r3 rB , r� zWebSocketResponse.compress�name�defaultc C s0 | j }|du r |S |j}|du r|S |�||�S )z}Get optional transport information.
If no value associated with ``name`` is found, ``default`` is returned.
N)rJ rk �get_extra_info)r1 r� r� rv rk r2 r2 r3 r� 0 s z WebSocketResponse.get_extra_infoc C r. r/ )rR r0 r2 r2 r3 � exception= r5 zWebSocketResponse.exception� �messagec � �* �| j d u r
td��| j �|�I d H d S �N�Call .prepare() first)rJ r� ri �r1 r� r2 r2 r3 ri @ s �
zWebSocketResponse.pingc � r� r� )rJ r� �pongr� r2 r2 r3 r� E s �
zWebSocketResponse.pong�datac � sJ �| j d u r
td��t|t�stdt|� ��| j j|d|d�I d H d S )Nr� z data argument must be str (%r)F��binaryrB )rJ r� �
isinstancer; � TypeError�type�send�r1 r� rB r2 r2 r3 �send_strK s �
zWebSocketResponse.send_strc � sP �| j d u r
td��t|tttf�stdt|� ��| j j|d|d�I d H d S )Nr� z#data argument must be byte-ish (%r)Tr� ) rJ r� r� �bytes� bytearray�
memoryviewr� r� r� r� r2 r2 r3 �
send_bytesR s �
z WebSocketResponse.send_bytes)�dumpsr� c � s �| j ||�|d�I d H d S )Nr� )r� )r1 r� rB r� r2 r2 r3 � send_jsonY s � zWebSocketResponse.send_jsonc � s6 �| j rd S | jd u rtd��| �� I d H d| _ d S )Nz Response has not been startedT)� _eof_sentrq r� ro r0 r2 r2 r3 � write_eofb s �
zWebSocketResponse.write_eof)�coder� r� c
� s� �| j d u r
td��| �� | j}|d usJ �| jd ur+| js+|�td� | jI d H | js�d| _z | j �||�I d H | j }|d usEJ �|�
� I d H W n) t j t j
fy] tj| _� tyv } ztj| _|| _W Y d }~dS d }~ww | jr|dS | j}|d us�J �z)t�| j�4 I d H � |�� I d H }W d �I d H n 1 I d H s�w Y W n& t j y� tj| _� ty� } ztj| _|| _W Y d }~dS d }~ww |jtjkr�|j| _dS tj| _t �
� | _dS dS )Nr� r TF) rJ r� r` rK rQ rL � feed_datar ro rq ru rm �CancelledErrorrn r rl rO � ExceptionrR rM �
async_timeoutr< rS �readr� r �CLOSEr� )r1 r� r� �readerrv �exc�msgr2 r2 r3 ro k sd �
�� (����
zWebSocketResponse.closec
� s� �| j d u r
td��| j}|d usJ � | jd ur td��| jr2| jd7 _| jtkr0td��tS | jr7t S zO|�
� | _z;t � |pD| j
�4 I d H � | j �� I d H }W d �I d H n 1 I d H scw Y | �� W | j}t|d� d | _n | j}t|d� d | _w W ny tjtjfy� tj| _� ty� tj| _| �� I d H ttj d d � Y S t y� } z|j | _| j|j d�I d H ttj |d �W Y d }~S d }~w t y� } z || _!d| _tj| _| �� I d H ttj |d �W Y d }~S d }~ww |j"tj#k�r d| _|j$| _| j�s | j%�r | �� I d H |S |j"tj&k�r+d| _|S |j"tj'k�r@| j(�r@| �)|j$�I d H q|j"tj*k�rL| j(�rLq|S )Nr� Tz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.)r� )+rK r� rP rQ rL rN r' r rM r �
create_futurer� r< rT r� rg r
rm r� rn r rl rO r �OKro r r �CLOSEDr r� �ERRORr� rR r� r� r� rU �CLOSING�PINGrV r� �PONG)r1 r< r� r� �waiterr� r2 r2 r3 �receive� s~ �
(�
�
��� ��zWebSocketResponse.receive�r< c � s> �| � |�I d H }|jtjkrtd�|j|j���tt|j�S )Nz.Received message {}:{!r} is not WSMsgType.TEXT) r� r� r �TEXTr� r� r� r r; �r1 r< r� r2 r2 r3 � receive_str� s � �� z WebSocketResponse.receive_strc � sD �| � |�I d H }|jtjkr td|j� d|j�d���tt|j�S )NzReceived message �:z
is not bytes)r� r� r �BINARYr� r� r r� r� r2 r2 r3 �
receive_bytes� s
� z WebSocketResponse.receive_bytes)�loadsr< r� c � s �| j |d�I d H }||�S )Nr� )r� )r1 r� r< r� r2 r2 r3 � receive_json� s �z WebSocketResponse.receive_jsonc � s
�t d��)Nz"Cannot call .write() for websocket)r� )r1 r� r2 r2 r3 �write� s �zWebSocketResponse.writec C s | S r/ r2 r0 r2 r2 r3 � __aiter__� s zWebSocketResponse.__aiter__c � s. �| � � I d H }|jtjtjtjfv rt�|S r/ )r� r� r r� r� r� �StopAsyncIteration)r1 r� r2 r2 r3 � __anext__� s
�zWebSocketResponse.__anext__r� c C s | j d ur
| j �|� d S d S r/ )rK �
set_exception)r1 r� r2 r2 r3 �_cancel s
�zWebSocketResponse._cancel)r- Nr/ )r� )r- r$ );r6 r7 r8 �
_length_check�floatr r9 r r; �intrG r` rg rc rj r r rs r r� r rr rt r% r� �propertyr� r� r� rB r r� �
BaseExceptionr� r� ri r� r� r� �jsonr� r r� r� r r� ro r r� r� r� r� r r� r� r� r� r� �
__classcell__r2 r2 r] r3 r$ : s� ������� �
� �
$
�
�U���
�
�����
�
7 =
���
�
r$ );rm r� r� r� r� �sys�typingr r r r r r �attr� multidictr ry r
�abcr �helpersr r
�httpr r r r r r r r r r r �logr �streamsr r �typedefsr r �web_exceptionsr r � web_requestr � web_responser! � version_infor� �__all__r' r� r: �sr% r$ r2 r2 r2 r3 �<module> s6
4