Format: application/x-python-code
1 o
2
3 2'h� � @ s4 d dl Zd dlmZ d dlmZ G dd� de�ZdS )� N)� BaseCache)�RedisSerializerc @ s� e Zd ZU dZdZejed< dZejed< e � Z
4 d4d ejd
5 e d ej e
6 d e d
7 e dej eje
8 ejg e
9 f f dejfdd�Zde
10 fdd�Zdej e de fdd�Zde
11 dejfdd�Zde
12 dejej fd d �Zd5de
13 d ejdej e dejfd d �Zd5de
14 d ejdej e dejfd!d"�Z d5d#eje
15 ejf dej e dejej fd$d%�Zde
16 defd&d'�Zde
17 dejej fd(d)�Z de
18 defd*d+�Z defd,d-�Z d6de
19 d/e dejfd0d1�Z d6de
20 d/e dejfd2d3�Z dS )7�
21 RedisCachea Uses the Redis key-value store as a cache backend.
22
23 The first argument can be either a string denoting address of the Redis
24 server or an object resembling an instance of a redis.Redis class.
25
26 Note: Python Redis API already takes care of encoding unicode strings on
27 the fly.
28
29 :param host: address of the Redis server or an object which API is
30 compatible with the official Python Redis client (redis-py).
31 :param port: port number on which Redis server listens for connections.
32 :param password: password authentication for the Redis server.
33 :param db: db (zero-based numeric index) on Redis Server to connect.
34 :param default_timeout: the default timeout that is used if no timeout is
35 specified on :meth:`~BaseCache.set`. A timeout of
36 0 indicates that the cache never expires.
37 :param key_prefix: A prefix that should be added to all keys.
38
39 Any additional keyword arguments will be passed to ``redis.Redis``.
40 N� _read_client�
41 _write_client� localhost�� r �, �host�port�password�db�default_timeout�
42 key_prefix�kwargsc
43
44 K s� t �| |� |d u rtd��t|t�rGzdd l}W n ty* } ztd�| �d } ~ ww |�dd �r5td��|j d||||d�|�� | _
45 | _ n| | _ | _
46 |pPd| _ d S ) Nz)RedisCache host parameter may not be Noner zno redis module foundZdecode_responsesz0decode_responses is not supported by RedisCache.)r
47 r r r
48 � � )
49 r �__init__�
50 ValueError�
51 isinstance�str�redis� ImportError� RuntimeError�getZRedisr r r )
52 �selfr
53 r r r
54 r r r r �errr r �DC:\Users\shubh\Videos\agent\nova\Lib\site-packages\cachelib/redis.pyr ! s&
55
56
57 �� �� zRedisCache.__init__�returnc C s t | jt�r | jS | �� S �N)r r r �r r r r � _get_prefix< s ��zRedisCache._get_prefix�timeoutc C s t �| |�}|dkr d}|S )z�Normalize timeout by setting it to default of 300 if
58 not defined (None) or -1 if explicitly set to zero.
59
60 :param timeout: timeout to normalize.
61 r �����)r �_normalize_timeout)r r" r r r r$ A s z RedisCache._normalize_timeout�keyc C s | j �| j�| �� � |� ���S r )�
62 serializer�loadsr r r! �r r% r r r r L s �zRedisCache.get�keysc s< � j r
63 � fdd�|D �}nt|�}� fdd�� j�|�D �S )Nc � g | ]
64 }� � � � |� ��qS r �r! ��.0r% r r r �
65 <listcomp>S � z'RedisCache.get_many.<locals>.<listcomp>c s g | ]}� j �|��qS r )r& r' )r- �xr r r r. V � )r �listr Zmget�r r) Z
66 prefixed_keysr r r �get_manyQ s zRedisCache.get_many�valuec C s` | � |�}| j�|�}|dkr | jj| �� � |� �|d�}|S | jj| �� � |� �||d�}|S )Nr# ��namer5 �r7 r5 �time)r$ r& �dumpsr �setr! �setex)r r% r5 r" �dump�resultr r r r; X s
67 ���zRedisCache.setc C s^ | � |�}| j�|�}| jj| �� � |� �|d�}|r-|dkr-| jj| �� � |� �|d� |S )Nr6 r# )r7 r9 )r$ r& r: r Zsetnxr! Zexpire)r r% r5 r" r= �createdr r r �adde s
68 � zRedisCache.add�mappingc C s� | � |�}| jjdd�}|�� D ]*\}}| j�|�}|dkr,|j| �� � |� �|d� q|j| �� � |� �||d� q|� � }dd� t
69 |� � |�D �S )NF)Z transactionr# r6 r8 c S s g | ]\}}|r|�qS r r )r- �kZwas_setr r r r. r1 z'RedisCache.set_many.<locals>.<listcomp>) r$ r Zpipeline�itemsr& r: r; r! r< �execute�zipr) )r rA r" �piper% r5 r= �resultsr r r �set_manyp s
70 zRedisCache.set_manyc C � t | j�| �� � |� ���S r )�boolr �deleter! r( r r r rK � � zRedisCache.deletec sN |sg S � j r� fdd�|D �}ndd� |D �}� jj|� � fdd�|D �S )Nc r* r r+ r, r r r r. � r/ z*RedisCache.delete_many.<locals>.<listcomp>c S s g | ]}|�qS r r �r- rB r r r r. � s c s g | ] }� � |�s|�qS r )�hasrM r r r r. � s )r r rK r3 r r r � delete_many� s zRedisCache.delete_manyc C rI r )rJ r �existsr! r( r r r rN � rL zRedisCache.hasc C sH d}| j r| j�| �� d �}|r| jj|� }t|�S | j�� }t|�S )Nr �*)r r r) r! r rK ZflushdbrJ )r �statusr) r r r �clear� s
71 �zRedisCache.clear� �deltac C s | j j| �� � |� �|d�S �N)r7 �amount�r �incrr! �r r% rU r r r �inc� rL zRedisCache.incc C s | j j| �� � |� �| d�S rV rX rZ r r r �dec� s zRedisCache.dec)r r Nr r Nr )rT )!�__name__�
72 __module__� __qualname__�__doc__r �_t�Any�__annotations__r r r& �int�Optionalr �Union�Callabler r! r$ r �Listr4 r; r@ �DictrH rJ rK rO rN rS r[ r\ r r r r r s\
73 ������ �
74 � &&
75 ���
76
77 �
78
79 r )�typingra Z
80 cachelib.baser Zcachelib.serializersr r r r r r �<module> s