File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/web_ws.cpython-310.pyc
Back
o ;j?k � @ sF U d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m Z m Z m Z m Z mZmZ d dlZd dlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lm Z m 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+m,Z,m-Z- dd l.m/Z/ dd l0m1Z1 ddl2m3Z3 ddl4m5Z5m6Z6m7Z7 ddl8m9Z9m:Z: ddl;m<Z< ddl=m>Z> ej?dkr�d dl m@Z@ nd dlAm@Z@ ej?dkr�d dlmBZB nd dl mCZC eCeDeEddf ZBej?dkr�d dl ZFd dl mGZG n d dlFZFd dlAmGZG d ZHd ZIe eJ eKd < e@d eLd e d d!�ZMejNd d d d"�G d#d$� d$��ZOG d%d&� d&e>e eM �ZPdS )'� N)�Callable�Iterable)�Any�Final�Generic�Literal�cast�overload)� CIMultiDict� )�hdrs)�WebSocketDataQueue)�AbstractStreamWriter)�WSMessageTypeError)�DEFAULT_CHUNK_SIZE�calculate_timeout_when� set_exception� set_result) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSCloseCode� WSMessage�WSMessageDecodeText�WSMessageNoDecodeText� WSMsgType� ws_ext_gen� ws_ext_parse)�_INTERNAL_RECEIVE_TYPES)� ws_logger)� EofStream)�JSONBytesEncoder� JSONDecoder� JSONEncoder)�HTTPBadRequest� HTTPException)� BaseRequest)�StreamResponse)� � )�TypeVar�r+ � )�Buffer)�Unionzmemoryview[int]zmemoryview[bytes])r+ � )�Self)�WebSocketResponse�WebSocketReadyr � �THRESHOLD_CONNLOST_ACCESS� _DecodeTextT)�bound� covariant�default)Z auto_attribs�frozen�slotsc @ s0 e Zd ZU eed< edB ed< defdd�ZdS )r5 �okN�protocol�returnc C � | j S �N)r> ��self� rE �B/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/web_ws.py�__bool__R � zWebSocketReady.__bool__)�__name__� __module__� __qualname__�bool�__annotations__�strrG rE rE rE rF r5 M s r5 c s� e Zd ZU dZeed< dZedB ed< dZe dB ed< dZ e dB ed< dZ eed< dZ eed< d Zeed < dZedB ed < dZejdB ed < dZeed < dZejd dB ed< dZedB ed< dZeed< dZej dB ed< dZ ej dB ed< dZ ej d dB ed< dZ eed< dZ!ej"dB ed< dddddddde#dd� d ed edB d ed ed edB d!e$e d"ed#ed$ed%ed&df� fd'd(�Z%d�d)d*�Z&d�d+d,�Z'd�d-d.�Z(d�d/d0�Z)d�d1d2�Z*d�d3d4�Z+d�d7d8�Z,d�d9d:�Z-d;ed&dfd<d=�Z.d�d>d?�Z/d@e0d&e1f� fdAdB� Z2d@e0d&e3dCedB eef fdDdE�Z4d@e0d&e3edB e f fdFdG�Z5d@e0dHedB dIe d&dfdJdK�Z6d@e0d&e7fdLdM�Z8e9d&efdNdO��Z:e9d&efdPdQ��Z;e9d&edB fdRdS��Z<e9d&edB fdTdU��Z=e9d&eeB fdVdW��Z>d�dXedYe?d&e?fdZd[�Z@d&edB fd\d]�ZAd�d_eBd&dfd`da�ZCd�d_eBd&dfdbdc�ZD d�d_eBddeEd"edB d&dfdedf�ZFd�dged"edB d&dfdhdi�ZGd�dgeBd"edB d&dfdjdk�ZH d�eIjJdl�dge?d"edB dmeKd&dfdndo�ZL d�dge?d"edB dmeMd&dfdpdq�ZNd�drds�ZOePjQd^ddt�dued_eBdved&efdwdx�ZRduePd&dfdydz�ZSduePd&dfd{d|�ZTd�d}d~�ZUeV d�dd�d edB d&eWfd�d���ZXeV d�dd�d edB d&eYfd�d���ZXeV d�dd�d edB d&eWeYB fd�d���ZX d�d edB d&eWeYB fd�d��ZXeVdd��dd�d edB d&efd�d���ZZeVdd��dd�d edB d&eBfd�d���ZZeVdd��dd�d edB d&eeBB fd�d���ZZdd��d edB d&eeBB fd�d��ZZdd��d edB d&eBfd�d��Z[eVd�dd��dd�d�e\d edB d&e?fd�d���Z]eVd�dd��dd�d�e^eBge?f d edB d&e?fd�d���Z]eVd�dd��dd�d�e\e^eBge?f B d edB d&e?fd�d���Z]eIj_dd��d�e\e^eBge?f B d edB d&e?fd�d��Z]dge`d&dfd�d��Zad&ebfd�d��ZceVdd�d&eWfd�d���ZdeVdd�d&eYfd�d���ZdeVdd�d&eWeYB fd�d���Zdd&eWeYB fd�d��Zdd;ed&dfd�d��Ze� ZfS )�r4 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_task�_need_heartbeat_reset�_heartbeat_reset_handleg $@TrE i @ ) �timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� max_msg_size� writer_limit� decode_textra rb rc rd re rf rg rh ri rj r@ c sl t � jdd� || _|| _|| _|| _|| _|| _|d ur"|d | _|| _ || _ | | _ | | _ d| _ d | _d S )N�e )�statusg @F)�super�__init__� _protocols�_timeout�_receive_timeout� _autoclose� _autoping� _heartbeat�_pong_heartbeat� _compress� _max_msg_size� _writer_limit� _decode_textr_ r` ) rD ra rb rc rd re rf rg rh ri rj �� __class__rE rF rn k s zWebSocketResponse.__init__c C sd | � � | jd ur| j�� d | _d| _| jd ur!| j�� d | _| jd ur0| j�� d | _d S d S �NF)�_cancel_pong_response_cbr` �cancelr_ r\ r^ rC rE rE rF �_cancel_heartbeat� s �z#WebSocketResponse._cancel_heartbeatc C s" | j d ur| j �� d | _ d S d S rB )r] r~ rC rE rE rF r} � s �z*WebSocketResponse._cancel_pong_response_cbc C s> | j d u s| jr d S | j}|d usJ �d| _|�| j�| _d S �NT)rt r_ rW Z call_soon�_flush_heartbeat_resetr` )rD �looprE rE rF �_on_data_received� s z#WebSocketResponse._on_data_receivedc C s" d | _ | jsd S | �� d| _d S r| )r` r_ �_reset_heartbeatrC rE rE rF r� � s z(WebSocketResponse._flush_heartbeat_resetc 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 )Nr6 ) rt r} �_req� _protocol�_timeout_ceil_thresholdrW �timer r[ r\ �call_at�_send_heartbeat)rD �req�timeout_ceil_thresholdr� �now�whenrE rE rF r� � s � �z"WebSocketResponse._reset_heartbeatc C s� d | _ | jrd S | j}|d ur| jd usJ �|�� }|| jk r*|�| j| j�| _ d S | j}|d ur5|j j nd}t || j |�}| � � |�|| j�| _| j�dtj�}tjdkr`tj||dd�}n|�|�}|�� st|| _|�| j� d S | �|� d S )Nr6 � r. T)r� Z eager_start) r\ r_ rW rQ r� r[ r� r� r� r� r� r ru r} �_pong_not_receivedr] � send_framer �PING�sys� version_info�asyncio�TaskZ create_taskZdoner^ Zadd_done_callback�_ping_task_done)rD r� r� r� r� r� �coroZ ping_taskrE rE rF r� � s2 �� z!WebSocketResponse._send_heartbeat�task�asyncio.Task[None]c C s( |� � s|�� }r| �|� d| _dS )z*Callback for when the ping task completes.N)Z cancelled� exception�_handle_ping_pong_exceptionr^ )rD r� �excrE rE rF r� � s z!WebSocketResponse._ping_task_donec C s< | j d ur| j jd ur | �t�d| j� d��� d S d S d S )NzNo PONG received after z seconds)r� � transportr� r� � TimeoutErrorru rC rE rE rF 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 ) rS � _set_closed�_set_code_close_transportr �ABNORMAL_CLOSURErZ rX rT rR � feed_datar r �ERROR�rD r� rE rE rF 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)rS r rC rE rE rF r� s z WebSocketResponse._set_closed�requestc � s^ �| j d ur | j S | �|�\}}t� �|�I d H }|d us J �| �|||� |�� I d H |S rB )�_payload_writer� _pre_startrm �prepare� _post_start�drain)rD r� r? �writerZpayload_writerrz rE rF r� s � zWebSocketResponse.preparezCIMultiDict[str]c C s� |j }d|�tjd��� �� kr td|�tj�� d�d��|jjs-td|�tj �� �d��d }tj |v rYdd� |tj � d �D �}|D ] }|| j v rN|} n qCt �d |j|| j � |�tjd�}|d vrltd |� �d��|�tj�}z|r~tt�|��d kr�td|��d��W n tjy� td|��d�d �w t�t�|�� t �� � �� � }t tjdtj dtj |i�} d} d} | j r�|�tj!�} t"| dd�\} } | r�t#| d| d�} | | tj!< |r�|| tj < | || | fS )NZ websocket� zNo WebSocket UPGRADE hdr: z$ Can "Upgrade" only to "WebSocket".)�textzNo CONNECTION upgrade hdr: c S s g | ]}t |�� ��qS rE )rN �strip)�.0�protorE rE rF � <listcomp>* s ��z0WebSocketResponse._handshake.<locals>.<listcomp>�,u< %s: Client protocols %r don’t overlap server-known ones %r)Z13�8�7zUnsupported version: � zHandshake error: �upgradeFr T)�isserver)rg r� Zserver_notakeover)$�headers�getr ZUPGRADE�lowerr� r' Z_messager� Z CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitro r" �warningZremoteZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder ZSEC_WEBSOCKET_ACCEPTrv ZSEC_WEBSOCKET_EXTENSIONSr r )rD r� r� r? Z req_protocolsr� �version�keyZ accept_valZresponse_headers� notakeoverrg � extensionsZ enabledextrE rE rF � _handshake s� �� � �� ������ � �z WebSocketResponse._handshakec C st |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}|d u r+t d��t |j|||| j d�}||fS )Nrk zConnection lost)rg r� �limit) rW r� Z set_statusr� �updateZ force_closerv r� r� �ConnectionResetErrorr rx )rD r� r� r? rg r� r� r� rE rE rF r� j s" �z WebSocketResponse._pre_startr? r� c C s� || _ || _| �� | j}|d usJ �t|jt|d�| _t| j| j t | j �| j d�}| j d u r1d n| j}|jj||d� |j�d� d S )N)r� )rg rj )Zdata_received_cbF)rP rQ r� rW r r� r rR r rw rL rv ry rt r� r? Z set_parserZ keep_alive)rD r� r? r� r� �parser�cbrE rE rF r� � s" ��z WebSocketResponse._post_startc C sP | j d ur td��z | �|�\}}}}W n ty" tdd � Y S w td|�S )NzAlready startedFT)rQ � RuntimeErrorr� r( r5 )rD r� �_r? rE rE rF � can_prepare� s � z WebSocketResponse.can_preparec C s | j d uS rB )rQ rC rE rE rF �prepared� s zWebSocketResponse.preparedc C rA rB )rS rC rE rE rF �closed� � zWebSocketResponse.closedc C rA rB )rV rC rE rE rF � close_code� r� z WebSocketResponse.close_codec C rA rB )rP rC rE rE rF � ws_protocol� r� z WebSocketResponse.ws_protocolc C rA rB )rv rC rE rE rF rg � r� zWebSocketResponse.compress�namer; c 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)rQ r� �get_extra_info)rD r� r; r� r� rE rE rF r� � s z WebSocketResponse.get_extra_infoc C rA rB )rZ rC rE rE rF r� � rH zWebSocketResponse.exceptionr� �messagec � �. �| j d u r td��| j �|tj�I d H d S �N�Call .prepare() first)rQ r� r� r r� �rD r� rE rE rF �ping� s � zWebSocketResponse.pingc � r� r� )rQ r� r� r �PONGr� rE rE rF �pong� s � zWebSocketResponse.pong�opcodec � s. �| j du r td��| j �|||�I dH dS )z Send a frame over the websocket.Nr� )rQ r� r� )rD r� r� rg rE rE rF 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-8�rg ) rQ r� � isinstancerN � TypeError�typer� r� r �TEXT�rD r� rg rE rE rF �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� ) rQ r� r� �bytes� bytearray� memoryviewr� r� r� r �BINARYr� rE rE rF � send_bytes� s � z WebSocketResponse.send_bytes)�dumpsr� c � s �| j ||�|d�I d H d S )Nr� )r� �rD r� rg r� rE rE rF � send_json� s � zWebSocketResponse.send_jsonc � s �| j ||�|d�I dH dS )z�Send JSON data using a bytes-returning encoder as a binary frame. Use this when your JSON encoder (like orjson) returns bytes instead of str, avoiding the encode/decode overhead. r� N)r� r� rE rE rF �send_json_bytes� s � z!WebSocketResponse.send_json_bytesc � s6 �| j rd S | jd u rtd��| �� I d H d| _ d S )Nz Response has not been startedT)Z _eof_sentr� r� �closerC rE rE rF � 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 ) rQ r� rS r� r� r� r� r� ZCancelledErrorr� r� r r� � ExceptionrZ rR rX rW rY Z create_futurer� r rT �_close_transport� async_timeoutra rp �readr� r �CLOSEr� )rD r� r� r� r� r� �reader�msgrE rE rF r� sh � � �� � � ��zWebSocketResponse.closec C s d| _ || _| �� dS )z6Set the close code and mark the connection as closing.TN)rT rV r �rD r� rE rE rF � _set_closing@ s z WebSocketResponse._set_closingc C s || _ | �� dS )z+Set the close code and close the transport.N)rV r� r rE rE rF r� F s z+WebSocketResponse._set_code_close_transportc C s. | j dur| j jdur| j j�� dS dS dS )zClose the transport.N)r� r� r� rC rE rE rF r� K s �z"WebSocketResponse._close_transportrD z WebSocketResponse[Literal[True]]c � � �d S rB rE �rD ra rE rE rF �receiveP � �zWebSocketResponse.receivez!WebSocketResponse[Literal[False]]c � r rB rE r rE rE rF r U r z WebSocketResponse[_DecodeText]c � r rB rE r rE rE rF r Z r c � s� �| j d u r td��|p| j} | jrtd��| jr,| jd7 _| jtkr*td��tS | jr1t S zTd| _z@|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| _| j rut| j d � nd| _| j r�t| j d � 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!vr�|S |j tj"u �r | � |j#� | j�s | j$�r | jdd�I d H |S |j tj%u �r-| � tj� |S |j tj&u �rB| j'�rB| �(|j#�I d H q|j tj)u �rN| j'�rNq|S ) Nr� Tz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.F)r� )r� )*rR r� rq rX rS rU r7 r rT r r� ra r� rY r r� r� r# r �OKrV r� r r �CLOSEDr r� r� r� rZ r r� r� r! r� r� rr �CLOSINGr� rs r� r� )rD ra rb r r� rE rE rF r _ s� � (��� ���� � �� � ��ra c � r rB rE r rE rE rF � receive_str� r z WebSocketResponse.receive_strc � r rB rE r rE rE rF r � r c � r rB rE r rE rE rF r � r c � sD �| � |�I dH }|jtjur td|j� d|j�d���tt|j�S )zrReceive TEXT message. Returns str when decode_text=True (default), bytes when decode_text=False. N�Received message �:z is not WSMsgType.TEXT)r r� r r� r r� r rN �rD ra r rE rE rF r � s � � c � sD �| � |�I d H }|jtjur td|j� d|j�d���tt|j�S )Nr r z is not WSMsgType.BINARY)r r� r r� r r� r r� r rE rE rF � receive_bytes� s � � z WebSocketResponse.receive_bytes.)�loadsra r c � r rB rE �rD r ra rE rE rF � receive_json� � �z WebSocketResponse.receive_jsonc � r rB rE r rE rE rF r � r c � r rB rE r rE rE rF r � r c � s �| j |d�I d H }||�S )Nr )r )rD r ra r� rE rE rF r � s �c � s �t d��)Nz"Cannot call .write() for websocket)r� )rD r� rE rE rF �write� s �zWebSocketResponse.writec C s | S rB rE rC rE rE rF � __aiter__� s zWebSocketResponse.__aiter__c � r rB rE rC rE rE rF � __anext__� r zWebSocketResponse.__anext__c � r rB rE rC rE rE rF r � r c � r rB rE rC rE rE rF r � r c � s. �| � � I d H }|jtjtjtjfv rt�|S rB )r r� r r� r r �StopAsyncIteration)rD r rE rE rF r s �c C s, d| _ | �� | jd urt| j|� d S d S r� )rT r rR r r� rE rE rF �_cancel s �zWebSocketResponse._cancel)r@ N)r� r� r@ NrB )r� )grI rJ rK rO rL rM rP rN rQ r rR r rS rT rU �intrV rW r� ZAbstractEventLooprX rY ZFuturerZ � BaseExceptionr[ �floatr\ Z TimerHandler] r^ r� r_ r` �Handler r rn r r} r� r� r� r� r� r� r� r� r) r r� �tupler� r� r� r5 r� �propertyr� r� r� r� rg 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 r r0 r r3 r r r � __classcell__rE rE rz rF r4 V s ������� � � � � � ) � � T��� � ���� � ����� � ���� � ���� �7 ��� ���� ���� ��� �K��� ���� ���� � ���� ��� �� ����� � ��� � � �� �� �r4 )Qr� r� r� r� r r� �collections.abcr r �typingr r r r r r �attrZ multidictr r� r Z_websocket.readerr �abcr Zclient_exceptionsr Zhelpersr r r r �httpr r r r r r r r r r r r r Zhttp_websocketr! �logr" Zstreamsr# Ztypedefsr$ r% r& Zweb_exceptionsr' r( Z web_requestr) Z web_responser* r� r- Ztyping_extensionsr0 r1 r� r� r� r3 �__all__r7 r rM rL r8 �sr5 r4 rE rE rE rF �<module> sR <
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings