63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
import inspect
|
|
|
|
from utils.storage import BaseStorage
|
|
|
|
|
|
def _is_cache_item(obj):
|
|
return isinstance(obj, CacheItem)
|
|
|
|
|
|
class CacheItem(object):
|
|
|
|
def __init__(self, cache=None, name=None):
|
|
self.cache = cache
|
|
self.name = name
|
|
|
|
def key_name(self, key):
|
|
if isinstance(key, (tuple, list)):
|
|
key = ':'.join(key)
|
|
|
|
k = '{0}:{1}'.format(self.cache.prefix, self.name)
|
|
if key is not None:
|
|
k = '{0}:{1}'.format(k, key)
|
|
return k
|
|
|
|
def get(self, key=None, default=None):
|
|
return self.cache.storage.get(self.key_name(key), default)
|
|
|
|
def set(self, key=None, value=None, ttl=None):
|
|
return self.cache.storage.set(self.key_name(key), value, ttl)
|
|
|
|
def delete(self, key=None):
|
|
return self.cache.storage.delete(self.key_name(key))
|
|
|
|
|
|
class BaseCache(object):
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
self = super(BaseCache, cls).__new__(cls)
|
|
api_endpoints = inspect.getmembers(self, _is_cache_item)
|
|
for name, api in api_endpoints:
|
|
api_cls = type(api)
|
|
api = api_cls(self, name)
|
|
setattr(self, name, api)
|
|
return self
|
|
|
|
def __init__(self, storage, prefix='client'):
|
|
assert isinstance(storage, BaseStorage)
|
|
self.storage = storage
|
|
self.prefix = prefix
|
|
|
|
|
|
class WeWorkCache(BaseCache):
|
|
access_token = CacheItem()
|
|
|
|
|
|
class DingDingCache(BaseCache):
|
|
access_token = CacheItem()
|
|
|
|
|