File: //usr/local/lib/python3.10/dist-packages/aiohttp/_websocket/__pycache__/writer.cpython-310.pyc
o
���g� � @ s� U d Z ddlZddlZddlZddlmZ ddlmZmZm Z m
Z
ddl m Z ddl
mZ ddlmZ d d
lmZmZmZmZmZmZmZmZ d d lmZm Z d Z ee e d
< dZ G dd� d�Z!dS )z%WebSocket protocol versions 13 and 8.� N)�partial)�Any�Final�Optional�Union� )� BaseProtocol)�ClientConnectionResetError)�ZLibCompressor� )�MASK_LEN�MSG_SIZE�PACK_CLOSE_CODE� PACK_LEN1� PACK_LEN2� PACK_LEN3�
PACK_RANDBITS�websocket_mask)�WS_DEFLATE_TRAILING� WSMsgType� �
DEFAULT_LIMITi c @ s� e Zd ZdZdee�� ddd�dedej de
de d ejd
e d e
d d
fdd�Z
dde
de d
ee d d
fdd�Zd
e d efdd�Zd de dee
ef d d
fdd�Zd
S ) �WebSocketWritera3 WebSocket writer.
The writer is responsible for sending messages to the client. It is
created by the protocol when a connection is established. The writer
should avoid implementing any application logic and should only be
concerned with the low-level details of the WebSocket protocol.
Fr )�use_mask�limit�random�compress�
notakeover�protocol� transportr r r r r �returnNc C sH || _ || _|| _t|jd�| _|| _|| _d| _|| _ d| _
d| _ dS )z Initialize a WebSocket writer.� Fr N) r r r r � getrandbits�get_random_bitsr r �_closing�_limit� _output_size� _compressobj)�selfr r r r r r r � r) �D/usr/local/lib/python3.10/dist-packages/aiohttp/_websocket/writer.py�__init__- s
zWebSocketWriter.__init__�message�opcodec
� s� �| j r
|tj@ s
td��d}|s| jrF|dk rFd}|r"| �|�}n
| js,| �| j�| _| j}|�|�I dH |�| jr>t j
nt j � � t
�}t|�}| j}|rQdnd}d|B |B } |dk rgt| ||B �}
d} n|d k rvt| d|B |�}
d
} n
t| d |B |�}
d } | j�� r�td��|r�t| �� �} t|�}t| |� | j�|
| | � | jt7 _n|t kr�| j�|
� | j�|� n| j�|
| � | j| | 7 _| j| j kr�d| _| j j r�| j � � I dH dS dS dS )
z<Send a frame over the websocket with message as its payload.z!Cannot write to closing transportr � �@ N� �~ r r � � �
)!r$ r �CLOSEr r �_make_compress_objr' �flushr �zlib� Z_FULL_FLUSH� Z_SYNC_FLUSH� removesuffixr �lenr r r r r �
is_closingr r# � bytearrayr �writer& r r
r% r �_paused�
_drain_helper)
r( r, r- r �rsv� compressobj�
msg_lengthr �mask_bit�
first_byte�header�
header_len�maskr) r) r* �
send_frameD sb � ���
�zWebSocketWriter.send_framec C s t tj| td�S )N)�level�wbits�max_sync_chunk_size)r
r8 � Z_BEST_SPEED� WEBSOCKET_MAX_SYNC_CHUNK_SIZE)r( r r) r) r* r6 � s
�z"WebSocketWriter._make_compress_obj�� � �codec � sJ �t |t�r |�d�}z| jt|�| tjd�I dH W d| _dS d| _w )z<Close the websocket, sending the specified code and message.zutf-8)r- NT)�
isinstance�str�encoderJ r r r5 r$ )r( rR r, r) r) r* �close� s �
�zWebSocketWriter.close)N)rP rQ )�__name__�
__module__� __qualname__�__doc__r r �Randomr �asyncio� Transport�bool�intr+ �bytesr rJ r
r6 r rT rV r) r) r) r* r $ sH
������� �
�����
�]$r )"rZ r\ r r8 � functoolsr �typingr r r r �
base_protocolr �client_exceptionsr �compression_utilsr
�helpersr r
r r r r r r �modelsr r r r_ �__annotations__rO r r) r) r) r* �<module> s (