34 lines
856 B
Python
34 lines
856 B
Python
|
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
|