mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-14 12:06:23 +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
98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
import json
|
|
|
|
from django.db import models
|
|
from django.db.utils import ProgrammingError, OperationalError
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.core.cache import cache
|
|
|
|
from common.utils import get_signer
|
|
|
|
signer = get_signer()
|
|
|
|
|
|
class SettingQuerySet(models.QuerySet):
|
|
def __getattr__(self, item):
|
|
instances = self.filter(name=item)
|
|
if len(instances) == 1:
|
|
return instances[0]
|
|
else:
|
|
return Setting()
|
|
|
|
|
|
class SettingManager(models.Manager):
|
|
def get_queryset(self):
|
|
return SettingQuerySet(self.model, using=self._db)
|
|
|
|
|
|
class Setting(models.Model):
|
|
name = models.CharField(max_length=128, unique=True, verbose_name=_("Name"))
|
|
value = models.TextField(verbose_name=_("Value"))
|
|
category = models.CharField(max_length=128, default="default")
|
|
encrypted = models.BooleanField(default=False)
|
|
enabled = models.BooleanField(verbose_name=_("Enabled"), default=True)
|
|
comment = models.TextField(verbose_name=_("Comment"))
|
|
|
|
objects = SettingManager()
|
|
cache_key_prefix = '_SETTING_'
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
@classmethod
|
|
def get(cls, item):
|
|
cached = cls.get_from_cache(item)
|
|
if cached is not None:
|
|
return cached
|
|
instances = cls.objects.filter(name=item)
|
|
if len(instances) == 1:
|
|
s = instances[0]
|
|
s.refresh_setting()
|
|
return s.cleaned_value
|
|
return None
|
|
|
|
@classmethod
|
|
def get_from_cache(cls, item):
|
|
key = cls.cache_key_prefix + item
|
|
cached = cache.get(key)
|
|
return cached
|
|
|
|
@property
|
|
def cleaned_value(self):
|
|
try:
|
|
value = self.value
|
|
if self.encrypted:
|
|
value = signer.unsign(value)
|
|
if not value:
|
|
return None
|
|
value = json.loads(value)
|
|
return value
|
|
except json.JSONDecodeError:
|
|
return None
|
|
|
|
@cleaned_value.setter
|
|
def cleaned_value(self, item):
|
|
try:
|
|
v = json.dumps(item)
|
|
if self.encrypted:
|
|
v = signer.sign(v)
|
|
self.value = v
|
|
except json.JSONDecodeError as e:
|
|
raise ValueError("Json dump error: {}".format(str(e)))
|
|
|
|
@classmethod
|
|
def refresh_all_settings(cls):
|
|
try:
|
|
settings_list = cls.objects.all()
|
|
for setting in settings_list:
|
|
setting.refresh_setting()
|
|
except (ProgrammingError, OperationalError):
|
|
pass
|
|
|
|
def refresh_setting(self):
|
|
key = self.cache_key_prefix + self.name
|
|
cache.set(key, self.cleaned_value, None)
|
|
|
|
class Meta:
|
|
db_table = "settings_setting"
|
|
verbose_name = _("Setting")
|