File: //proc/self/root/usr/lib/python3.6/site-packages/pyzor/engines/__pycache__/redis_v0.cpython-36.pyc
3
�Ugb: � @ s� d Z ddlZddlZddlZyddlZdZW n ek
rH dZdZY nX ddlT dZdd� Z d d� Z
d
d � Z G d d
� d
e �Z
G dd� de
�Zes�eddddd�Znee
ede
d�ZdS )z0Redis database engine.
XXX Deprecated version.
� NTF)�*z
pyzord.digestc C s | d kr dS | j d�S )N� z%Y-%m-%d %H:%M:%S)Zstrftime)�d� r � /usr/lib/python3.6/redis_v0.py�<lambda> s r c C s | dkr d S t j j| d�S )Nr z%Y-%m-%d %H:%M:%S)�datetimeZstrptime)�xr r r r s 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 dd � �Z d
d � Z d d
� Z
dd� Zdd� Ze dd� �Zedd� �Zedd� �Zedd� �Zedd� �Zed d d ��ZdS ) �
RedisDBHandleFZpyzordNc C sV || _ |jd�}|d pd| _|d p(d| _|d p6d | _|d pDd| _| j� | _d S ) N�,r Z localhost� Z6379� � �0)�max_age�split�host�port�passwd�db_name�_get_new_connection�db)r �fn�moder r r r �__init__1 s
zRedisDBHandle.__init__c C s4 d| j t| j�t| j�| jt| j�t| j�f j� S )Nz%s,%s,%s,%s,%s,%s)�r_count� encode_date� r_entered� r_updated�wl_count�
wl_entered�
wl_updated�encode)�rr r r �_encode_record= s z RedisDBHandle._encode_recordc C s` | d krt � S | j� jd�}t t|d �t|d �t|d �t|d �t|d �t|d �d�S ) Nr r r r r � � )r' r) r* r+ r, r- )ZRecord�decoder �int� decode_date)r/ Zfieldsr r r �_decode_recordG 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 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 r< Zexr r r �
_iteritemsW s
zRedisDBHandle._iteritemsc C s | j � S )N)r? )r r r r � iteritems^ s zRedisDBHandle.iteritemsc C s t | j� �S )N)�listr? )r r r r �itemsa s zRedisDBHandle.itemsc C s dt | f S )Nz%s.%s)� NAMESPACE)r< r r r r: d 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# rE )r r
Z StrictRedisr4 r! r r )r r r r r" h s
z!RedisDBHandle._get_new_connectionc C s | j | jj| j|���S )N)r6 r# �getr: )r r< r r r � __getitem__p s zRedisDBHandle.__getitem__c C sH | j d kr&| jj| j|�| j|�� n | jj| j|�| j | j|�� d S )N)r r# �setr: r0 Zsetex)r r< �valuer r r � __setitem__t s
zRedisDBHandle.__setitem__c C s | j j| j|�� d S )N)r# �deleter: )r r< r r r � __delitem__| s zRedisDBHandle.__delitem__c 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_connections)N)N)r �
__module__� __qualname__Zabsolute_sourceZhandles_one_step�loggingZ getLoggerr r& � staticmethodr0 r6 r= r? r@ rB r: r r" rG rJ rL � classmethodrP r r r r r + 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 rQ rR r& r r r r rV � s rV )Zsingle_threadedZmulti_threadedZmulti_processingZprefork)�__doc__rS r rM r
Z
_has_redis� ImportErrorZpyzor.engines.commonrC r( r5 r Z
BaseEnginer rV ZDBHandleZhandler r r r �<module> s2
^