mirror of https://github.com/jumpserver/jumpserver
[Update] 优化 dynamic settings (#4107)
parent
08904c2a9f
commit
7ebe1c2916
@ -1,12 +1,46 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
#
|
#
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
from werkzeug.local import LocalProxy
|
||||||
|
|
||||||
from .conf import ConfigManager
|
from .conf import ConfigManager
|
||||||
|
from common.local import thread_local
|
||||||
|
|
||||||
__all__ = ['BASE_DIR', 'PROJECT_DIR', 'VERSION', 'CONFIG', 'DYNAMIC']
|
__all__ = ['BASE_DIR', 'PROJECT_DIR', 'VERSION', 'CONFIG', 'DYNAMIC', 'LOCAL_DYNAMIC_SETTINGS']
|
||||||
|
|
||||||
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||||
PROJECT_DIR = os.path.dirname(BASE_DIR)
|
PROJECT_DIR = os.path.dirname(BASE_DIR)
|
||||||
VERSION = '2.0.0'
|
VERSION = '2.0.0'
|
||||||
CONFIG = ConfigManager.load_user_config()
|
CONFIG = ConfigManager.load_user_config()
|
||||||
DYNAMIC = ConfigManager.get_dynamic_config(CONFIG)
|
DYNAMIC = ConfigManager.get_dynamic_config(CONFIG)
|
||||||
|
|
||||||
|
|
||||||
|
class _Settings:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def get_dynamic_cfg_from_thread_local():
|
||||||
|
KEY = 'dynamic_config'
|
||||||
|
|
||||||
|
try:
|
||||||
|
cfg = getattr(thread_local, KEY)
|
||||||
|
except AttributeError:
|
||||||
|
cfg = _Settings()
|
||||||
|
setattr(thread_local, KEY, cfg)
|
||||||
|
|
||||||
|
return cfg
|
||||||
|
|
||||||
|
|
||||||
|
class DynamicDefaultLocalProxy(LocalProxy):
|
||||||
|
def __getattr__(self, item):
|
||||||
|
try:
|
||||||
|
value = super().__getattr__(item)
|
||||||
|
except AttributeError:
|
||||||
|
value = getattr(DYNAMIC, item)()
|
||||||
|
setattr(self, item, value)
|
||||||
|
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
LOCAL_DYNAMIC_SETTINGS = DynamicDefaultLocalProxy(get_dynamic_cfg_from_thread_local)
|
||||||
|
Loading…
Reference in new issue