mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-10-25 01:40:16 +00:00
4944ac8e75
* [Update] 修改config * [Update] 移动存储设置到到terminal中 * [Update] 修改permission 查看 * [Update] pre merge * [Update] 录像存储 * [Update] 命令存储 * [Update] 添加存储测试可连接性 * [Update] 修改 meta 值的 key 为大写 * [Update] 修改 Terminal 相关 Storage 配置 * [Update] 删除之前获取录像/命令存储的代码 * [Update] 修改导入失败 * [Update] 迁移文件添加default存储 * [Update] 删除之前代码，添加help_text信息 * [Update] 删除之前代码 * [Update] 删除之前代码 * [Update] 抽象命令/录像存储 APIView * [Update] 抽象命令/录像存储 APIView 1 * [Update] 抽象命令/录像存储 DictField * [Update] 抽象命令/录像存储列表页面 * [Update] 修复CustomDictField的bug * [Update] RemoteApp 页面添加 hidden * [Update] 用户页面添加用户关联授权 * [Update] 修改存储测试可连接性 target * [Update] 修改配置 * [Update] 修改存储前端 Form 渲染逻辑 * [Update] 修改存储细节 * [Update] 统一存储类型到 const 文件 * [Update] 修改迁移文件及Model，创建默认存储 * [Update] 修改迁移文件及Model初始化默认数据 * [Update] 修改迁移文件 * [Update] 修改迁移文件 * [Update] 修改迁移文件 * [Update] 修改迁移文件 * [Update] 修改迁移文件 * [Update] 修改迁移文件 * [Update] 修改迁移文件 * [Update] 限制删除默认存储配置，只允许创建扩展的存储类型 * [Update] 修改ip字段长度 * [Update] 修改ip字段长度 * [Update] 修改一些css * [Update] 修改关联 * [Update] 添加操作日志定时清理 * [Update] 修改记录syslog的instance encoder * [Update] 忽略登录产生的操作日志 * [Update] 限制更新存储时不覆盖原有AK SK 等字段 * [Update] 修改迁移文件添加comment字段 * [Update] 修改迁移文件 * [Update] 添加 comment 字段 * [Update] 修改默认存储no -> null * [Update] 修改细节 * [Update] 更新翻译（存储配置 * [Update] 修改定时任务注册，修改系统用户资产、节点关系api * [Update] 添加监控磁盘任务 * [Update] 修改session * [Update] 拆分serializer * [Update] 还原setting原来的manager
105 lines
3.0 KiB
Python
105 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from rest_framework import serializers
|
|
from django.utils import six
|
|
|
|
|
|
__all__ = [
|
|
'StringIDField', 'StringManyToManyField', 'ChoiceDisplayField',
|
|
'CustomMetaDictField'
|
|
]
|
|
|
|
|
|
class StringIDField(serializers.Field):
|
|
def to_representation(self, value):
|
|
return {"pk": value.pk, "name": value.__str__()}
|
|
|
|
|
|
class StringManyToManyField(serializers.RelatedField):
|
|
def to_representation(self, value):
|
|
return value.__str__()
|
|
|
|
|
|
class ChoiceDisplayField(serializers.ChoiceField):
|
|
def __init__(self, *args, **kwargs):
|
|
super(ChoiceDisplayField, self).__init__(*args, **kwargs)
|
|
self.choice_strings_to_display = {
|
|
six.text_type(key): value for key, value in self.choices.items()
|
|
}
|
|
|
|
def to_representation(self, value):
|
|
if value is None:
|
|
return value
|
|
return {
|
|
'value': self.choice_strings_to_values.get(six.text_type(value), value),
|
|
'display': self.choice_strings_to_display.get(six.text_type(value), value),
|
|
}
|
|
|
|
|
|
class DictField(serializers.DictField):
|
|
def to_representation(self, value):
|
|
if not value or not isinstance(value, dict):
|
|
value = {}
|
|
return super().to_representation(value)
|
|
|
|
|
|
class CustomMetaDictField(serializers.DictField):
|
|
"""
|
|
In use:
|
|
RemoteApp params field
|
|
CommandStorage meta field
|
|
ReplayStorage meta field
|
|
"""
|
|
type_map_fields = {}
|
|
default_type = None
|
|
need_convert_key = False
|
|
|
|
def filter_attribute(self, attribute, instance):
|
|
fields = self.type_map_fields.get(instance.type, [])
|
|
for field in fields:
|
|
if field.get('write_only', False):
|
|
attribute.pop(field['name'], None)
|
|
return attribute
|
|
|
|
def get_attribute(self, instance):
|
|
"""
|
|
序列化时调用
|
|
"""
|
|
attribute = super().get_attribute(instance)
|
|
attribute = self.filter_attribute(attribute, instance)
|
|
return attribute
|
|
|
|
def convert_value_key(self, dictionary, value):
|
|
if not self.need_convert_key:
|
|
# remote app
|
|
return value
|
|
tp = dictionary.get('type')
|
|
_value = {}
|
|
for k, v in value.items():
|
|
prefix = '{}_'.format(tp)
|
|
_k = k
|
|
if k.lower().startswith(prefix):
|
|
_k = k.lower().split(prefix, 1)[1]
|
|
_k = _k.upper()
|
|
_value[_k] = value[k]
|
|
return _value
|
|
|
|
def filter_value_key(self, dictionary, value):
|
|
tp = dictionary.get('type', self.default_type)
|
|
fields = self.type_map_fields.get(tp, [])
|
|
fields_names = [field['name'] for field in fields]
|
|
no_need_keys = [k for k in value.keys() if k not in fields_names]
|
|
for k in no_need_keys:
|
|
value.pop(k)
|
|
return value
|
|
|
|
def get_value(self, dictionary):
|
|
"""
|
|
反序列化时调用
|
|
"""
|
|
value = super().get_value(dictionary)
|
|
value = self.convert_value_key(dictionary, value)
|
|
value = self.filter_value_key(dictionary, value)
|
|
return value
|