| 12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- from __future__ import division
-
- from datetime import datetime
- from pip._vendor.cachecontrol.cache import BaseCache
-
-
- def total_seconds(td):
- """Python 2.6 compatability"""
- if hasattr(td, 'total_seconds'):
- return int(td.total_seconds())
-
- ms = td.microseconds
- secs = (td.seconds + td.days * 24 * 3600)
- return int((ms + secs * 10**6) / 10**6)
-
-
- 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, total_seconds(expires), 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
|
