File: //lib/python3.9/site-packages/redis/asyncio/__pycache__/connection.cpython-39.pyc
a
��b& �
@ s| U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZ d dl m Z m Z m Z m Z d dl!Z!d dl"m#Z# d dl$m%Z% d dl&m'Z'm(Z( d d l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 d d
l9m:Z:m;Z; d d l<m=Z=m>Z> dZ?e=�rld dl?Z?e@ejAejBd ejCd ejDd iZEeFeE�G� �ZHd
ZIdZJdZKdZLdZMdZNG dd� dejO�ZPePjQZRdZSdZTdZUdZVG dd� de(dd �ZWG d d � d �ZXeeYeeeZ eeYeeZ f f f Z[G d d � d �Z\G d!d"� d"�Z]G d#d$� d$e\�Z^G d%d&� d&e\�Z_eee^e_f e`d'< e=�rve_Zane^ZaG d(d)� d)e'�ZbG d*d+� d+e'�Zceebecf ZdG d,d-� d-�ZeG d.d/� d/ee�ZfG d0d1� d1�ZgG d2d3� d3ee�Zhd4Zieej d5�d6d7�Zkeelememekekelelekd8��ZneeYed9eof f e`d:< G d;d<� d<e(dd �ZpeYepd=�d>d?�Zqed@dAdB�ZrG dCdA� dA�ZsG dDdE� dEes�ZtdS )F� N)�chain)�MappingProxyType) �Any�Callable�Iterable�List�Mapping�Optional�Set�Tuple�Type�TypeVar�Union)� ParseResult�parse_qs�unquote�urlparse)�Retry)� NoBackoff)�Protocol� TypedDict)�AuthenticationError�$AuthenticationWrongNumberOfArgsError�BusyLoadingError�ChildDeadlockedError�ConnectionError� DataError�ExecAbortError�InvalidResponse� ModuleError�NoPermissionError�
NoScriptError�
ReadOnlyError�
RedisError�
ResponseError� TimeoutError)�
EncodableT�EncodedT)�HIREDIS_AVAILABLE� str_if_bytes� � *� $s
�
� z Connection closed by server.c @ s e Zd Ze� ZdS )� _SentinelN)�__name__�
__module__� __qualname__�object�sentinel� r5 r5 �</usr/lib/python3.9/site-packages/redis/asyncio/connection.pyr/ N s r/ z: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 unloadc @ sJ e Zd ZU eegef ed< eegef ed< ee ed< ee ed< dS )�_HiredisReaderArgs�
protocolError�
replyError�encoding�errorsN)r0 r1 r2 r �str� Exception�__annotations__r r5 r5 r5 r6 r7 ] s
r7 F)�totalc @ sH e Zd ZdZdZeeed�dd�Zee d�dd�Z
d eed�d d
�Z d S )
�Encoderz=Encode strings to bytes-like and decode bytes-like to strings�r: �encoding_errors�decode_responsesc C s || _ || _|| _d S �NrA )�selfr: rB rC r5 r5 r6 �__init__i s zEncoder.__init__)�value�returnc C st t |t�r|�| j| j�S t |ttf�r,|S t |ttf�rXt |t �rLt
d��t |��� S |j j
}t
d|�d���dS )z=Return a bytestring or bytes-like representation of the valuezNInvalid input of type: 'bool'. Convert to a bytes, string, int or float first.zInvalid input of type: z1. Convert to a bytes, string, int or float first.N)�
isinstancer< �encoder: rB �bytes�
memoryview�int�float�boolr �repr� __class__r0 )rE rG �typenamer5 r5 r6 rJ n s
�
�zEncoder.encodeFc C sF | j s
|rBt|t�r$|�| j| j�S t|t�rB|�� �| j| j�S |S )z:Return a unicode string from the bytes-like representation)rC rI rK �decoder: rB rL �tobytes)rE rG �forcer5 r5 r6 rS � s
zEncoder.decodeN)F) r0 r1 r2 �__doc__� __slots__r< rO rF r& r' rJ rS r5 r5 r5 r6 r@ d s
r@ c @ s� e Zd ZU dZdZdededededeee e
e e e e e i e
eeeeed�Zeed < ed
�d d �Zd
d� Zeed�dd�Zdd� Zdd�dd�Z e e d�dd�Z d e e e!ed e"e! f d �d d �Z#d S )!�
BaseParser�Plain Python parsing class��_stream�_buffer�
_read_sizez max number of clients reachedz(Client sent AUTH, but no password is setzinvalid passwordz,wrong number of arguments for 'auth' commandz,wrong number of arguments for 'AUTH' command)ZERRZ EXECABORTZLOADINGZNOSCRIPTZREADONLYZNOAUTHZNOPERM�EXCEPTION_CLASSES��socket_read_sizec C s d | _ d | _|| _d S rD rZ �rE r` r5 r5 r6 rF � s zBaseParser.__init__c C s$ z | � � W n ty Y n0 d S rD )�
on_disconnectr= �rE r5 r5 r6 �__del__� s zBaseParser.__del__)�responserH c C s\ |� d�d }|| jv rT|t|�d d� }| j| }t|t�rL|�|t�}||�S t|�S )zParse an error response� r � N)�splitr^ �lenrI �dict�getr$ )rE re Z
error_codeZexception_classr5 r5 r6 � parse_error� s
zBaseParser.parse_errorc C s
t � �d S rD ��NotImplementedErrorrc r5 r5 r6 rb � s zBaseParser.on_disconnect�
Connection��
connectionc C s
t � �d S rD rm �rE rq r5 r5 r6 �
on_connect� s zBaseParser.on_connect��timeoutrH c � s
t � �d S rD rm �rE ru r5 r5 r6 �can_read� s zBaseParser.can_readFN��disable_decodingrH c � s
t � �d S rD rm )rE ry r5 r5 r6 �
read_response� s zBaseParser.read_response)F)$r0 r1 r2 rV rW r r r �MODULE_LOAD_ERRORr � MODULE_EXPORTS_DATA_TYPES_ERROR�NO_SUCH_MODULE_ERROR� MODULE_UNLOAD_NOT_POSSIBLE_ERRORr r r! r" r r^ �ExceptionMappingTr> rM rF rd r< r$ rl rb rs rN rO rw r r& r rz r5 r5 r5 r6 rX � s<
�� ��rX c @ s� e Zd ZdZejeee d�dd�Z e
dd� �Z de dfee e
edef eed �d
d �Zeed �d
d�Zeed�dd�Zed�dd�Zdd� Zdd� ZdS )� SocketBufferz�Async-friendly re-impl of redis-py's SocketBuffer.
TODO: We're currently passing through two buffers,
the asyncio.StreamReader and this. I imagine we can reduce the layers here
while maintaining compliance with prior art.
)�
stream_readerr` �socket_timeoutc C s, || _ || _|| _t�� | _d| _d| _d S �Nr )r[ r` r� �io�BytesIOr\ �
bytes_written�
bytes_read)rE r� r` r� r5 r5 r6 rF � s
zSocketBuffer.__init__c C s | j | j S rD )r� r� rc r5 r5 r6 �length� s zSocketBuffer.lengthNT)r� ru �raise_on_timeoutrH c
� s� | j }|d u s| jd u r td��|�| j� d}|tur<|n| j}z�t�|�4 I d H �* | j� | j
�I d H }W d �I d H q�1 I d H s�0 Y t |t �r�t
|�dkr�tt��|�|� t
|�}| j|7 _||7 }|d ur�||kr�qDW dS tjtjf�y |�rtd��Y dS t�y� } zJt�|jd�} |�s\|j| k�r\W Y d }~dS td|j� ���W Y d }~n
d }~0 0 d S )N�Buffer is closed.r T�Timeout reading from socketF����!Error while reading from socket: )r\ r[ r# �seekr� �SENTINELr� �
async_timeoutru �readr` rI rK ri r � SERVER_CLOSED_CONNECTION_ERROR�write�socket�asyncior% �NONBLOCKING_EXCEPTIONS�#NONBLOCKING_EXCEPTION_ERROR_NUMBERSrk rQ �errno�args)
rE r� ru r� �bufZmarker�dataZ data_length�ex�allowedr5 r5 r6 �_read_from_socket� s6 >
z SocketBuffer._read_from_socketrt c � s t | j�p | j|dd�I d H S )NF�ru r� )rO r� r� rv r5 r5 r6 rw s �zSocketBuffer.can_read)r� rH c � s� |d }|| j kr(| �|| j �I d H | jd u r:td��| j�| j� | j�|�}| jt|�7 _| j| jkrz| � � |d d� S )Nr* r� ���)
r� r� r\ r# r� r� r� ri r� �purge)rE r� r� r5 r5 r6 r� s
zSocketBuffer.read�rH c � s� | j }|d u rtd��|�| j� |�� }|�t�sX| �� I d H |�| j� |�� }q*| jt|�7 _| j| j kr~| �
� |d d� S )Nr� r� ) r\ r# r� r� �readline�endswith�SYM_CRLFr� ri r� r� )rE r� r� r5 r5 r6 r� 0 s
zSocketBuffer.readlinec C s8 | j d u rtd��| j �d� | j �� d| _d| _d S )Nr� r )r\ r# r� �truncater� r� rc r5 r5 r6 r� F s
zSocketBuffer.purgec C s: z| � � | j�� W n ty( Y n0 d | _d | _d S rD )r� r\ �closer= r[ rc r5 r5 r6 r� O s zSocketBuffer.close)r0 r1 r2 rV r� � StreamReaderrM r rN rF �propertyr� r� r r/ rO r� rw rK r� r� r� r� r5 r5 r5 r6 r� � s* �
� � * r� c sp e Zd ZdZejd Zed�� fdd� Zdd�dd �Zd
d � Z e
d �d
d�Z de e
eedf d�dd�Z� ZS )� PythonParserrY )�encoderr_ c s t � �|� d | _d S rD )�superrF r� ra �rQ r5 r6 rF c s zPythonParser.__init__ro rp c C s: |j | _| jdu rtd��t| j| j|j�| _|j| _dS )z Called when the stream connectsNr� )�_readerr[ r# r� r] r� r\ r� rr r5 r5 r6 rs g s
�zPythonParser.on_connectc C s4 | j durd| _ | jdur*| j�� d| _d| _dS )z"Called when the stream disconnectsN)r[ r\ r� r� rc r5 r5 r6 rb r s
zPythonParser.on_disconnect�ru c � s | j ot| j �|�I d H �S rD )r\ rO rw rv r5 r5 r6 rw { s zPythonParser.can_readFNrx c � sB �j r �jstt���j �� I d H }|s0tt��|d d� |dd � }}|dvr`td|����|dkr�|jddd�}��|�}t|t�r�|�|S |dkr�n�|d kr�t |�}nn|d
kr�t |�}|d kr�d S �j �
|�I d H }n>|d k�r t |�}|d k� r�d S � �fd
d�t |�D �I d H }t|t ��r>� du �r>�j�|�}|S )Nrg )� -� +� :r, r+ zProtocol Error: r� �utf-8�replace)r; r� r� r, r� r+ c � s g | ]}�� � �I d H �qS rD )rz )�.0�_�ry rE r5 r6 �
<listcomp>� s z.PythonParser.read_response.<locals>.<listcomp>F)
r\ r� r r� r� r rS rl rI rM r� �rangerK )rE ry �rawZbytere �errorr� r5 r� r6 rz ~ sB
� zPythonParser.read_response)F)r0 r1 r2 rV rX rW rM rF rs rb rN rw rO r r& r$ rz �
__classcell__r5 r5 r� r6 r� ^ s
� �r� c s� e Zd ZU dZejd Zeed< ed�� fdd� Z dd�d d
�Z
d d � Z e d
�dd�Z
edfee def ed�dd�Zdeeeee f d�dd�Z� ZS )�
HiredisParserz*Parser class for connections using Hiredis)�_next_responser� �_socket_timeoutr� r_ c s* t s td��t� j|d� d | _d | _d S )NzHiredis is not available.r_ )r( r# r� rF r� r� ra r� r5 r6 rF � s
zHiredisParser.__init__ro rp c C sX |j | _t| jd�}|jjr4|jj|d< |jj|d< tj f i |��| _ d| _
|j | _ d S )N)r8 r9 r: r; F)
r� r[ r rl r� rC r: rB �hiredis�Readerr� r� r� )rE rq �kwargsr5 r5 r6 rs � s � zHiredisParser.on_connectc C s d | _ d | _d| _d S )NF)r[ r� r� rc r5 r5 r6 rb � s zHiredisParser.on_disconnectr� c � sL | j r | jstt��| jdu r*| j�� | _| jdu rH| j|dd�I d H S dS )NFr� T)r[ r� r r� r� �gets�read_from_socketrv r5 r5 r6 rw � s
zHiredisParser.can_readTNr� c
� sR |t u r| jn|}z�|d u r2| j�| j�I d H }nRt�|�4 I d H �* | j�| j�I d H }W d �I d H q�1 I d H sz0 Y |r�t|t�s�t t
�d �| j � |� W dS t
jy� � Y n� tjt
jfy� |r�td�d �Y dS t�yL } zJt�|jd�}|�s(|j|k�r(W Y d }~dS t d|j� ���W Y d }~n
d }~0 0 d S )NTr� Fr� r� )r� r� r[ r� r] r� ru rI rK r r� r� Zfeedr� �CancelledErrorr� r% r� r� rk rQ r� r� )rE ru r� �bufferr� r� r5 r5 r6 r� � s* >
z HiredisParser.read_from_socketFrx c � s� | j r | js | �� tt�d �| jdur8| j}d| _|S | j�� }|du rd| �� I d H | j�� }qBt|t�rt|�n$t|t �r�|r�t|d t�r�|d �|S )NFr )
r[ r� rb r r� r� r� r� rI �list)rE ry re r5 r5 r6 rz � s*
�� �zHiredisParser.read_response)F)r0 r1 r2 rV rX rW rO r> rM rF rs rb rN rw r� r r/ r� r& r rz r� r5 r5 r� r6 r� � s$
� � # ��r� �
DefaultParserc @ s e Zd Zdd�dd�ZdS )�ConnectCallbackProtocolro rp c C s d S rD r5 rr r5 r5 r6 �__call__' s z ConnectCallbackProtocol.__call__N�r0 r1 r2 r� r5 r5 r5 r6 r� & s r� c @ s e Zd Zdd�dd�ZdS )� AsyncConnectCallbackProtocolro rp c � s d S rD r5 rr r5 r5 r6 r� , s z%AsyncConnectCallbackProtocol.__call__Nr� r5 r5 r5 r6 r� + s r� c @ s� e Zd ZdZdZdddddddddddd ded
ddddded �eeee f eee f e
e e
e e
e e e
e
e ee ef f e e eee ee e e e
e e
e e
e e
e ee d �d d
�Zdd� Zdd� Zdd� Zedd� �Zdd� Zdd� Zee dd�dd �Zd d � Z d d � Z d!d"� Z dd#�d$d%�Z dd#�d&d'�Z d(d)� Z!d*d+� Z"d,d-� Z#e$e dd.�d/d0�Z%dFeeee$e f e dd2�d3d4�Z&e'e'dd5�d6d7�Z(dGe d8�d9d:�Z)dHe d;�d<d=�Z*dIe d;�d>d?�Z+e,e-e d@�dAdB�Z.e$e$e, e-e dC�dDdE�Z/dS )Jro z4Manages TCP communication to and from a Redis server)�pid�host�port�db�username� client_name�passwordr� �socket_connect_timeout�socket_keepalive�socket_keepalive_options� socket_type�redis_connect_func�retry_on_timeout�health_check_interval�next_health_checkZlast_active_atr� � ssl_contextr� �_writer�_parser�_connect_callbacks�_buffer_cutoff�_lock�_socket_read_size�__dict__Z localhosti� r NFr� �strict� )r� r� r� r� r� r� r� r� r� r� r: rB rC � parser_classr` r� r� r� �retryr� �
encoder_classc C s� t �� | _|| _t|�| _|| _|| _|| _|| _ || _
|pB|pBd | _ || _ |pRi | _
| | _|
| _|
r�|sztt� d�| _q�t�|�| _ntt� d�| _|| _d| _d | _|| | |
�| _|| _d | _d | _|| _ | � |� g | _ d| _ t �!� | _"d S )Nrg r r� �p )#�os�getpidr� r� rM r� r� r� r� r� r� r� r� r� r� r� r r r� �copy�deepcopyr� r� r� r� r� r� r� r� �
set_parserr� r� r� �Lockr� )rE r� r� r� r� r� r� r� r� r� r� r: rB rC r� r` r� r� r� r� r� r� r5 r5 r6 rF T s<
zConnection.__init__c C s, d� dd� | �� D ��}| jj� d|� d�S )N�,c s s | ]\}}|� d |� �V qdS )�=Nr5 )r� �k�vr5 r5 r6 � <genexpr>� r. z&Connection.__repr__.<locals>.<genexpr>�<�>)�join� repr_piecesrQ r0 )rE Z repr_argsr5 r5 r6 �__repr__� s zConnection.__repr__c C s6 d| j fd| jfd| jfg}| jr2|�d| jf� |S )Nr� r� r� r� )r� r� r� r� �append�rE �piecesr5 r5 r6 r� � s zConnection.repr_piecesc C sP z8| j r6t�� }| �� }|�� r,|�|� n
|�|� W n tyJ Y n0 d S rD )� is_connectedr� �get_event_loop�
disconnectZ
is_runningZ create_taskZrun_until_completer= )rE Zloop�coror5 r5 r6 rd � s zConnection.__del__c C s | j o
| jS rD )r� r� rc r5 r5 r6 r� � s zConnection.is_connectedc C s | j �t�|�� d S rD )r� r� �weakref�
WeakMethod)rE �callbackr5 r5 r6 �register_connect_callback� s z$Connection.register_connect_callbackc C s
g | _ d S rD )r� rc r5 r5 r6 �clear_connect_callbacks� s z"Connection.clear_connect_callbacks)r� rH c C s || j d�| _dS )z�
Creates a new instance of parser_class with socket size:
_socket_read_size and assigns it to the parser for the connection
:param parser_class: The required parser class
r_ N)r� r� )rE r� r5 r5 r6 r� � s zConnection.set_parserc
� sR | j r
dS z| �� I dH W n� tjy2 � Y n~ tjtjfyR td��Y n^ ty� } zt| � |���W Y d}~n6d}~0 t
y� } zt|�|�W Y d}~n
d}~0 0 z@| j s�| � � I dH n&t�
| j �r�| � | �I dH n| � | � W n$ t�y | �� I dH � Y n0 | jD ]0}|� }|| �}|�r t�|��r |I dH �q dS )z5Connects to the Redis server if not already connectedNz Timeout connecting to server)r� �_connectr� r� r� ru r% �OSErrorr �_error_messager= r� rs Ziscoroutinefunctionr# r� r� �inspectZ isawaitable)rE �e�exc�refr Ztaskr5 r5 r6 �connect� s8 " ��
zConnection.connectc � s t �| j�4 I dH �D tj| j| j| jr2| j�� ndd�I dH \}}W d �I dH qn1 I dH sd0 Y || _ || _
|j � d�}|r�|�
tjtjd� z@| jr�|�
tjtjd� | j�� D ]\}}|�
tj||� q�W n ttfy� |�� � Y n0 dS )z Create a TCP socket connectionN)r� r� �sslr� rg )r� ru r� r� Zopen_connectionr� r� r� rk r� r� Z transportZget_extra_infoZ
setsockoptr� Z IPPROTO_TCPZ TCP_NODELAYr� Z
SOL_SOCKETZ SO_KEEPALIVEr� �itemsZSOL_TCPr � TypeErrorr� )rE �reader�writerZsockr� r� r5 r5 r6 r � s& �: zConnection._connectc C sb t |j�dkr0d| j� d| j� d|jd � d�S d|jd � d| j� d| j� d|jd � d� S d S ) Nrg zError connecting to �:�. r �.�Error z connecting to )ri r� r� r� �rE � exceptionr5 r5 r6 r � s " ��zConnection._error_messager� c � s4 | j �| � | js| jr�| jr0| j| jp*df}n | jp8df}| jdg|�R ddi�I dH z| �� I dH }W n8 ty� | jd| jdd�I dH | �� I dH }Y n0 t|�dkr�td��| j r�| �d d
| j �I dH t| �� I dH �dkr�t
d ��| j �r0| �d | j �I dH t| �� I dH �dk�r0t
d
��dS )z=Initialize the connection, authenticate and select a database� ZAUTH� check_healthFN�r ZOKz Invalid Username or PasswordZCLIENTZSETNAMEzError setting client nameZSELECTzInvalid Database) r� rs r� r� � send_commandrz r r) r r� r r� )rE Z auth_argsZ
auth_responser5 r5 r6 rs s* zConnection.on_connectc � s z�t �| j�4 I dH �� | j�� | js@W d �I dH W dS z\z8t�� | jkrx| j �
� t | j d�rx| j � � I dH W n t
y� Y n0 W d| _d| _ nd| _d| _ 0 W d �I dH q�1 I dH s�0 Y W n& tjy� td| j� ��d�Y n0 dS )z!Disconnects from the Redis serverN� wait_closedz#Timed out closing connection after )r� ru r� r� rb r� r� r� r� r� r� �hasattrr r r� r� r% rc r5 r5 r6 r� * s,
�6
��zConnection.disconnectc � s6 | j ddd�I dH t| �� I dH �dkr2td��dS )z Send PING, expect PONG in returnZPINGFr NZPONGz#Bad response from PING health check)r r) rz r rc r5 r5 r6 �
_send_pingA s zConnection._send_pingc � s | � � I dH dS )z Function to call when PING failsN�r� )rE r� r5 r5 r6 � _ping_failedG s zConnection._ping_failedc � sR t jdd� dkrtj}ntj}| jrN|� �� | jkrN| j� | j
| j �I dH dS )z3Check the health of the connection with a PING/PONGr r* �� � N) �sys� version_infor� r� �get_running_loopr� �timer� r� Zcall_with_retryr r! )rE �funcr5 r5 r6 r K s
zConnection.check_health)�commandrH c � s | j �|� | j �� I d H d S rD )r� �
writelines�drain)rE r* r5 r5 r6 �_send_packed_commandU s z Connection._send_packed_commandT)r* r rH c
� s^ | j s| �� I d H n|r(| �� I d H zdt|t�r<|�� }t|t�rL|g}| jrnt� | �
|�| j�I d H n | j � |� | j �
� I d H W n� tjy� | �� I d H td�d �Y n� t�y6 } zf| �� I d H t|j�dkr�d|jd }}n|jd }|jd }td|� d|� d��|�W Y d }~n,d }~0 t�yX | �� I d H � Y n0 d S )NzTimeout writing to socketrg ZUNKNOWNr r z while writing to socket. r )r� r
r rI r< rJ rK r� r� �wait_forr- r� r+ r, r% r� r ri r� r �
BaseException)rE r* r r
Zerr_no�errmsgr5 r5 r6 �send_packed_commandY s@
�
��z Connection.send_packed_command)r� r� rH c � s& | j | j|� |�dd�d�I dH dS )z+Pack and send a command to the Redis serverr Tr N)r1 � pack_commandrk )rE r� r� r5 r5 r6 r ~ s �zConnection.send_commandr� c
� s� | j s| �� I dH z| j�|�I dH W S tyz } z:| �� I dH td| j� d| j� d|j � ���W Y d}~n
d}~0 0 dS )z8Poll the socket to see if there's data that can be read.N�Error while reading from r z: )
r� r
r� rw r r� r r� r� r� )rE ru r
r5 r5 r6 rw � s �zConnection.can_read�ry c
� s� z�| j 4 I dH �� | jrnt�| j�4 I dH �* | jj|d�I dH }W d �I dH q�1 I dH sb0 Y n| jj|d�I dH }W d �I dH q�1 I dH s�0 Y W n� tjy� | �� I dH td| j � d| j
� ���Y nv t �y: } z:| �� I dH t d| j � d| j
� d|j
� ���W Y d}~n,d}~0 t�y\ | �� I dH � Y n0 | j�r�tjdd� d k�r�tj}ntj}|� �� | j | _t|t��r�|d�|S �
z0Read the response from a previously sent commandNr4 zTimeout reading from r r3 z : r r* r" )r� r� r� ru r� rz r� r% r� r� r� r r r� r/ r� r% r& r� r' r( r� rI r$ �rE ry re r
r) r5 r5 r6 rz � s: �8�: � zConnection.read_responsec
� st zt| j r^t�| j �4 I dH �* | jj|d�I dH }W d �I dH qr1 I dH sR0 Y n| jj|d�I dH }W n� tjy� | �� I dH td| j� d| j � ���Y nt t
y� } z:| �� I dH t d| j� d| j � d|j � ���W Y d}~n,d}~0 t
�y | �� I dH � Y n0 | j�r^tjdd� d k�rFtj}ntj}|� �� | j | _t|t��rp|d�|S r5 )r� r� ru r� rz r� r% r� r� r� r r r� r/ r� r% r& r� r' r( r� rI r$ r6 r5 r5 r6 �read_response_without_lock� s8 �8� � z%Connection.read_response_without_lock)r� rH c G s* g }t |d t�rJ �t |d t�rFt|d �� �� �|dd� }n(d|d v rnt|d �� �|dd� }t�ttt |���� t
f�}| j }t | j
j|�D ]|}t |�}t |�|ks�||ks�t |t�r�t�|tt|��� t
f�}|�|� |�|� t
}q�t�|tt|��� t
|t
f�}q�|�|� |S )z2Pack a series of arguments into the Redis protocolr rg N� )rI rN r< �tuplerJ rh � SYM_EMPTYr� �SYM_STARri r� r� �mapr� rL �
SYM_DOLLARr� )rE r� �outputZbuff�
buffer_cutoff�argZ
arg_lengthr5 r5 r6 r2 � sB "
����
��
zConnection.pack_command)�commandsrH c C s� g }g }d}| j }|D ]~}| j|� D ]n}t|�}||ksJ||ksJt|t�rb|�t�|�� d}g }||kstt|t�r�|�|� q$|�|� ||7 }q$q|r�|�t�|�� |S )z.Pack multiple commands into the Redis protocolr )r� r2 ri rI rL r� r: r� ) rE rA r> r� Z
buffer_lengthr? �cmd�chunkZchunklenr5 r5 r6 �
pack_commands s. ���
zConnection.pack_commands)T)r )F)F)0r0 r1 r2 rV rW r� r@ r< r rM r rN rO r rK r rX r �ConnectCallbackTrF r� r� rd r� r� r r r� r
r r rs r� r r! r r r- r1 r r rw rz r7 r& r r2 rD r5 r5 r5 r6 ro 3 s� !�
� ;
% )
�� % $#.ro c s� e Zd Zdee ee eee ee ed�� fdd�
Zedd� �Zed d
� �Z ed d � �Z
ed
d� �Z edd� �Z edd� �Z
� ZS )�
SSLConnectionN�requiredF)� ssl_keyfile� ssl_certfile�
ssl_cert_reqs� ssl_ca_certs� ssl_ca_data�ssl_check_hostnamec s, t � jf i |�� t||||||d�| _d S )N��keyfile�certfile� cert_reqs�ca_certs�ca_data�check_hostname)r� rF �RedisSSLContextr� )rE rH rI rJ rK rL rM r� r� r5 r6 rF $ s
�zSSLConnection.__init__c C s | j jS rD )r� rO rc r5 r5 r6 rO 8 s zSSLConnection.keyfilec C s | j jS rD )r� rP rc r5 r5 r6 rP < s zSSLConnection.certfilec C s | j jS rD )r� rQ rc r5 r5 r6 rQ @ s zSSLConnection.cert_reqsc C s | j jS rD )r� rR rc r5 r5 r6 rR D s zSSLConnection.ca_certsc C s | j jS rD )r� rS rc r5 r5 r6 rS H s zSSLConnection.ca_datac C s | j jS rD )r� rT rc r5 r5 r6 rT L s z SSLConnection.check_hostname)NNrG NNF)r0 r1 r2 r r< rO rF r� rO rP rQ rR rS rT r� r5 r5 r� r6 rF # s4 ��
rF c @ sN e Zd ZdZd
ee ee ee ee ee ed�dd�Zej d�dd �Z
dS ) rU )rO rP rQ rR rS �contextrT NFrN c C sv || _ || _|d u r tj| _n<t|t�rZtjtjtjd�}||vrPt d|� ���|| | _|| _
|| _ || _ d | _
d S )N)ZnoneZoptionalrG z+Invalid SSL Certificate Requirements Flag: )rO rP r Z CERT_NONErQ rI r< Z
CERT_OPTIONALZ
CERT_REQUIREDr# rR rS rT rV )rE rO rP rQ rR rS rT Z CERT_REQSr5 r5 r6 rF \ s$
��
zRedisSSLContext.__init__r� c C sf | j s`t�� }| j|_| j|_| jr<| jr<|j| j| jd� | j sH| j
rZ|j | j | j
d� || _ | j S )N)rP rO )ZcafileZcadata) rV r Zcreate_default_contextrT rQ Z verify_moderP rO Zload_cert_chainrR rS Zload_verify_locations)rE rV r5 r5 r6 rk y s zRedisSSLContext.get)NNNNNF) r0 r1 r2 rW r r< rO rF r Z
SSLContextrk r5 r5 r5 r6 rU Q s �� rU c @ s� e Zd Zddddddddddedddddd �eeeef ee ee ee ee eee e e
e eeeee d
�d d �Z
eeeeeef f d
�dd�Zdd� Zdd� ZdS )�UnixDomainSocketConnectionr r Nr� r� Fr� g )�pathr� r� r� r� r� r: rB rC r� r� r` r� r� r� r� )rX r� r� r� r� r� r: rB rC r� r� r` r� r� r� c C s� t �� | _|| _|| _|| _|| _|| _|| _|p8|p8d| _ |
| _
|
rl|du r^t t � d�| _
qzt�|�| _
nt t � d�| _
|
| _d| _|| _t||| �| _d| _d| _d| _| | _| �| � g | _d| _t � � | _ dS )z�
Initialize a new UnixDomainSocketConnection.
To specify a retry policy, first set `retry_on_timeout` to `True`
then set `retry` to a valid `Retry` object
Nrg r r� r� ) r� r� r� rX r� r� r� r� r� r� r� r r r� r� r� r� r� r� r@ r� Z_sockr� r� r� r� r� r� r� r� r� )rE rX r� r� r� r� r� r: rB rC r� r� r` r� r� r� r� r5 r5 r6 rF � s4
z#UnixDomainSocketConnection.__init__r� c C s. d| j fd| jfg}| jr*|�d| jf� |S )NrX r� r� )rX r� r� r� r� r5 r5 r6 r� � s z&UnixDomainSocketConnection.repr_piecesc � sv t �| j�4 I d H �. tj| jd�I d H \}}W d �I d H qX1 I d H sN0 Y || _|| _| �� I d H d S )N)rX ) r� ru r� r� Zopen_unix_connectionrX r� r� rs )rE r r r5 r5 r6 r � s
Bz#UnixDomainSocketConnection._connectc C sR t |j�dkr(d| j� d|jd � d�S d|jd � d| j� d|jd � d�S d S )Nrg z!Error connecting to unix socket: r r r r z connecting to unix socket: )ri r� rX r r5 r5 r6 r � s ���z)UnixDomainSocketConnection._error_message)r0 r1 r2 r� r< r rM r rN rO r rX r rF r r r� r r r5 r5 r5 r6 rW � sH �
� 7"rW )�0�FZFALSE�NZNOr� c C s6 | d u s| dkrd S t | t�r.| �� tv r.dS t| �S )Nr F)rI r< �upper�
FALSE_STRINGSrO )rG r5 r5 r6 �to_bool� s
r^ )r� r� r� r� r� �max_connectionsr� rM .�URL_QUERY_ARGUMENT_PARSERSc @ sJ e Zd ZU eed< eed< ee ed< eed< eed< eed< eed< dS ) �
ConnectKwargsr� r� �connection_classr� r� r� rX N)r0 r1 r2 r< r> r ro rM r5 r5 r5 r6 ra � s
ra )�urlrH c
C s� t | �}i }t|j��� D ]t\}}|rt|�dkrt|d �}t�|�}|r�z||�||< W q� tt fy� t d|� d���Y q�0 q|||< q|j
r�t|j
�|d< |j r�t|j �|d< |j dkr�|j
r�t|j
�|d< t|d< n�|j d v �rx|j�rt|j�|d
< |j�rt|j�|d < |j
�rbd |v�rbz tt|j
��d
d��|d < W n tt f�y` Y n0 |j dk�r�t|d< nd}t d|� d���|S )Nr zInvalid value for `z` in connection URL.r� r� ZunixrX rb )Zredis�redissr� r� r� �/r rd z redis://, rediss://, unix://z5Redis URL must specify one of the following schemes (�))r r �queryr ri r r` rk r �
ValueErrorr� r� �schemerX rW �hostnamer� rM r� �AttributeErrorrF )rc �parsedr� �nameZ
value_listrG �parserZ
valid_schemesr5 r5 r6 � parse_url� sJ
�ro �_CP�ConnectionPool)�boundc @ s� e Zd ZdZeee eed�dd��Ze dfee e
e d�dd�Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Ze d�dd�Ze d�dd�Zd ed�d d �ZdS ) rq a�
Create a connection pool. ``If max_connections`` is set, then this
object raises :py:class:`~redis.ConnectionError` when the pool's
limit is reached.
By default, TCP connections are created unless ``connection_class``
is specified. Use :py:class:`~redis.UnixDomainSocketConnection` for
unix sockets.
Any additional keyword arguments are passed to the constructor of
``connection_class``.
)�clsrc rH c K s t |�}|�|� | f i |��S )a
Return a connection pool configured from the given URL.
For example::
redis://[[username]:[password]]@localhost:6379/0
rediss://[[username]:[password]]@localhost:6379/0
unix://[[username]:[password]]@/path/to/socket.sock?db=0
Three URL schemes are supported:
- `redis://` creates a TCP socket connection. See more at:
<https://www.iana.org/assignments/uri-schemes/prov/redis>
- `rediss://` creates a SSL wrapped TCP socket connection. See more at:
<https://www.iana.org/assignments/uri-schemes/prov/rediss>
- ``unix://``: creates a Unix Domain Socket connection.
The username, password, hostname, path and all querystring values
are passed through urllib.parse.unquote in order to replace any
percent-encoded values with their corresponding characters.
There are several ways to specify a database number. The first value
found will be used:
1. A ``db`` querystring option, e.g. redis://localhost?db=0
2. If using the redis:// or rediss:// schemes, the path argument
of the url, e.g. redis://localhost/0
3. A ``db`` keyword argument to this function.
If none of these options are specified, the default db=0 is used.
All querystring options are cast to their appropriate Python types.
Boolean arguments can be specified with string values "True"/"False"
or "Yes"/"No". Values that cannot be properly cast cause a
``ValueError`` to be raised. Once parsed, the querystring arguments
and keyword arguments are passed to the ``ConnectionPool``'s
class initializer. In the case of conflicting arguments, querystring
arguments always win.
)ro �update)rs rc r� Z url_optionsr5 r5 r6 �from_url? s (
zConnectionPool.from_urlN�rb r_ c K sp |pd}t |t�r|dk r"td��|| _|| _|| _t�� | _t �� | _
| | | | � � | j� dt
�| _d S )Nl r z,"max_connections" must be a positive integerr� )rI rM rh rb �connection_kwargsr_ � threadingr� �
_fork_lockr� r� �resetrk r@ r� )rE rb r_ rw r5 r5 r6 rF k s
zConnectionPool.__init__c C s" | j j� d| jf i | j���d�S )Nr� r� )rQ r0 rb rw rc r5 r5 r6 r� � s
��zConnectionPool.__repr__c C s, t �� | _d| _g | _t� | _t�� | _ d S r� )
r� r� r� �_created_connections�_available_connections�set�_in_use_connectionsr� r� r� rc r5 r5 r6 rz � s
zConnectionPool.resetc C sZ | j t�� krV| jjdd�}|s$t�z$| j t�� kr<| �� W | j�� n | j�� 0 d S )N� r� )r� r� r� ry �acquirer rz �release)rE Zacquiredr5 r5 r6 � _checkpid� s #
zConnectionPool._checkpidc � s | � � | j4 I dH �L z| j�� }W n ty@ | �� }Y n0 | j�|� W d �I dH qx1 I dH sn0 Y zv|�� I dH z |� � I dH r�t
d�d�W nF t
y� |� � I dH |�� I dH |� � I dH r�t
d�d�Y n0 W n& t �y | �
|�I dH � Y n0 |S )z Get a connection from the poolN�Connection has data�Connection not ready)r� r� r| �pop�
IndexError�make_connectionr~ �addr
rw r r� r/ r� �rE Z command_name�keys�optionsrq r5 r5 r6 �get_connection� s* 6 z ConnectionPool.get_connectionc C s. | j }| j|�dd�|�dd�|�dd�d�S )z,Return an encoder based on encoding settingsr: r� rB r� rC FrA )rw r� rk )rE r� r5 r5 r6 � get_encoder� s
�zConnectionPool.get_encoderc C s4 | j | jkrtd��| j d7 _ | jf i | j��S )zCreate a new connectionzToo many connectionsrg )r{ r_ r rb rw rc r5 r5 r6 r� � s z ConnectionPool.make_connectionrp c � s� | � � | j4 I dH �� z| j�|� W n ty: Y n0 | �|�rT| j�|� n2| jd8 _|� � I dH W d �I dH dS W d �I dH q�1 I dH s�0 Y dS )z(Releases the connection back to the poolNrg )
r� r� r~ �remove�KeyError�owns_connectionr| r� r{ r� rr r5 r5 r6 r� s
zConnectionPool.releasec C s |j | j kS rD )r� rr r5 r5 r6 r� s z ConnectionPool.owns_connectionT��inuse_connectionsc � s� | � � | j4 I dH �l |r,t| j| j�}n| j}tjdd� |D �ddi�I dH }tdd� |D �d�}|rn|�W d �I dH q�1 I dH s�0 Y dS )z�
Disconnects connections in the pool
If ``inuse_connections`` is True, disconnect connections that are
current in use, potentially by other tasks. Otherwise only disconnect
connections that are idle in the pool.
Nc s s | ]}|� � V qd S rD r �r� rq r5 r5 r6 r� ) r. z,ConnectionPool.disconnect.<locals>.<genexpr>�return_exceptionsTc s s | ]}t |t�r|V qd S rD �rI r/ �r� �rr5 r5 r6 r� , r. )r� r� r r| r~ r� �gather�next)rE r� Z connections�respr r5 r5 r6 r� s � �� zConnectionPool.disconnect)T)r0 r1 r2 rV � classmethodr rp r< ru ro r rM rF r� rz r� r� r� r� r� r� rO r� r5 r5 r5 r6 rq 1 s$
-�� /! c s| e Zd ZdZddeejfeee e e e ej
d�� fdd�
Z dd� Z d d
� Z
d d � Zed
�dd�Zded�dd�Z� ZS )�BlockingConnectionPoola
Thread-safe blocking connection pool::
>>> from redis.client import Redis
>>> client = Redis(connection_pool=BlockingConnectionPool())
It performs the same function as the default
:py:class:`~redis.ConnectionPool` implementation, in that,
it maintains a pool of reusable connections that can be shared by
multiple redis clients (safely across threads if required).
The difference is that, in the event that a client tries to get a
connection from the pool when all of connections are in use, rather than
raising a :py:class:`~redis.ConnectionError` (as the default
:py:class:`~redis.ConnectionPool` implementation does), it
makes the client wait ("blocks") for a specified number of seconds until
a connection becomes available.
Use ``max_connections`` to increase / decrease the pool size::
>>> pool = BlockingConnectionPool(max_connections=10)
Use ``timeout`` to tell it either how many seconds to wait for a connection
to become available, or to block forever:
>>> # Block forever.
>>> pool = BlockingConnectionPool(timeout=None)
>>> # Raise a ``ConnectionError`` after five seconds if a connection is
>>> # not available.
>>> pool = BlockingConnectionPool(timeout=5)
�2 � )r_ ru rb � queue_classc s, || _ || _| t� jf ||d�|�� d S )Nrv )r� ru r� rF )rE r_ ru rb r� rw r� r5 r6 rF S s ��z BlockingConnectionPool.__init__c C sN | � | j�| _z| j�d � W q tjy6 Y q:Y q0 qg | _t�� | _ d S rD )
r� r_ �pool�
put_nowaitr� � QueueFull� _connectionsr� r� r� rc r5 r5 r6 rz e s z BlockingConnectionPool.resetc C s" | j f i | j��}| j�|� |S )zMake a fresh connection.)rb rw r� r� rr r5 r5 r6 r� } s z&BlockingConnectionPool.make_connectionc � s: | � � d}zTt�| j�4 I dH �& | j�� I dH }W d �I dH q^1 I dH sT0 Y W n" tjtjfy� td��Y n0 |du r�| � � }zz|�
� I dH z |� � I dH r�td�d�W nJ t�y
|� � I dH |�
� I dH |� � I dH �rtd�d�Y n0 W n& t
�y4 | �|�I dH � Y n0 |S )a7
Get a connection, blocking for ``self.timeout`` until a connection
is available from the pool.
If the connection returned is ``None`` then creates a new connection.
Because we use a last-in first-out queue, the existing connections
(having been returned to the pool after the initial ``None`` values
were added) will be returned before ``None`` values. This means we only
create new connections when we need to, i.e.: the actual number of
connections will only increase in response to demand.
NzNo connection available.r� r� )r� r� ru r� rk r� Z
QueueEmptyr% r r� r
rw r� r/ r� r� r5 r5 r6 r� � s.
>z%BlockingConnectionPool.get_connectionrp c � sZ | � � | �|�s0|�� I dH | j�d� dS z| j�|� W n tjyT Y n0 dS )z)Releases the connection back to the pool.N)r� r� r� r� r� r� r� rr r5 r5 r6 r� � s
z BlockingConnectionPool.releaseTr� c � s� | � � | j4 I dH �T tjdd� | jD �ddi�I dH }tdd� |D �d�}|rV|�W d �I dH q�1 I dH sv0 Y dS )z(Disconnects all connections in the pool.Nc s s | ]}|� � V qd S rD r r� r5 r5 r6 r� � r. z4BlockingConnectionPool.disconnect.<locals>.<genexpr>r� Tc s s | ]}t |t�r|V qd S rD r� r� r5 r5 r6 r� � r. )r� r� r� r� r� r� )rE r� r� r r5 r5 r6 r� � s �� z!BlockingConnectionPool.disconnect)T)r0 r1 r2 rV ro r� Z LifoQueuerM r r ZQueuerF rz r� r� r� rO r� r� r5 r5 r� r6 r� 1 s #��5r� )ur� r� �enumr� r r� r� r� r r% rx r � itertoolsr �typesr �typingr r r r r r r
r r r
r � urllib.parser r r r r� Zredis.asyncio.retryr Z
redis.backoffr Z redis.compatr r Zredis.exceptionsr r r r r r r r r r r! r" r# r$ r% Z redis.typingr& r' Z redis.utilsr( r) r� �BlockingIOErrorZ EWOULDBLOCKZSSLWantReadErrorZSSLWantWriteErrorZSSLErrorr� r9 r� r� r; r= r� ZSYM_LFr: r� �Enumr/ r4 r� r{ r} r~ r| r7 r@ r< r= r rX r� r� r� r> r� r� r� rE ro rF rU rW r] rO r^ rM rN r` r3 ra ro rp rq r� r5 r5 r5 r6 �<module> s�
4
D� �)$B
Tm s.5Q��
4