mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-06 01:15:05 -05:00
35 lines
791 B
Python
35 lines
791 B
Python
|
"""
|
||
|
|
The cache object API for implementing caches. The default is just a
|
||
|
|
dictionary, which in turns means it is not threadsafe for writing.
|
||
|
|
"""
|
||
|
|
from threading import Lock
|
||
|
|
|
||
|
|
class BaseCache(object):
|
||
|
|
|
||
|
|
def get(self, key):
|
||
|
|
raise NotImplemented()
|
||
|
|
|
||
|
|
def set(self, key, value):
|
||
|
|
raise NotImplemented()
|
||
|
|
|
||
|
|
def delete(self, key):
|
||
|
|
raise NotImplemented()
|
||
|
|
|
||
|
|
class DictCache(BaseCache):
|
||
|
|
|
||
|
|
def __init__(self, init_dict=None):
|
||
|
|
self.lock = Lock()
|
||
|
|
self.data = init_dict or {}
|
||
|
|
|
||
|
|
def get(self, key):
|
||
|
|
return self.data.get(key, None)
|
||
|
|
|
||
|
|
def set(self, key, value):
|
||
|
|
with self.lock:
|
||
|
|
self.data.update({key: value})
|
||
|
|
|
||
|
|
def delete(self, key):
|
||
|
|
with self.lock:
|
||
|
|
if key in self.data:
|
||
|
|
self.data.pop(key)
|