jumpserver/apps/applications/forms/remote_app.py

#  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')
            })
        }