Your IP : 216.73.216.213
B
�aG� � @ s� d dl mZ d dlmZ d dlmZ d dlmZmZmZ d dl m
Z
m Z m Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z" d dl#m$Z$m%Z% yd dl&Z&d Z'W n e(k
r� d
Z'Y nX e)e
j*iZ+e'�r6e,e&d ��r,d e+e&j-< d e+e&j.< n
d e+e&j/< e0e+�1� �Z2e$�r�d dl3Z3ee3j4�Z5e5ed
�kZ6e5ed�kZ7e5ed�kZ8e7�s�de3j4 Z9e�:e9� d Z;e7�s�d
Z;dZ<dZ=dZ>dZ?dZ@eA� ZBdZCdZDdZEdZFG dd� d�ZGG d d � d �ZHG d d � d �ZIG d d!� d!eH�ZJG d"d#� d#eH�ZKe$�r&eKZLneJZLG d$d%� d%�ZMG d&d'� d'eM�ZNG d(d)� d)eM�ZOd*ZPd+d,� ZQeReSeSeQeQeReReQd-�ZTd.d/� ZUG d0d1� d1�ZVG d2d3� d3eV�ZWdS )4� )�
StrictVersion)�chain)�time)� LifoQueue�Empty�Full)�parse_qs�unquote�urlparseN)�AuthenticationError�$AuthenticationWrongNumberOfArgsError�BusyLoadingError�ChildDeadlockedError�ConnectionError� DataError�ExecAbortError�InvalidResponse�NoPermissionError�
NoScriptError�
ReadOnlyError�
RedisError�
ResponseError� TimeoutError� ModuleError)�HIREDIS_AVAILABLE� str_if_bytesTF�SSLWantReadError� z0.1.3z0.1.4z1.0.0z`redis-py works best with hiredis >= 0.1.4. You're running hiredis %s. Please consider upgrading.� *� $s
� z Connection closed by server.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 @ s* e Zd ZdZdd� Zdd� Zd
dd�Zd S ) �Encoderz=Encode strings to bytes-like and decode bytes-like to stringsc C s || _ || _|| _d S )N)�encoding�encoding_errors�decode_responses)�selfr" r# r$ � r& �A/opt/alt/python37/lib/python3.7/site-packages/redis/connection.py�__init__] s zEncoder.__init__c C s� t |ttf�r|S t |t�r&td��n<t |ttf�rBt|��� }n t |t �sbt
|�j }td| ��t |t �r||�| j | j
�}|S )z=Return a bytestring or bytes-like representation of the valuezNInvalid input of type: 'bool'. Convert to a bytes, string, int or float first.zLInvalid input of type: '%s'. Convert to a bytes, string, int or float first.)�
isinstance�bytes�
memoryview�boolr �int�float�repr�encode�str�type�__name__r" r# )r% �value�typenamer& r& r' r0 b s
zEncoder.encodeFc C s: | j s
|r6t|t�r |�� }t|t�r6|�| j| j�}|S )z:Return a unicode string from the bytes-like representation)r$ r) r+ �tobytesr* �decoder" r# )r% r4 �forcer& r& r' r7 u s
zEncoder.decodeN)F)r3 �
__module__� __qualname__�__doc__r( r0 r7 r& r& r& r' r! Z s r! c @ sL e Zd Zdededededeeeeee ee
ei e e e
eeed�Zdd� Zd S )
�
BaseParserz 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 EXECABORTZLOADINGZNOSCRIPTZREADONLYZNOAUTHZNOPERMc C s\ |� d�d }|| jkrT|t|�d d� }| j| }t|t�rL|�|t�}||�S t|�S )zParse an error response� r � N)�split�EXCEPTION_CLASSES�lenr) �dict�getr )r% �responseZ
error_codeZexception_classr& r& r' � parse_error� s
zBaseParser.parse_errorN)r3 r9 r: 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@ rE r& r& r& r' r< s$ r< c @ sX e Zd Zdd� Zedd� �Zdedfdd�Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )� SocketBufferc C s, || _ || _|| _t�� | _d| _d| _d S )Nr )�_sock�socket_read_size�socket_timeout�io�BytesIO�_buffer�
bytes_written�
bytes_read)r% �socketrL rM r& r& r' r( � s
zSocketBuffer.__init__c C s | j | j S )N)rQ rR )r% r& r& r' �length� s zSocketBuffer.lengthNTc
C s@ | j }| j}| j}|�| j� d}|tk }� z�y||r>|�|� xh| j �|�} t| t �rjt
| �dkrjt t ��|�
| � t
| �}
| j|
7 _||
7 }|d k r�||kr�q@dS W nx tjk
r� |r�td��dS tk
�r" } z8t�| jd�} |�s| j| k�rdS t d| jf ��W d d } ~ X Y nX W d |�r:|�| j� X d S )Nr TzTimeout reading from socketF���z#Error while reading from socket: %s)rK rL rP �seekrQ �SENTINEL�
settimeout�recvr) r* rA r � SERVER_CLOSED_CONNECTION_ERROR�writerS �timeoutr �NONBLOCKING_EXCEPTIONS�#NONBLOCKING_EXCEPTION_ERROR_NUMBERSrC � __class__�errno�argsrM )
r% rT r\ �raise_on_timeout�sockrL �bufZmarker�custom_timeout�dataZ data_length�ex�allowedr& r& r' �_read_from_socket� s@
"z SocketBuffer._read_from_socketc C s t | j�p| j|dd�S )NF)r\ rb )r, rT ri )r% r\ r& r& r' �can_read� s
zSocketBuffer.can_readc C sn |d }|| j kr"| �|| j � | j�| j� | j�|�}| jt|�7 _| j| jkrb| �� |d d� S )Nr ���) rT ri rP rV rR �readrA rQ �purge)r% rT rf r& r& r' rl � s
zSocketBuffer.readc C sx | j }|�| j� |�� }x*|�t�sD| �� |�| j� |�� }q W | jt|�7 _| j| jkrl| � � |d d� S )Nrk )
rP rV rR �readline�endswith�SYM_CRLFri rA rQ rm )r% rd rf r& r& r' rn � s zSocketBuffer.readlinec C s& | j �d� | j �� d| _d| _d S )Nr )rP rV �truncaterQ rR )r% r& r& r' rm s
zSocketBuffer.purgec C s< y| � � | j�� W n tk
r* Y nX d | _d | _d S )N)rm rP �close� ExceptionrK )r% r& r& r' rr s zSocketBuffer.close)
r3 r9 r: r( �propertyrT rW ri rj rl rn rm rr r& r& r& r' rJ � s
*rJ c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� PythonParserzPlain Python parsing classc C s || _ d | _d | _d | _d S )N)rL �encoderrK rP )r% rL r& r& r' r( s zPythonParser.__init__c C s& y | � � W n tk
r Y nX d S )N)�
on_disconnectrs )r% r& r& r' �__del__& s zPythonParser.__del__c C s( |j | _ t| j | j|j�| _|j| _dS )z Called when the socket connectsN)rK rJ rL rM rP rv )r% �
connectionr& r& r' �
on_connect, s
zPythonParser.on_connectc C s* d| _ | jdk r | j�� d| _d| _dS )z"Called when the socket disconnectsN)rK rP rr rv )r% r& r& r' rw 4 s
zPythonParser.on_disconnectc C s | j o| j �|�S )N)rP rj )r% r\ r& r& r' rj < s zPythonParser.can_readc s
� j �� }|stt��|d d� |dd � }}|dkrDtd| ��|dkrv|jddd�}� �|�}t|t�rr|�|S |dkr�nn|d kr�t|�}n\|d
kr�t|�}|d kr�d S � j � |�}n2|d kr�t|�}|d kr�d S � fd
d�t
|�D �}t|t ��r� j �|�}|S )Nr> )� -� +� :r r zProtocol Error: %rr{ zutf-8�replace)�errorsr| r} r rU r c s g | ] }� � � �qS r& )�
read_response)�.0�i)r% r& r' �
<listcomp>g s z.PythonParser.read_response.<locals>.<listcomp>)
rP rn r rZ r r7 rE r) r- rl �ranger* rv )r% �rawZbyterD �errorrT r& )r% r' r� ? s:
zPythonParser.read_responseN)
r3 r9 r: r; r( rx rz rw rj r� r& r& r& r' ru s ru c @ sN e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze d fd
d�Z
dd� Z dS )�
HiredisParserz*Parser class for connections using Hiredisc C s$ t s td��|| _tr t|�| _d S )NzHiredis is not installed)r r rL �HIREDIS_USE_BYTE_BUFFER� bytearrayrP )r% rL r& r& r' r( o s
zHiredisParser.__init__c C s& y | � � W n tk
r Y nX d S )N)rw rs )r% r& r& r' rx w s zHiredisParser.__del__c C sd |j | _ |j| _t| jd�}ts(t|d< |jjr<|jj |d< t
rL|jj |d< t j
f |�| _d| _d S )N)Z
protocolError�
replyErrorr� r" r F)rK rM �_socket_timeoutr rE � HIREDIS_SUPPORTS_CALLABLE_ERRORSr rv r$ r" � HIREDIS_SUPPORTS_ENCODING_ERRORSr# �hiredis�Reader�_reader�_next_response)r% ry �kwargsr& r& r' rz } s
zHiredisParser.on_connectc C s d | _ d | _d| _d S )NF)rK r� r� )r% r& r& r' rw � s zHiredisParser.on_disconnectc C s@ | j stt��| jdkr<| j �� | _| jdkr<| j|dd�S dS )NF)r\ rb T)r� r rZ r� �gets�read_from_socket)r% r\ r& r& r' rj � s
zHiredisParser.can_readTc
C s$ | j }|tk }� z�y�|r"|�|� trX| j �| j�}|dkrDtt��| j� | jd|� n8| j �
| j �}t |t
�r|t|�dkr�tt��| j� |� dS tjk
r� |r�td��dS tk
�r } z4t�|jd�}|s�|j|kr�dS td|jf ��W d d }~X Y nX W d |�r |�| j� X d S )Nr TzTimeout reading from socketFrU z#Error while reading from socket: %s)rK rW rX r� Z recv_intorP r rZ r� ZfeedrY rL r) r* rA rS r\ r r] r^ rC r_ r` ra r� ) r% r\ rb rc re Zbufflen�bufferrg rh r& r& r' r� � s6
"z HiredisParser.read_from_socketc C s� | j stt��| jdk r(| j}d| _|S | j �� }x |dkrP| �� | j �� }q4W ts�t|t�rr| � |j
d �}n4t|t �r�|r�t|d t�r�| � |d j
d �|d< t|t�r�|�n$t|t �r�|r�t|d t�r�|d �|S )NFr ) r� r rZ r� r� r� r� r) r rE ra �list)r% rD r& r& r' r� � s,
zHiredisParser.read_responseN) r3 r9 r: r; r( rx rz rw rj rW r� r� r& r& r& r' r� m s &r� c @ s� e Zd ZdZddddddddddddded dddfd
d �Zd d
� Zdd� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
d d � Zd d � Zd d!� Zd/d#d$�Zd%d&� Zd0d'd(�Zd)d*� Zd+d,� Zd-d.� ZdS )1�
Connectionz4Manages TCP communication to and from a Redis serverZ localhosti� r NFzutf-8�stricti c C s� t �� | _|| _t|�| _|| _|| _|| _|| _ || _
|p>|| _ || _ |pNi | _
| | _|
| _|| _d| _t| | |
�| _d | _||d�| _g | _d| _d S )Nr )rL ip )�os�getpid�pid�hostr- �port�db�username� client_name�passwordrM �socket_connect_timeout�socket_keepalive�socket_keepalive_options� socket_type�retry_on_timeout�health_check_interval�next_health_checkr! rv rK �_parser�_connect_callbacks�_buffer_cutoff)r% r� r� r� r� rM r� r� r� r� r� r" r# r$ � parser_classrL r� r� r� r& r& r' r( � s(
zConnection.__init__c C s( d� dd� | �� D ��}d| jj|f S )N�,c S s g | ]\}}d ||f �qS )z%s=%sr& )r� �k�vr& r& r' r�
s z'Connection.__repr__.<locals>.<listcomp>z%s<%s>)�join� repr_piecesr_ r3 )r% Z repr_argsr& r& r' �__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)r% �piecesr& r& r' r� s zConnection.repr_piecesc C s& y | � � W n tk
r Y nX d S )N)�
disconnectrs )r% r& r& r' rx s zConnection.__del__c C s | j �|� d S )N)r� r� )r% �callbackr& r& r' �register_connect_callback s z$Connection.register_connect_callbackc C s
g | _ d S )N)r� )r% r& r& r' �clear_connect_callbacks# s z"Connection.clear_connect_callbacksc
C s� | j r
dS y | �� }W nN tjk
r4 td��Y n2 tjk
rd } zt| �|���W dd}~X Y nX || _ y | �� W n t k
r� | �
� � Y nX x| j D ] }|| � q�W dS )z5Connects to the Redis server if not already connectedNz Timeout connecting to server) rK �_connectrS r\ r r� r �_error_messagerz r r� r� )r% rc �er� r& r& r' �connect& s zConnection.connectc C s d}x�t �| j| j| jt j�D ]�}|\}}}}}d}y�t � |||�}|�t jt jd� | j r�|�t j
t j d� x&| j �
� D ]\} }
|�t j| |
� qzW |�| j� |�|� |�| j� |S tk
r� } z| }|dk r�|�� W dd} ~ X Y q X q W |dk �r|�td��dS )z Create a TCP socket connectionNr> z)socket.getaddrinfo returned an empty list)rS Z getaddrinfor� r� r� � SOCK_STREAMZ
setsockoptZ IPPROTO_TCPZ TCP_NODELAYr� Z
SOL_SOCKETZ SO_KEEPALIVEr� �itemsrX r� r� rM �OSErrorrr ) r% �err�resZfamilyZsocktype�protoZ canonnameZsocket_addressrc r� r� �_r& r& r' r� > s.
zConnection._connectc C sJ t |j�dkr&d| j| j|jd f S d|jd | j| j|jd f S d S )Nr> z Error connecting to %s:%s. %s.r z!Error %s connecting to %s:%s. %s.)rA ra r� r� )r% � exceptionr& r& r' r� f s
zConnection._error_messagec C s� | j �| � | js| jr�| jr0| j| jp*df}n| jf}| jd|�ddi� y | �� }W n. tk
r� | jd| jdd� | �� }Y nX t|�dkr�td��| j r�| �dd | j � t| �� �dkr�t
d
��| j r�| �d | j � t| �� �dkr�t
d ��d
S )z=Initialize the connection, authenticate and select a database� �AUTH� check_healthF)r� ZOKz Invalid Username or PasswordZCLIENTZSETNAMEzError setting client nameZSELECTzInvalid DatabaseN)r� ) r� rz r� r� � send_commandr� r r r r� r r� )r% Z auth_argsZ
auth_responser& r& r' rz p s* zConnection.on_connectc C sb | j �� | jdkrdS y*t�� | jkr6| j�tj� | j� � W n t
k
rV Y nX d| _dS )z!Disconnects from the Redis serverN) r� rw rK r� r� r� ZshutdownrS Z SHUT_RDWRrr r� )r% r& r& r' r� � s
zConnection.disconnectc C s� | j r�t� | jkr�y*| jddd� t| �� �dkr:td��W nF ttfk
r� | �� | jddd� t| �� �dkr~td��Y nX dS )z3Check the health of the connection with a PING/PONGZPINGF)r� ZPONGz#Bad response from PING health checkN) r� r r� r� r r� r r r� )r% r& r& r' r� � s
zConnection.check_healthTc
C s� | j s| �� |r| �� y.t|t�r,|g}x|D ]}| j �|� q2W W n� tjk
rn | �� t d��Y n� tj
k
r� } zP| �� t |j �dkr�d|j d }}n|j d }|j d }t
d||f ��W dd}~X Y n tk
r� | �� � Y nX dS )z2Send an already packed command to the Redis serverzTimeout writing to socketr> ZUNKNOWNr z%Error %s while writing to socket. %s.N)rK r� r� r) r1 ZsendallrS r\ r� r r� rA ra r �
BaseException)r% Zcommandr� �itemr� r` �errmsgr& r& r' �send_packed_command� s.
z Connection.send_packed_commandc O s | j | j|� |�dd�d� dS )z+Pack and send a command to the Redis serverr� T)r� N)r� � pack_commandrC )r% ra r� r& r& r' r� � s zConnection.send_commandc C s$ | j }|s| �� | j }| j�|�S )z8Poll the socket to see if there's data that can be read.)rK r� r� rj )r% r\ rc r& r& r' rj � s
zConnection.can_readc
C s� y| j �� }W n� tjk
r@ | �� td| j| jf ��Y n` tjk
r� } z$| �� t d| j| j|j
f ��W dd}~X Y n t k
r� | �� � Y nX | j r�t
� | j | _t|t�r�|�|S )z0Read the response from a previously sent commandzTimeout reading from %s:%sz#Error while reading from %s:%s : %sN)r� r� rS r\ r� r r� r� r� r ra r� r� r r� r) r )r% rD r� r& r& r' r� � s$ $
zConnection.read_responsec G s g }t |d t�r4t|d �� �� �|dd� }n(d|d kr\t|d �� �|dd� }t�ttt|���� t f�}| j
}x�t | j j|�D ]|}t|�}t|�|ks�||ks�t |t
�r�t�|tt|��� t f�}|�|� |�|� t }q�t�|tt|��� t |t f�}q�W |�|� |S )z2Pack a series of arguments into the Redis protocolr r> N� )r) r1 �tupler0 r? � SYM_EMPTYr� �SYM_STARrA rp r� �maprv r+ �
SYM_DOLLARr� )r% ra �outputZbuff�
buffer_cutoff�argZ
arg_lengthr& r& r' r� � s* "
zConnection.pack_commandc C s� g }g }d}| j }x�|D ]�}x|| j|� D ]n}t|�}||ksN||ksNt|t�rf|�t�|�� d}g }||ksxt|t�r�|�|� q(|�|� ||7 }q(W qW |r�|�t�|�� |S )z.Pack multiple commands into the Redis protocolr )r� r� rA r) r+ r� r� r� ) r% Zcommandsr� r� Z
buffer_lengthr� �cmd�chunkZchunklenr& r& r' �
pack_commands s&
zConnection.pack_commands)T)r )r3 r9 r: r; �
DefaultParserr( r� r� rx r� r� r� r� r� rz r� r� r� r� rj r� r� r� r& r& r& r' r� � s2
(
'
"r� c s* e Zd Zd� fdd� Z� fdd�Z� ZS ) �
SSLConnectionN�requiredFc s� t s td��t� jf |� || _|| _|d kr6tj}n8t|t �rntjtj
tj d�}||krftd| ��|| }|| _ || _
|| _d S )Nz$Python wasn't built with SSL support)ZnoneZoptionalr� z-Invalid SSL Certificate Requirements Flag: %s)�
ssl_availabler �superr( �keyfile�certfile�sslZ CERT_NONEr) r1 Z
CERT_OPTIONALZ
CERT_REQUIRED� cert_reqs�ca_certs�check_hostname)r% Z ssl_keyfileZ ssl_certfileZ
ssl_cert_reqsZ ssl_ca_certs�ssl_check_hostnamer� Z CERT_REQS)r_ r& r' r( 2 s&
zSSLConnection.__init__c sb t � �� }t�� }| j|_| j|_| jr@| jr@|j | j| jd� | j
rR|� | j
� |j || j
d�S )z Wrap the socket with SSL support)r� r� )Zserver_hostname)r� r� r� Zcreate_default_contextr� r� Z verify_moder� r� Zload_cert_chainr� Zload_verify_locationsZ wrap_socketr� )r% rc �context)r_ r& r' r� M s
zSSLConnection._connect)NNr� NF)r3 r9 r: r( r� �
__classcell__r& r& )r_ r' r� 0 s r� c
@ sH e Zd Zdddddddddedddf
dd �Zd
d � Zd d
� Zdd� ZdS )�UnixDomainSocketConnectionr� r Nzutf-8r� Fi c C sp t �� | _|| _|| _|| _|
| _|| _|| _| | _ | | _
d| _ t |||�| _
d | _|
| d�| _g | _d| _d S )Nr )rL ip )r� r� r� �pathr� r� r� r� rM r� r� r� r! rv rK r� r� r� )r% r� r� r� r� rM r" r# r$ r� r� rL r� r� r& r& r' r( ] s
z#UnixDomainSocketConnection.__init__c C s. d| j fd| jfg}| jr*|�d| jf� |S )Nr� r� r� )r� r� r� r� )r% r� r& r& r' r� s s
z&UnixDomainSocketConnection.repr_piecesc C s, t � t jt j�}|�| j� |�| j� |S )z&Create a Unix domain socket connection)rS ZAF_UNIXr� rX rM r� r� )r% rc r& r& r' r� | s z#UnixDomainSocketConnection._connectc C sB t |j�dkr"d| j|jd f S d|jd | j|jd f S d S )Nr> z(Error connecting to unix socket: %s. %s.r z+Error %s connecting to unix socket: %s. %s.)rA ra r� )r% r� r& r& r' r� � s
z)UnixDomainSocketConnection._error_message)r3 r9 r: r� r( r� r� r� r& r& r& r' r� [ s r� )�0�FZFALSE�NZNOc C s6 | d ks| dkrd S t | t�r.| �� tkr.dS t| �S )Nr� F)r) r1 �upper�
FALSE_STRINGSr, )r4 r& r& r' �to_bool� s
r� )r� rM r� r� r� �max_connectionsr� r� c
C s� t | �} i }x�t| j��� D ]r\}}|r t|�dkr t|d �}t�|�}|r�y||�||< W q� tt fk
r� t d| ��Y q�X q |||< q W | j
r�t| j
�|d< | j r�t| j �|d< | j dkr�| j
r�t| j
�|d< t|d< n�| j dk�r|| j�rt| j�|d < | j�rt| j�|d
< | j
�rfd |k�rfy tt| j
��d d
��|d < W n tt fk
�rd Y nX | j dk�r�t|d< nd}t d| ��|S )Nr z)Invalid value for `%s` in connection URL.r� r� Zunixr� �connection_class)Zredis�redissr� r� r� �/r� r� z redis://, rediss://, unix://z8Redis URL must specify one of the following schemes (%s))r
r Zqueryr� rA r �URL_QUERY_ARGUMENT_PARSERSrC � TypeError�
ValueErrorr� r� Zschemer� r� Zhostnamer� r- r~ �AttributeErrorr� )�urlr� �namer4 �parserZ
valid_schemesr& r& r' � parse_url� sJ
r� c @ st e Zd ZdZedd� �Zedfdd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zddd�ZdS )�ConnectionPoola�
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``.
c K s t |�}|�|� | f |�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.
)r� �update)�clsr� r� Z url_optionsr& r& r' �from_url� s (
zConnectionPool.from_urlNc K sJ |pd}t |t�r|dk r"td��|| _|| _|| _t�� | _| � � d S )Nl r z,"max_connections" must be a positive integer)
r) r- r� r� �connection_kwargsr� � threading�Lock�
_fork_lock�reset)r% r� r� r r& r& r' r( s
zConnectionPool.__init__c C s dt | �jt| jf | j��f S )Nz%s<%s>)r2 r3 r/ r� r )r% r& r& r' r� ( s zConnectionPool.__repr__c C s, t �� | _d| _g | _t� | _t�� | _ d S )Nr )
r r �_lock�_created_connections�_available_connections�set�_in_use_connectionsr� r� r� )r% r& r& r' r . s
zConnectionPool.resetc C sP | j t�� krL| jjdd�}|s$t�z| j t�� kr<| �� W d | j�� X d S )N� )r\ )r� r� r� r �acquirer r �release)r% Zacquiredr& r& r' � _checkpid? s # zConnectionPool._checkpidc O s� | � � | j�> y| j�� }W n tk
r: | �� }Y nX | j�|� W dQ R X yV|�� y|� � rnt
d��W n4 t
k
r� |� � |�� |� � r�t
d��Y nX W n t k
r� | �
|� � Y nX |S )z Get a connection from the poolNzConnection has datazConnection not ready)r
r r �pop�
IndexError�make_connectionr �addr� rj r r� r� r )r% � command_name�keys�optionsry r& r& r' �get_connectionn s*
z ConnectionPool.get_connectionc C s, | j }t|�dd�|�dd�|�dd�d�S )z,Return an encoder based on encoding settingsr" zutf-8r# r� r$ F)r" r# r$ )r r! rC )r% r� r& r& r' � get_encoder� s
zConnectionPool.get_encoderc C s0 | j | jkrtd��| j d7 _ | jf | j�S )zCreate a new connectionzToo many connectionsr> )r r� r r� r )r% r& r& r' r � s z ConnectionPool.make_connectionc C sv | � � | j�^ y| j�|� W n tk
r4 Y nX | �|�rN| j�|� n| jd8 _|� � dS W dQ R X dS )z(Releases the connection back to the poolr> N)
r
r r �remove�KeyError�owns_connectionr r� r r� )r% ry r& r& r' r � s
zConnectionPool.releasec C s |j | j kS )N)r� )r% ry r& r& r' r � s z ConnectionPool.owns_connectionTc C sN | � � | j�6 |r$t| j| j�}n| j}x|D ] }|�� q0W W dQ R X dS )z�
Disconnects connections in the pool
If ``inuse_connections`` is True, disconnect connections that are
current in use, potentially by other threads. Otherwise only disconnect
connections that are idle in the pool.
N)r
r r r r r� )r% Zinuse_connectionsZ connectionsry r& r& r' r� � s
zConnectionPool.disconnect)T)r3 r9 r: r; � classmethodr� r� r( r� r r
r r r r r r� r& r& r& r' r� � s ,/! r� c sR e Zd ZdZddeef� fdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z � Z S )�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 � c s( || _ || _t� jf ||d�|�� d S )N)r� r� )� queue_classr\ r� r( )r% r� r\ r� r r )r_ r& r' r( � s z BlockingConnectionPool.__init__c C sP | � | j�| _x,y| j�d � W q tk
r6 P Y qX qW g | _t�� | _d S )N) r r� �pool�
put_nowaitr � _connectionsr� r� r� )r% r& r& r' r � s z BlockingConnectionPool.resetc C s | j f | j�}| j�|� |S )zMake a fresh connection.)r� r r! r� )r% ry r& r& r' r s z&BlockingConnectionPool.make_connectionc O s� | � � d}y| jjd| jd�}W n tk
r> td��Y nX |dkrP| �� }yV|�� y|�� rltd��W n4 tk
r� |� � |�� |�� r�td��Y nX W n t
k
r� | � |� � Y nX |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.
NT)�blockr\ zNo connection available.zConnection has datazConnection not ready) r
r rC r\ r r r r� rj r� r� r )r% r r r ry r& r& r' r s,
z%BlockingConnectionPool.get_connectionc C sT | � � | �|�s*|�� | j�d� dS y| j�|� W n tk
rN Y nX dS )z)Releases the connection back to the pool.N)r
r r� r r r )r% ry r& r& r' r I s
z BlockingConnectionPool.releasec C s$ | � � x| jD ] }|�� qW dS )z(Disconnects all connections in the pool.N)r
r! r� )r% ry r& r& r' r� ^ s z!BlockingConnectionPool.disconnect)
r3 r9 r: r; r� r r( r r r r r� r� r& r& )r_ r' r � s
4r )XZdistutils.versionr � itertoolsr r Zqueuer r r Z urllib.parser r r
r` rN r� rS r �warningsZredis.exceptionsr r r
r r r r r r r r r r r r Z redis.utilsr r r� r� � ImportError�BlockingIOErrorZ EWOULDBLOCKr^ �hasattrr ZSSLWantWriteErrorZSSLErrorr� r r] r� � __version__Zhiredis_versionr� Z HIREDIS_SUPPORTS_BYTE_BUFFERr� �msg�warnr� r� r� rp r� rZ �objectrW rF rH rI rG r! r< rJ ru r� r� r� r� r� r� r� r- r. r� r� r� r r& r& r& r' �<module> s� D
%'xOz E+3 5 r