mirror of https://github.com/jumpserver/jumpserver
112 lines
3.3 KiB
Python
112 lines
3.3 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 Asset, SystemUser
|
|
|
|
from ..models import RemoteApp
|
|
|
|
|
|
__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(
|
|
protocol=Asset.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')
|
|
})
|
|
}
|
|
|