File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/web_ws.cpython-310.pyc
o
���g�W � @ s� U d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m
Z
m Z m Z m
Z
d dlZd dlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lm Z m Z m Z dd l m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z* dd l+m,Z, dd
l-m.Z. ddl/m0Z0 ddl1m2Z2m3Z3 ddl4m5Z5m6Z6 ddl7m8Z8 ddl9m:Z: ej;dkr�d dl Z<nd dl<Z<dZ=dZ>ee? e@d< ejAdddd�G dd� d��ZBG dd � d e:�ZCdS ) � N)�Any�Final�Iterable�Optional�Tuple�Union�cast)� CIMultiDict� )�hdrs)�WebSocketDataQueue)�
DEFAULT_LIMIT)�AbstractStreamWriter)�WSMessageTypeError)�calculate_timeout_when�
set_exception�
set_result) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSCloseCode� WSMessage� WSMsgType�
ws_ext_gen� ws_ext_parse)�_INTERNAL_RECEIVE_TYPES)� ws_logger)� EofStream)� 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)r0 ��self� r7 �9/usr/local/lib/python3.10/dist-packages/aiohttp/web_ws.py�__bool__: � zWebSocketReady.__bool__N)�__name__�
__module__� __qualname__�bool�__annotations__r �strr9 r7 r7 r7 r8 r* 5 s
r* c s~ e Zd ZU dZeed< dZee ed< dZ ee
ed< dZ ee ed< dZ
eed< dZeed< d Zeed
< dZee ed < dZeej ed < dZeed
< dZeejd ed< dZee ed< dZeed< dZ eej ed< dZ eej ed< dZ eej d ed< dddddddde!d� dedee d ed ed ee d e"e d ed!ed"ed#df� fd$d%�Z#d�d&d'�Z$d�d(d)�Z%d�d*d+�Z&d�d,d-�Z'd�d0d1�Z(d�d2d3�Z)d4ed#dfd5d6�Z*d�d7d8�Z+d9e,d#e-f� fd:d;� Z.d9e,d#e/d<ee eef fd=d>�Z0d9e,d#e/ee e
f fd?d@�Z1d9e,dAee dBe
d#dfdCdD�Z2d9e,d#e3fdEdF�Z4e5d#efdGdH��Z6e5d#ee fdIdJ��Z7e5d#ee fdKdL��Z8e5d#e9eef fdMdN��Z:d�dOedPe;d#e;fdQdR�Z<d#ee fdSdT�Z=d�dVe>d#dfdWdX�Z?d�dVe>d#dfdYdZ�Z@ d�dVe>d[eAd ee d#dfd\d]�ZBd�d^ed ee d#dfd_d`�ZCd�d^e>d ee d#dfdadb�ZD d�eEjFdc�d^e;d ee ddeGd#dfdedf�ZHd�dgdh�ZIeJjKdUddi�djedVe>dked#efdldm�ZLdjeJd#dfdndo�ZMdjeJd#dfdpdq�ZNd�drds�ZOd�dee d#ePfdtdu�ZQddv�dee d#efdwdx�ZRddv�dee d#e>fdydz�ZSeEjTdd{�d|eUdee d#e;fd}d~�ZVd^e>d#dfdd��ZWd�d�d��ZXd#ePfd�d��ZYd4ed#dfd�d��ZZ� Z[S )�r) F�
_length_checkN� _ws_protocol�_writer�_reader�_closed�_closingr �
_conn_lost� _close_code�_loop�_waiting� _close_wait�
_exceptiong �_heartbeat_when�
_heartbeat_cb�_pong_response_cb�
_ping_taskg $@Tr7 i @ ) �timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� max_msg_size� writer_limitrQ rR rS rT rU rV rW rX rY r2 c
sZ t � jdd� || _|| _|| _|| _|| _|| _|d ur"|d | _|| _ || _
| | _ d S )N�e )�statusg @) �super�__init__�
_protocols�_timeout�_receive_timeout�
_autoclose� _autoping�
_heartbeat�_pong_heartbeat� _compress�
_max_msg_size�
_writer_limit)
r6 rQ rR rS rT rU rV rW rX rY �� __class__r7 r8 r] Q s
zWebSocketResponse.__init__c C sD | � � | jd ur| j�� d | _| jd ur | j�� d | _d S d S r4 )�_cancel_pong_response_cbrN �cancelrP r5 r7 r7 r8 �_cancel_heartbeatk s
�z#WebSocketResponse._cancel_heartbeatc C s" | j d ur| j �� d | _ d S d S r4 )rO rk r5 r7 r7 r8 rj t s
�z*WebSocketResponse._cancel_pong_response_cbc C s� | j d u rd S | �� | j}|d ur|jjnd}| j}|d us!J �|�� }t|| j |�}|| _| j d u r>|�
|| j �| _ d S d S )Nr+ ) rc rj �_req� _protocol�_timeout_ceil_thresholdrI �timer rM rN �call_at�_send_heartbeat)r6 �req�timeout_ceil_threshold�loop�now�whenr7 r7 r8 �_reset_heartbeaty s
�
�z"WebSocketResponse._reset_heartbeatc C s� d | _ | j}|d ur| jd usJ �|�� }|| jk r%|�| j| j�| _ d S | j}|d ur0|jj nd}t
|| j |�}| � � |�|| j
�| _| j�dtj�}tjdkr[tj||dd�}n|�|�}|�� so|| _|�| j� d S | �|� d S )Nr+ � )r' � T)ru � eager_start)rN rI rC rp rM rq rr rm rn ro r rd rj �_pong_not_receivedrO �
send_framer �PING�sys� version_info�asyncio�Task� create_task�donerP �add_done_callback�_ping_task_done)r6 ru rv rs rt rw �coro� ping_taskr7 r7 r8 rr � s.
��
z!WebSocketResponse._send_heartbeat�task�asyncio.Task[None]c C s( |� � s|�� }r| �|� d| _dS )z*Callback for when the ping task completes.N)� cancelled� exception�_handle_ping_pong_exceptionrP )r6 r� �excr7 r7 r8 r� � s
z!WebSocketResponse._ping_task_donec C s0 | j d ur| j jd ur| �t�� � d S d S d S r4 )rm � transportr� r� � TimeoutErrorr5 r7 r7 r8 r| � s �z$WebSocketResponse._pong_not_receivedr� c C sb | j rdS | �� | �tj� || _| jr+| js-| jdur/| j� t
t j |d�d� dS dS dS dS )z5Handle exceptions raised during ping/pong processing.Nr )
rE � _set_closed�_set_code_close_transportr �ABNORMAL_CLOSURErL rJ rF rD � feed_datar r �ERROR�r6 r� r7 r7 r8 r� � s �z-WebSocketResponse._handle_ping_pong_exceptionc C s d| _ | �� dS )zdSet the connection to closed.
Cancel any heartbeat timers and set the closed flag.
TN)rE rl r5 r7 r7 r8 r� � s z WebSocketResponse._set_closed�requestc � s^ �| j d ur | j S | �|�\}}t� �|�I d H }|d us J �| �|||� |�� I d H |S r4 )�_payload_writer�
_pre_startr\ �prepare� _post_start�drain)r6 r� r1 �writer�payload_writerrh r7 r8 r� � 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 r7 )r@ �strip)�.0�protor7 r7 r8 �
<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)rW r� �server_notakeover)"�headers�getr �UPGRADE�lowerr� r# �format�
CONNECTION�SEC_WEBSOCKET_PROTOCOL�splitr^ r �warning�SEC_WEBSOCKET_VERSION�SEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder �SEC_WEBSOCKET_ACCEPTre �SEC_WEBSOCKET_EXTENSIONSr r )r6 r� r� r1 �
req_protocolsr� �version�key�
accept_val�response_headers�
notakeoverrW �
extensions�
enabledextr7 r7 r8 �
_handshake� s� �
��
�
�� ������ �
�z WebSocketResponse._handshakec C sp |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}|d us)J �t |j|||| j
d�}||fS )NrZ )rW r� �limit) rI r� �
set_statusr� �update� force_closere rn r� r rg )r6 r� r� r1 rW r� r� r� r7 r7 r8 r� . s
�z WebSocketResponse._pre_startr1 r� c C sh || _ || _| �� | j}|d usJ �t|jd|d�| _|j�t | j| j
t | j �d�� |j�
d� d S )Ni )ru �rW F)rB rC rx rI r rn rD r1 �
set_parserr rf r> re �
keep_alive)r6 r� r1 r� ru r7 r7 r8 r� C s ��z WebSocketResponse._post_startc C sP | j d ur td��z | �|�\}}}}W n ty" tdd � Y S w td|�S )NzAlready startedFT)rC � RuntimeErrorr� r$ r* )r6 r� �_r1 r7 r7 r8 � can_prepareV s
�
z WebSocketResponse.can_preparec C r3 r4 )rE r5 r7 r7 r8 �closed` � zWebSocketResponse.closedc C r3 r4 )rH r5 r7 r7 r8 �
close_coded r� z WebSocketResponse.close_codec C r3 r4 )rB r5 r7 r7 r8 � ws_protocolh r� z WebSocketResponse.ws_protocolc C r3 r4 )re r5 r7 r7 r8 rW l 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)rC r� �get_extra_info)r6 r� r� r� r� r7 r7 r8 r� p s z WebSocketResponse.get_extra_infoc C r3 r4 )rL r5 r7 r7 r8 r� } r: zWebSocketResponse.exceptionry �messagec � �. �| j d u r
td��| j �|tj�I d H d S �N�Call .prepare() first)rC r� r} r r~ �r6 r� r7 r7 r8 �ping� s �
zWebSocketResponse.pingc � r� r� )rC r� r} r �PONGr� r7 r7 r8 �pong� s �
zWebSocketResponse.pong�opcodec � s. �| j du r
td��| j �|||�I dH dS )z Send a frame over the websocket.Nr� )rC r� r} )r6 r� r� rW r7 r7 r8 r} � s �
z WebSocketResponse.send_frame�datac � sR �| j d u r
td��t|t�stdt|� ��| j j|�d�tj |d�I d H d S )Nr� z data argument must be str (%r)zutf-8r� )
rC r� �
isinstancer@ � TypeError�typer} r� r �TEXT�r6 r� rW r7 r7 r8 �send_str� s �
�zWebSocketResponse.send_strc � sR �| j d u r
td��t|tttf�stdt|� ��| j j|t j
|d�I d H d S )Nr� z#data argument must be byte-ish (%r)r� ) rC r� r� �bytes� bytearray�
memoryviewr� r� r} r �BINARYr� r7 r7 r8 �
send_bytes� s �
z WebSocketResponse.send_bytes)�dumpsr� c � s �| j ||�|d�I d H d S )Nr� )r� )r6 r� rW r� r7 r7 r8 � send_json� s � zWebSocketResponse.send_jsonc � s6 �| j rd S | jd u rtd��| �� I d H d| _ d S )Nz Response has not been startedT)� _eof_sentr� r� �closer5 r7 r7 r8 � write_eof� s �
zWebSocketResponse.write_eof)�coder� r� r� r� c
� s� �| j du r
td��| jrdS | �� z | j �||�I dH | j}|dus'J �|r0|�� I dH W n- tjtj fyC | �
t j � � t
y^ } z|| _| �
t j � W Y d}~dS d}~ww | j}|dushJ �| jr�| jdusrJ �| jdu syJ �| j�� | _|�td� | jI dH | jr�| �� dS z;t�| j�4 I dH �# |�� I dH }|j t j u r�| �
|j � W d �I dH W dS q�1 I dH s�w Y W dS tjy� | �
t j � � t
y� } z|| _| �
t j � W Y d}~dS d}~ww )zClose websocket connection.Nr� FTr ) rC r� rE r� r� r� r� r� �CancelledErrorr� r� r r� � ExceptionrL rD rJ rI rK �
create_futurer� r rF �_close_transport�
async_timeoutrQ r_ �readr� r �CLOSEr� )r6 r� r� r� r� r� �reader�msgr7 r7 r8 r� � sh �
� �� � � ��zWebSocketResponse.closec C s d| _ || _| �� dS )z6Set the close code and mark the connection as closing.TN)rF rH rl �r6 r� r7 r7 r8 � _set_closing� s z WebSocketResponse._set_closingc C s || _ | �� dS )z+Set the close code and close the transport.N)rH r� r r7 r7 r8 r� � s z+WebSocketResponse._set_code_close_transportc C s. | j dur| j jdur| j j�� dS dS dS )zClose the transport.N)rm r� r� r5 r7 r7 r8 r� � s �z"WebSocketResponse._close_transportc
� s� �| j d u r
td��|p| j} | jrtd��| jr,| jd7 _| jtkr*td��tS | jr1t S zXd| _zD|r`t
� |�4 I d H � | j � � I d H }W d �I d H n 1 I d H sZw Y n| j � � I d H }| �
� W d| _| jryt| jd � nd| _| jr�t| jd � w w W nq tjy� � ty� tj| _| �� I d H ttjd d � Y S ty� } z|j| _| j|jd�I d H ttj |d �W Y d }~S d }~w t y� } z || _ | � tj � | �� I d H ttj |d �W Y d }~S d }~ww |j!t"v�r|S |j!tj#u �r#| � |j$� | j�s!| j%�r!| jdd�I d H |S |j!tj&u �r2| � tj� |S |j!tj'u �rG| j(�rG| �)|j$�I d H q|j!tj*u �rS| j(�rSq|S ) Nr� Tz+Concurrent call to receive() is not allowedr
z WebSocket connection is closed.F)r� )r� )+rD r� r` rJ rE rG r, r rF r r� rQ r rx rK r r� r� r r �OKrH r� r r �CLOSEDr r� r� r� rL r r� r� r r r� ra �CLOSINGr~ rb r� r� )r6 rQ rR r r� r7 r7 r8 �receive� s� �
(��
���� � �� � �zWebSocketResponse.receive�rQ c � �D �| � |�I d H }|jtjur td|j� d|j�d���tt|j�S )N�Received message �:z is not WSMsgType.TEXT)r r� r r� r r� r r@ �r6 rQ r r7 r7 r8 � receive_strF � � � z WebSocketResponse.receive_strc � r )Nr r
z is not WSMsgType.BINARY)r r� r r� r r� r r� r r7 r7 r8 �
receive_bytesN r z WebSocketResponse.receive_bytes)�loadsrQ r c � s �| j |d�I d H }||�S )Nr
)r )r6 r rQ r� r7 r7 r8 � receive_jsonV s �z WebSocketResponse.receive_jsonc � s
�t d��)Nz"Cannot call .write() for websocket)r� )r6 r� r7 r7 r8 �write\ s �zWebSocketResponse.writec C s | S r4 r7 r5 r7 r7 r8 � __aiter___ s zWebSocketResponse.__aiter__c � s. �| � � I d H }|jtjtjtjfv rt�|S r4 )r r� r r r r �StopAsyncIteration)r6 r r7 r7 r8 � __anext__b s
�zWebSocketResponse.__anext__c C s, d| _ | �� | jd urt| j|� d S d S )NT)rF rl rD r r� r7 r7 r8 �_cancelh s
�zWebSocketResponse._cancel)r2 N)r� r� r2 Nr4 )ry )r2 r) )\r; r< r= rA r> r? rB r r@ rC r rD r rE rF rG �intrH rI r� �AbstractEventLooprJ rK �FuturerL �
BaseExceptionrM �floatrN � TimerHandlerO rP r� r
r r] rl rj rx rr r� r| r� r� r% r r� r r� r� r� r* r� �propertyr� r� r� r rW r r� r� r� r� r� r r} r� r� �jsonr� r" r� r� r r r� r r� r� r r r r r r! r r r r r �
__classcell__r7 r7 rh r8 r) > s
������� �
� � �
#
�
� U���
�
����
�
�����
�
����
�7
J ���
�
r) )Dr� r� r� r� r r �typingr r r r r r r �attr� multidictr r� r �_websocket.readerr �_websocket.writerr
�abcr �client_exceptionsr �helpersr r r �httpr r r r r r r r r r r �http_websocketr �logr �streamsr �typedefsr! r" �web_exceptionsr# r$ � web_requestr% � web_responser& r� r� �__all__r, r r? �sr* r) r7 r7 r7 r8 �<module> s>
$ 4