mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-15 14:47:24 +00:00
164f48e131
* [Update] 统一url地址 * [Update] 修改api * [Update] 使用规范的签名 * [Update] 修改url * [Update] 修改swagger * [Update] 添加serializer class避免报错 * [Update] 修改token * [Update] 支持api key * [Update] 支持生成api key * [Update] 修改api重定向 * [Update] 修改翻译 * [Update] 添加说明文档 * [Update] 修复浏览器关闭后session不失效的问题 * [Update] 修改一些内容 * [Update] 修改 jms脚本 * [Update] 修改重定向 * [Update] 修改搜索trim * [Update] 修改搜索trim * [Update] 添加sys log * [Bugfix] 修改登陆错误 * [Update] 优化User操作private_token的接口 (#3091) * [Update] 优化User操作private_token的接口 * [Update] 优化User操作private_token的接口 2 * [Bugfix] 解决授权了一个节点，当移动节点后，被移动的节点下的资产会放到未分组节点下的问题 * [Update] 升级jquery * [Update] 默认使用page * [Update] 修改使用Orgmodel view set * [Update] 支持 nv的硬盘 https://github.com/jumpserver/jumpserver/issues/1804 * [UPdate] 解决命令执行宽度问题 * [Update] 优化节点 * [Update] 修改nodes过多时创建比较麻烦 * [Update] 修改导入 * [Update] 节点获取更新 * [Update] 修改nodes * [Update] nodes显示full value * [Update] 统一使用nodes select2 函数 * [Update] 修改磁盘大小小数 * [Update] 修改 Node service * [Update] 优化授权节点 * [Update] 修改 node permission * [Update] 修改asset permission * [Stash] * [Update] 修改node assets api * [Update] 修改tree service，支持资产数量 * [Update] 修改暂时完成 * [Update] 修改一些bug
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django import forms
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from orgs.mixins.forms import OrgModelForm
|
|
from ..models import Domain, Asset, Gateway
|
|
from .user import PasswordAndKeyAuthForm
|
|
|
|
__all__ = ['DomainForm', 'GatewayForm']
|
|
|
|
|
|
class DomainForm(forms.ModelForm):
|
|
assets = forms.ModelMultipleChoiceField(
|
|
queryset=Asset.objects.all(), label=_('Asset'), required=False,
|
|
widget=forms.SelectMultiple(
|
|
attrs={'class': 'select2', 'data-placeholder': _('Select assets')}
|
|
)
|
|
)
|
|
|
|
class Meta:
|
|
model = Domain
|
|
fields = ['name', 'comment', 'assets']
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
if kwargs.get('instance', None):
|
|
initial = kwargs.get('initial', {})
|
|
initial['assets'] = kwargs['instance'].assets.all()
|
|
super().__init__(*args, **kwargs)
|
|
|
|
# 前端渲染优化, 防止过多资产
|
|
assets_field = self.fields.get('assets')
|
|
if not self.data:
|
|
instance = kwargs.get('instance')
|
|
if instance:
|
|
assets_field.queryset = instance.assets.all()
|
|
else:
|
|
assets_field.queryset = Asset.objects.none()
|
|
|
|
def save(self, commit=True):
|
|
instance = super().save(commit=commit)
|
|
assets = self.cleaned_data['assets']
|
|
instance.assets.set(assets)
|
|
return instance
|
|
|
|
|
|
class GatewayForm(PasswordAndKeyAuthForm, OrgModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
password_field = self.fields.get('password')
|
|
password_field.help_text = _('Password should not contain special characters')
|
|
protocol_field = self.fields.get('protocol')
|
|
protocol_field.choices = [Gateway.PROTOCOL_CHOICES[0]]
|
|
|
|
def save(self, commit=True):
|
|
# Because we define custom field, so we need rewrite :method: `save`
|
|
instance = super().save()
|
|
password = self.cleaned_data.get('password')
|
|
private_key, public_key = super().gen_keys()
|
|
instance.set_auth(password=password, private_key=private_key)
|
|
return instance
|
|
|
|
class Meta:
|
|
model = Gateway
|
|
fields = [
|
|
'name', 'ip', 'port', 'username', 'protocol', 'domain', 'password',
|
|
'private_key', 'is_active', 'comment',
|
|
]
|
|
help_texts = {
|
|
'protocol': _("SSH gateway support proxy SSH,RDP,VNC")
|
|
}
|
|
widgets = {
|
|
'name': forms.TextInput(attrs={'placeholder': _('Name')}),
|
|
'username': forms.TextInput(attrs={'placeholder': _('Username')}),
|
|
}
|