mirror of https://github.com/jumpserver/jumpserver
parent
ba11e646d6
commit
b3e4c10bc2
|
@ -12,7 +12,10 @@ from common.utils.timezone import as_current_tz
|
||||||
from jumpserver.utils import current_request
|
from jumpserver.utils import current_request
|
||||||
from orgs.models import Organization
|
from orgs.models import Organization
|
||||||
from orgs.utils import get_current_org_id
|
from orgs.utils import get_current_org_id
|
||||||
|
from settings.models import Setting
|
||||||
from settings.serializers import SettingsSerializer
|
from settings.serializers import SettingsSerializer
|
||||||
|
from users.models import Preference
|
||||||
|
from users.serializers import PreferenceSerializer
|
||||||
from .backends import get_operate_log_storage
|
from .backends import get_operate_log_storage
|
||||||
|
|
||||||
logger = get_logger(__name__)
|
logger = get_logger(__name__)
|
||||||
|
@ -87,19 +90,15 @@ class OperatorLogHandler(metaclass=Singleton):
|
||||||
return log_id, before, after
|
return log_id, before, after
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def get_resource_display_from_setting(resource):
|
def get_resource_display(resource):
|
||||||
resource_display = None
|
if isinstance(resource, Setting):
|
||||||
setting_serializer = SettingsSerializer()
|
serializer = SettingsSerializer()
|
||||||
label = setting_serializer.get_field_label(resource)
|
resource_display = serializer.get_field_label(resource.name)
|
||||||
if label is not None:
|
elif isinstance(resource, Preference):
|
||||||
resource_display = label
|
serializer = PreferenceSerializer()
|
||||||
return resource_display
|
resource_display = serializer.get_field_label(resource.name)
|
||||||
|
else:
|
||||||
def get_resource_display(self, resource):
|
resource_display = str(resource)
|
||||||
resource_display = str(resource)
|
|
||||||
return_value = self.get_resource_display_from_setting(resource_display)
|
|
||||||
if return_value is not None:
|
|
||||||
resource_display = return_value
|
|
||||||
return resource_display
|
return resource_display
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
# coding: utf-8
|
|
||||||
from django.core.cache import cache
|
from django.core.cache import cache
|
||||||
from django.utils import translation
|
from django.utils import translation
|
||||||
from django.utils.translation import gettext_noop, gettext_lazy as _
|
from django.utils.translation import gettext_noop, gettext_lazy as _
|
||||||
|
from rest_framework import serializers
|
||||||
|
|
||||||
from common.utils import i18n_fmt
|
from common.utils import i18n_fmt
|
||||||
from .auth import (
|
from .auth import (
|
||||||
|
@ -20,11 +20,53 @@ from .security import SecuritySettingSerializer
|
||||||
from .terminal import TerminalSettingSerializer
|
from .terminal import TerminalSettingSerializer
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
|
'BaseSerializerWithFieldLabel',
|
||||||
'SettingsSerializer',
|
'SettingsSerializer',
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class BaseSerializerWithFieldLabel:
|
||||||
|
CACHE_KEY: str
|
||||||
|
ignore_iter_fields = True
|
||||||
|
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
|
self.fields_label_mapping = None
|
||||||
|
|
||||||
|
def extract_fields(self, serializer):
|
||||||
|
fields = {}
|
||||||
|
for field_name, field in serializer.get_fields().items():
|
||||||
|
if isinstance(field, serializers.Serializer):
|
||||||
|
fields.update(self.extract_fields(field))
|
||||||
|
else:
|
||||||
|
fields.update({field_name: field})
|
||||||
|
return fields
|
||||||
|
|
||||||
|
def get_field_label(self, field_name):
|
||||||
|
self.fields_label_mapping = cache.get(self.CACHE_KEY, None)
|
||||||
|
if self.fields_label_mapping is None:
|
||||||
|
self.fields_label_mapping = {}
|
||||||
|
with translation.override('en'):
|
||||||
|
cls = self.__class__
|
||||||
|
base_name = getattr(cls, 'PREFIX_TITLE', cls.__name__)
|
||||||
|
|
||||||
|
for subclass in cls.__bases__:
|
||||||
|
ignore_iter_fields = getattr(subclass, 'ignore_iter_fields', False)
|
||||||
|
if ignore_iter_fields:
|
||||||
|
continue
|
||||||
|
|
||||||
|
prefix = getattr(subclass, 'PREFIX_TITLE', base_name)
|
||||||
|
fields = self.extract_fields(subclass())
|
||||||
|
for name, item in fields.items():
|
||||||
|
label = getattr(item, 'label', '')
|
||||||
|
detail = i18n_fmt(gettext_noop('[%s] %s'), prefix, label)
|
||||||
|
self.fields_label_mapping[name] = detail
|
||||||
|
cache.set(self.CACHE_KEY, self.fields_label_mapping, 3600 * 24)
|
||||||
|
return self.fields_label_mapping.get(field_name, field_name)
|
||||||
|
|
||||||
|
|
||||||
class SettingsSerializer(
|
class SettingsSerializer(
|
||||||
|
BaseSerializerWithFieldLabel,
|
||||||
BasicSettingSerializer,
|
BasicSettingSerializer,
|
||||||
LDAPSettingSerializer,
|
LDAPSettingSerializer,
|
||||||
AuthSettingSerializer,
|
AuthSettingSerializer,
|
||||||
|
@ -49,25 +91,5 @@ class SettingsSerializer(
|
||||||
CustomSMSSettingSerializer,
|
CustomSMSSettingSerializer,
|
||||||
PasskeySettingSerializer
|
PasskeySettingSerializer
|
||||||
):
|
):
|
||||||
|
PREFIX_TITLE = _('Setting')
|
||||||
CACHE_KEY = 'SETTING_FIELDS_MAPPING'
|
CACHE_KEY = 'SETTING_FIELDS_MAPPING'
|
||||||
|
|
||||||
# encrypt_fields 现在使用 write_only 来判断了
|
|
||||||
def __init__(self, **kwargs):
|
|
||||||
super().__init__(**kwargs)
|
|
||||||
self.fields_label_mapping = None
|
|
||||||
|
|
||||||
# 单次计算量不大,搞个缓存,以防操作日志大量写入时,这里影响性能
|
|
||||||
def get_field_label(self, field_name):
|
|
||||||
self.fields_label_mapping = cache.get(self.CACHE_KEY, None)
|
|
||||||
if self.fields_label_mapping is None:
|
|
||||||
self.fields_label_mapping = {}
|
|
||||||
with translation.override('en'):
|
|
||||||
for subclass in SettingsSerializer.__bases__:
|
|
||||||
prefix = getattr(subclass, 'PREFIX_TITLE', _('Setting'))
|
|
||||||
fields = subclass().get_fields()
|
|
||||||
for name, item in fields.items():
|
|
||||||
label = getattr(item, 'label', '')
|
|
||||||
detail = i18n_fmt(gettext_noop('[%s] %s'), prefix, label)
|
|
||||||
self.fields_label_mapping[name] = detail
|
|
||||||
cache.set(self.CACHE_KEY, self.fields_label_mapping, 3600 * 24)
|
|
||||||
return self.fields_label_mapping.get(field_name)
|
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
from .koko import *
|
from .koko import *
|
||||||
from .lina import *
|
from .lina import *
|
||||||
from .luna import *
|
from .luna import *
|
||||||
|
from .preference import *
|
||||||
|
|
|
@ -0,0 +1,20 @@
|
||||||
|
from django.utils.translation import gettext_lazy as _
|
||||||
|
|
||||||
|
from settings.serializers import BaseSerializerWithFieldLabel
|
||||||
|
from .koko import KokoSerializer
|
||||||
|
from .lina import LinaSerializer
|
||||||
|
from .luna import LunaSerializer
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
'PreferenceSerializer',
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
class PreferenceSerializer(
|
||||||
|
BaseSerializerWithFieldLabel,
|
||||||
|
LinaSerializer,
|
||||||
|
LunaSerializer,
|
||||||
|
KokoSerializer
|
||||||
|
):
|
||||||
|
PREFIX_TITLE = _('Preference')
|
||||||
|
CACHE_KEY = 'PREFERENCE_FIELDS_MAPPING'
|
Loading…
Reference in New Issue