mirror of
https://github.com/capricornxl/ad-password-self-service.git
synced 2025-08-09 03:43:36 +08:00
### 2021/05/19 -- 更新:
+ 添加了企业微信支持,修改pwdselfservice/local_settings.py中的SCAN_CODE_TYPE = 'DING'或SCAN_CODE_TYPE = 'WEWORK',区分使用哪个应用扫码验证 + 添加Reids缓存Token支持,如果不配置Redis则使用MemoryStorage缓存到内存中
This commit is contained in:
23
utils/storage/__init__.py
Normal file
23
utils/storage/__init__.py
Normal file
@@ -0,0 +1,23 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from __future__ import absolute_import, unicode_literals
|
||||
|
||||
|
||||
class BaseStorage(object):
|
||||
|
||||
def get(self, key, default=None):
|
||||
raise NotImplementedError()
|
||||
|
||||
def set(self, key, value, ttl=None):
|
||||
raise NotImplementedError()
|
||||
|
||||
def delete(self, key):
|
||||
raise NotImplementedError()
|
||||
|
||||
def __getitem__(self, key):
|
||||
self.get(key)
|
||||
|
||||
def __setitem__(self, key, value):
|
||||
self.set(key, value)
|
||||
|
||||
def __delitem__(self, key):
|
||||
self.delete(key)
|
58
utils/storage/cache.py
Normal file
58
utils/storage/cache.py
Normal file
@@ -0,0 +1,58 @@
|
||||
# -*- 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()
|
||||
|
||||
|
37
utils/storage/kvstorage.py
Normal file
37
utils/storage/kvstorage.py
Normal file
@@ -0,0 +1,37 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from __future__ import absolute_import, unicode_literals
|
||||
|
||||
import json
|
||||
|
||||
from dingtalk.core.utils import to_text
|
||||
from utils.storage import BaseStorage
|
||||
|
||||
|
||||
class KvStorage(BaseStorage):
|
||||
|
||||
def __init__(self, kvdb, prefix='wework'):
|
||||
for method_name in ('get', 'set', 'delete'):
|
||||
assert hasattr(kvdb, method_name)
|
||||
self.kvdb = kvdb
|
||||
self.prefix = prefix
|
||||
|
||||
def key_name(self, key):
|
||||
return '{0}:{1}'.format(self.prefix, key)
|
||||
|
||||
def get(self, key, default=None):
|
||||
key = self.key_name(key)
|
||||
value = self.kvdb.get(key)
|
||||
if value is None:
|
||||
return default
|
||||
return json.loads(to_text(value))
|
||||
|
||||
def set(self, key, value, ttl=None):
|
||||
if value is None:
|
||||
return
|
||||
key = self.key_name(key)
|
||||
value = json.dumps(value)
|
||||
self.kvdb.set(key, value, ttl)
|
||||
|
||||
def delete(self, key):
|
||||
key = self.key_name(key)
|
||||
self.kvdb.delete(key)
|
32
utils/storage/memorystorage.py
Normal file
32
utils/storage/memorystorage.py
Normal file
@@ -0,0 +1,32 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from __future__ import absolute_import, unicode_literals
|
||||
|
||||
import time
|
||||
|
||||
from utils.storage import BaseStorage
|
||||
|
||||
|
||||
class MemoryStorage(BaseStorage):
|
||||
|
||||
def __init__(self):
|
||||
self._data = {}
|
||||
|
||||
def get(self, key, default=None):
|
||||
ret = self._data.get(key, None)
|
||||
if ret is None or len(ret) != 2:
|
||||
return default
|
||||
else:
|
||||
value = ret[0]
|
||||
expires_at = ret[1]
|
||||
if expires_at is None or expires_at > time.time():
|
||||
return value
|
||||
else:
|
||||
return default
|
||||
|
||||
def set(self, key, value, ttl=None):
|
||||
if value is None:
|
||||
return
|
||||
self._data[key] = (value, int(time.time()) + ttl)
|
||||
|
||||
def delete(self, key):
|
||||
self._data.pop(key, None)
|
Reference in New Issue
Block a user