mirror of
https://github.com/moparisthebest/SickRage
synced 2024-08-13 16:53:54 -04:00
47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
from __future__ import division
|
|
|
|
from datetime import datetime
|
|
|
|
try:
|
|
from cPickle import loads, dumps
|
|
except ImportError: # Python 3.x
|
|
from pickle import loads, dumps
|
|
|
|
|
|
def total_seconds(td):
|
|
"""Python 2.6 compatability"""
|
|
if hasattr(td, 'total_seconds'):
|
|
return td.total_seconds()
|
|
|
|
ms = td.microseconds
|
|
secs = (td.seconds + td.days * 24 * 3600)
|
|
return (ms + secs * 10**6) / 10**6
|
|
|
|
|
|
class RedisCache(object):
|
|
|
|
def __init__(self, conn):
|
|
self.conn = conn
|
|
|
|
def get(self, key):
|
|
val = self.conn.get(key)
|
|
if val:
|
|
return loads(val)
|
|
return None
|
|
|
|
def set(self, key, value, expires=None):
|
|
if not expires:
|
|
self.conn.set(key, dumps(value))
|
|
else:
|
|
expires = expires - datetime.now()
|
|
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)
|