zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from __future__ import division
 
from datetime import datetime
from pip._vendor.cachecontrol.cache import BaseCache
 
 
class RedisCache(BaseCache):
 
    def __init__(self, conn):
        self.conn = conn
 
    def get(self, key):
        return self.conn.get(key)
 
    def set(self, key, value, expires=None):
        if not expires:
            self.conn.set(key, value)
        else:
            expires = expires - datetime.utcnow()
            self.conn.setex(key, int(expires.total_seconds()), value)
 
    def delete(self, key):
        self.conn.delete(key)
 
    def clear(self):
        """Helper for clearing all the keys in a database. Use with
        caution!"""
        for key in self.conn.keys():
            self.conn.delete(key)
 
    def close(self):
        """Redis uses connection pooling, no need to close the connection."""
        pass