mirror of https://github.com/jumpserver/jumpserver
133 lines
3.9 KiB
Python
133 lines
3.9 KiB
Python
# coding: utf-8
|
|
#
|
|
|
|
from django.utils.translation import ugettext as _
|
|
from django import forms
|
|
|
|
from orgs.mixins import OrgModelForm
|
|
from assets.models import SystemUser, Protocol
|
|
|
|
from ..models import RemoteApp
|
|
from .. import const
|
|
|
|
|
|
__all__ = [
|
|
'RemoteAppCreateUpdateForm',
|
|
]
|
|
|
|
|
|
class RemoteAppTypeChromeForm(forms.ModelForm):
|
|
chrome_target = forms.CharField(
|
|
max_length=128, label=_('Target URL'), required=False
|
|
)
|
|
chrome_username = forms.CharField(
|
|
max_length=128, label=_('Login username'), required=False
|
|
)
|
|
chrome_password = forms.CharField(
|
|
widget=forms.PasswordInput, strip=True,
|
|
max_length=128, label=_('Login password'), required=False
|
|
)
|
|
|
|
|
|
class RemoteAppTypeMySQLWorkbenchForm(forms.ModelForm):
|
|
mysql_workbench_ip = forms.CharField(
|
|
max_length=128, label=_('Database IP'), required=False
|
|
)
|
|
mysql_workbench_name = forms.CharField(
|
|
max_length=128, label=_('Database name'), required=False
|
|
)
|
|
mysql_workbench_username = forms.CharField(
|
|
max_length=128, label=_('Database username'), required=False
|
|
)
|
|
mysql_workbench_password = forms.CharField(
|
|
widget=forms.PasswordInput, strip=True,
|
|
max_length=128, label=_('Database password'), required=False
|
|
)
|
|
|
|
|
|
class RemoteAppTypeVMwareForm(forms.ModelForm):
|
|
vmware_target = forms.CharField(
|
|
max_length=128, label=_('Target address'), required=False
|
|
)
|
|
vmware_username = forms.CharField(
|
|
max_length=128, label=_('Login username'), required=False
|
|
)
|
|
vmware_password = forms.CharField(
|
|
widget=forms.PasswordInput, strip=True,
|
|
max_length=128, label=_('Login password'), required=False
|
|
)
|
|
|
|
|
|
class RemoteAppTypeCustomForm(forms.ModelForm):
|
|
custom_cmdline = forms.CharField(
|
|
max_length=128, label=_('Operating parameter'), required=False
|
|
)
|
|
custom_target = forms.CharField(
|
|
max_length=128, label=_('Target address'), required=False
|
|
)
|
|
custom_username = forms.CharField(
|
|
max_length=128, label=_('Login username'), required=False
|
|
)
|
|
custom_password = forms.CharField(
|
|
widget=forms.PasswordInput, strip=True,
|
|
max_length=128, label=_('Login password'), required=False
|
|
)
|
|
|
|
|
|
class RemoteAppTypeForms(
|
|
RemoteAppTypeChromeForm,
|
|
RemoteAppTypeMySQLWorkbenchForm,
|
|
RemoteAppTypeVMwareForm,
|
|
RemoteAppTypeCustomForm
|
|
):
|
|
pass
|
|
|
|
|
|
class RemoteAppCreateUpdateForm(RemoteAppTypeForms, OrgModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
# 过滤RDP资产和系统用户
|
|
super().__init__(*args, **kwargs)
|
|
field_asset = self.fields['asset']
|
|
field_asset.queryset = field_asset.queryset.filter(
|
|
protocols__name=Protocol.PROTOCOL_RDP
|
|
)
|
|
field_system_user = self.fields['system_user']
|
|
field_system_user.queryset = field_system_user.queryset.filter(
|
|
protocol=SystemUser.PROTOCOL_RDP
|
|
)
|
|
|
|
class Meta:
|
|
model = RemoteApp
|
|
fields = [
|
|
'name', 'asset', 'system_user', 'type', 'path', 'comment'
|
|
]
|
|
widgets = {
|
|
'asset': forms.Select(attrs={
|
|
'class': 'select2', 'data-placeholder': _('Asset')
|
|
}),
|
|
'system_user': forms.Select(attrs={
|
|
'class': 'select2', 'data-placeholder': _('System user')
|
|
})
|
|
}
|
|
|
|
def _clean_params(self):
|
|
app_type = self.data.get('type')
|
|
fields = const.REMOTE_APP_TYPE_MAP_FIELDS.get(app_type, [])
|
|
params = {}
|
|
for field in fields:
|
|
name = field['name']
|
|
value = self.cleaned_data[name]
|
|
params.update({name: value})
|
|
return params
|
|
|
|
def _save_params(self, instance):
|
|
params = self._clean_params()
|
|
instance.params = params
|
|
instance.save()
|
|
return instance
|
|
|
|
def save(self, commit=True):
|
|
instance = super().save(commit=commit)
|
|
instance = self._save_params(instance)
|
|
return instance
|