File: //opt/saltstack/salt/extras-3.10/redis/_parsers/__pycache__/base.cpython-310.pyc
o
��e8 � @ s, d dl Z d dlmZ d dlmZmZmZ d dlmZm Z m
Z
e j j dkr1e j j
dkr1d dlmZ nd dlmZ dd lmZmZmZmZmZmZmZmZmZmZmZm Z dd
lm Z d d l m Z d d
l m!Z!m"Z" dZ#dZ$dZ%dZ&eed�Z'G dd� de�Z(G dd� de(�Z)G dd� de(�Z*G dd� de*�Z+dS )� N)�ABC)�IncompleteReadError� StreamReader� TimeoutError)�List�Optional�Union� � )�timeout� ) �AuthenticationError�$AuthenticationWrongNumberOfArgsError�BusyLoadingError�ConnectionError�ExecAbortError� ModuleError�NoPermissionError�
NoScriptError�OutOfMemoryError�
ReadOnlyError�
RedisError�
ResponseError)�
EncodableT� )�Encoder)� SERVER_CLOSED_CONNECTION_ERROR� SocketBufferz:Error loading the extension. Please check the server logs.z5Error unloading module: no such module with that namez/Error unloading module: operation not possible.z[Error unloading module: the module exports one or more module-side data types, can't unload)zxAUTH <password> called without any password configured for the default user. Are you sure your configuration is correct?z(Client sent AUTH, but no password is setc @ sd e Zd Zdedededeeeeee ee
eie �e ee
eeeeed� Zedd� �Zdd � Zd
d � Zd S )
�
BaseParserz max number of clients reachedzinvalid passwordz,wrong number of arguments for 'auth' commandz,wrong number of arguments for 'AUTH' command) �ERRZOOMZ WRONGPASSZ EXECABORTZLOADINGZNOSCRIPTZREADONLYZNOAUTHZNOPERMc C s\ |� d�d }|| jv r*|t|�d d� }| j| }t|t�r&|�|t�}||�S t|�S )zParse an error response� r r N)�split�EXCEPTION_CLASSES�len�
isinstance�dict�getr )�cls�response�
error_codeZexception_class� r* �6/opt/saltstack/salt/extras-3.10/redis/_parsers/base.py� parse_errorM s
zBaseParser.parse_errorc C � t � ��N��NotImplementedError��selfr* r* r+ �
on_disconnectY � zBaseParser.on_disconnectc C r- r. r/ �r2 �
connectionr* r* r+ �
on_connect\ r4 zBaseParser.on_connectN)�__name__�
__module__� __qualname__r r
r �MODULE_LOAD_ERRORr � MODULE_EXPORTS_DATA_TYPES_ERROR�NO_SUCH_MODULE_ERROR� MODULE_UNLOAD_NOT_POSSIBLE_ERROR�NO_AUTH_SET_ERRORr r r r r r r" � classmethodr, r3 r7 r* r* r* r+ r 0 s6 ���
r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� _RESPBasez&Base class for sync-based resp parsingc C s || _ d | _d | _d | _d S r. )�socket_read_size�encoder�_sock�_buffer�r2 rB r* r* r+ �__init__c s
z_RESPBase.__init__c C s$ z| � � W d S ty Y d S w r. )r3 � Exceptionr1 r* r* r+ �__del__i s
�z_RESPBase.__del__c C s( |j | _ t| j | j|j�| _|j| _dS )z Called when the socket connectsN)rD r rB �socket_timeoutrE rC r5 r* r* r+ r7 o s
� z_RESPBase.on_connectc C s* d| _ | jdur| j�� d| _d| _dS )z"Called when the socket disconnectsN)rD rE �closerC r1 r* r* r+ r3 w s
z_RESPBase.on_disconnectc C s | j o| j �|�S r. )rE �can_read)r2 r r* r* r+ rL s z_RESPBase.can_readN) r8 r9 r: �__doc__rG rI r7 r3 rL r* r* r* r+ rA ` s rA c
@ sV e Zd ZdZdZdefdd�Zdefdd�Z dd
ede e
e d e e
f fd d
�Z
d S )�AsyncBaseParserz5Base parsing class for the python-backed async parser��_streamZ
_read_sizerB c C s d | _ || _d S r. rO rF r* r* r+ rG � s
zAsyncBaseParser.__init__�returnc � � �t � �r. r/ r1 r* r* r+ �can_read_destructive� s �z$AsyncBaseParser.can_read_destructiveF�disable_decodingNc � rR r. r/ )r2 rT r* r* r+ �
read_response� s �z AsyncBaseParser.read_response)F)r8 r9 r: rM � __slots__�intrG �boolrS r r r r rU r* r* r* r+ rN � s ���rN c sv e Zd ZdZejd Zdef� fdd� Zdd� Zdd � Z d
d � Z
d e fd
d�Z ded e
fdd�Zd e
fdd�Z� ZS )�_AsyncRESPBasez!Base class for async resp parsing)rC rE �_pos�_chunksrB c s( t � �|� d | _d| _g | _d| _d S )N� r )�superrG rC rE r[ rZ rF �� __class__r* r+ rG � s
z_AsyncRESPBase.__init__c C s d| _ | j�� d S )Nr\ )rE r[ �clearr1 r* r* r+ �_clear� s z_AsyncRESPBase._clearc C s4 |j | _| jdu r
td��|j| _| �� d| _dS )z Called when the stream connectsN�Buffer is closed.T)�_readerrP r rC ra �
_connectedr5 r* r* r+ r7 � s
z_AsyncRESPBase.on_connectc C s
d| _ dS )z"Called when the stream disconnectsFN)rd r1 r* r* r+ r3 � s
z _AsyncRESPBase.on_disconnectrQ c � s� �| j std��| jr
dS z+td�4 I d H � | j�d�I d H W d �I d H W S 1 I d H s1w Y W d S tyB Y dS w )Nrb Tr r F)rd r rE �
async_timeoutrP �readr r1 r* r* r+ rS � s �4� �z#_AsyncRESPBase.can_read_destructive�lengthc
� s� �|d }| j | }t| j�|kr | j| j |d � }n7| j| j d� }z| j�|t|� �I dH }W n tyD } ztt�|�d}~ww || dd� }| j� |� | j |7 _ |S )z�
Read `length` bytes of data. These are assumed to be followed
by a '
' terminator which is subsequently discarded.
r N���)
rZ r# rE rP � readexactlyr r r r[ �append)r2 rg Zwant�end�result�tail�data�errorr* r* r+ �_read� s �
�� z_AsyncRESPBase._readc � s� �| j �d| j�}|dkr| j | j|� }n'| j | jd� }| j�� I dH }|�d�s/tt��|| dd� }| j� |� | jt
|�d 7 _|S )zq
read an unknown number of bytes up to the next '
'
line separator, which is discarded.
s
r Nrh r ) rE �findrZ rP �readline�endswithr r r[ rj r# )r2 �foundrl rm rn r* r* r+ � _readline� s �
z_AsyncRESPBase._readline)r8 r9 r: rM rN rV rW rG ra r7 r3 rX rS �bytesrp ru �
__classcell__r* r* r^ r+ rY � s
rY ),�sys�abcr �asyncior r r �typingr r r � version_info�major�minorr re �
exceptionsr
r r r r r r r r r r r r �encodersr �socketr r r; r= r> r< r? r rA rN rY r* r* r* r+ �<module> s. 8 �
�
0#