File: //proc/self/root/usr/lib/python3.6/site-packages/pyzor/engines/__pycache__/redis_.cpython-36.pyc
3
�Ugb[ � @ s� d Z ddlZddlZddlZddlZyddlZdZW n ek
rP dZdZY nX ddlT dZ de Z
dd � Z d
d � Z d d
� Z
G dd� de�ZG dd� de�Zes�eddddd�Zneeeded�ZdS )zRedis database engine.� NTF)�*�1zpyzord.digest_v%sc C s | dkr dS t tj| j� ��S )z"Convert the date to Unix TimestampNr )�int�timeZmktimeZ timetuple)Zdate� r � /usr/lib/python3.6/redis_.py� encode_date s r c C s t | �} | dkrdS tjj| �S )z/Return a datetime object from a Unix Timestamp.r N)r �datetimeZ
fromtimestamp)Zstampr r r � decode_date s r
c s � fdd�}|S )z?Decorator that wraps a method for handling database operations.c sV y� | f|�|�S t jjk
rP } z | jjd� j|� td��W Y d d }~X nX d S )Nz Redis error while calling %s: %sz!Database temporarily unavailable.)�redis�
exceptionsZ
RedisError�log�error�__name__Z
DatabaseError)�self�args�kwargs�e)�fr r � wrapped_f' s
z safe_call.<locals>.wrapped_fr )r r r )r r � safe_call$ s
r c @ s� e Zd ZdZdZejd�Zd%dd�Ze dd� �Z
e d d
� �Z d d � Z d
d� Z
dd� Zdd� Ze dd� �Zedd� �Zedd� �Zedd� �Zedd � �Zed d � �Zed d � �Zed&d!d"��Zd#d$� ZdS )'�
RedisDBHandleFTZpyzordNc C sd || _ || _|jd�}|d p d| _|d p.d| _|d p<d | _|d pJd| _| j� | _| j � d S ) N�,r Z localhost� Z6379� � �0)
�max_age�_dsn�split�host�port�passwd�db_name�_get_new_connection�db�_check_version)r �fn�moder r r r �__init__: s
zRedisDBHandle.__init__c C s. | j t| j�t| j�| jt| j�t| j�d�S )N)�r_count� r_entered� r_updated�wl_count�
wl_entered�
wl_updated)r* r r+ r, r- r. r/ )�rr r r �_encode_recordH s z RedisDBHandle._encode_recordc
C sf | s
t � S t t| jdd��t| jdd��t| jdd��t| jdd��t| jdd��t| jdd��d�S ) Ns r_countr s r_entereds r_updateds wl_counts
wl_entereds
wl_updated)r* r+ r, r- r. r/ )ZRecordr �getr
)r0 r r r �_decode_recordR s z RedisDBHandle._decode_recordc c s2 x,| j j| jd��D ]}|jdd�d V qW d S )Nr �.r ���)r% �keys� _real_key�rsplit)r �keyr r r �__iter__] s zRedisDBHandle.__iter__c c sX xR| D ]J}y|| | fV W q t k
rN } z| jjd||� W Y d d }~X qX qW d S )NzInvalid record %s: %s)� Exceptionr
Zwarning)r r9 Zexr r r �
_iteritemsa s
zRedisDBHandle._iteritemsc C s | j � S )N)r< )r r r r � iteritemsh s zRedisDBHandle.iteritemsc C s t | j� �S )N)�listr< )r r r r �itemsk s zRedisDBHandle.itemsc C s dt | f S )Nz%s.%s)� NAMESPACE)r9 r r r r7 n s zRedisDBHandle._real_keyc C sF d| j kr$tj| j t| j�| jd�S tj| j t| j�t| j�| jd�S )N�/)Zunix_socket_pathr% �password)r r! r% rB )r r Z StrictRedisr r# r" r! )r r r r r$ r s
z!RedisDBHandle._get_new_connectionc C s | j | jj| j|���S )N)r3 r% Zhgetallr7 )r r9 r r r � __getitem__z s zRedisDBHandle.__getitem__c C s< | j |�}| jj|| j|�� | jd k r8| jj|| j� d S )N)r7 r% Zhmsetr1 r �expire)r r9 �value�real_keyr r r � __setitem__~ s
zRedisDBHandle.__setitem__c C s | j j| j|�� d S )N)r% �deleter7 )r r9 r r r � __delitem__� s zRedisDBHandle.__delitem__c C sl t tj� �}xZ|D ]R}| j|�}| jj|d� | jj|d|� | jj|d|� | jr| jj|| j� qW d S )Nr* r+ r, ) r r r7 r% �hincrby�hsetnx�hsetr rD )r r6 �nowr9 rF r r r �report� s
zRedisDBHandle.reportc C sl t tj� �}xZ|D ]R}| j|�}| jj|d� | jj|d|� | jj|d|� | jr| jj|| j� qW d S )Nr- r. r/ ) r r r7 r% rJ rK rL r rD )r r6 rM r9 rF r r r � whitelist� s
zRedisDBHandle.whitelistc c s xt j| |||d�V qW dS )z^Yields a number of database connections suitable for a Pyzor
pre-fork server.
)r N)� functools�partial)�clsr' r( r r r r �get_prefork_connections� s z%RedisDBHandle.get_prefork_connectionsc C sD t | jjd��}|r@d| j| jf }| jjd|� | jjd|� dS )z8Check if there are deprecated records and warn the user.zpyzord.digest.*z?pyzor-migrate --delete --se=redis_v0 --sd=%s --de=redis --dd=%szBYou have %s records in the deprecated version of the redis engine.z#Please migrate the records with: %rN)�lenr% r6 r r
Zcritical)r Zold_keys�cmdr r r r& � s z RedisDBHandle._check_version)N)N)r �
__module__� __qualname__Zabsolute_sourceZhandles_one_step�loggingZ getLoggerr
r) � staticmethodr1 r3 r: r< r= r? r7 r r$ rC rG rI rN rO � classmethodrS r& r r r r r 4 s(
r c @ s e Zd Zddd�ZdS )�ThreadedRedisDBHandleNc C s t j| |||d� d S )N)r )r r) )r r' r( r Zboundr r r r) � s z ThreadedRedisDBHandle.__init__)NN)r rV rW r) r r r r r[ � s r[ )Zsingle_threadedZmulti_threadedZmulti_processingZprefork)�__doc__r rX r rP r Z
_has_redis� ImportErrorZpyzor.engines.common�VERSIONr@ r r
r Z
BaseEnginer r[ ZDBHandleZhandler r r r �<module> s6
~