mirror of
https://github.com/RobbieHan/sandboxMP.git
synced 2026-02-03 19:03:15 +08:00
105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
# @Time : 2018/12/19 16:13
|
||
# @Author : RobbieHan
|
||
# @File : forms.py
|
||
|
||
from django import forms
|
||
|
||
from .models import Code, DeviceInfo, ConnectionInfo, DeviceFile
|
||
|
||
|
||
class CodeCreateForm(forms.ModelForm):
|
||
class Meta:
|
||
model = Code
|
||
fields = '__all__'
|
||
|
||
error_messages = {
|
||
'key': {'required': 'key不能为空'},
|
||
'value': {'required': 'value不能为空'}
|
||
}
|
||
|
||
def clean(self):
|
||
cleaned_data = super(CodeCreateForm, self).clean()
|
||
key = cleaned_data.get('key')
|
||
value = cleaned_data.get('value')
|
||
|
||
if Code.objects.filter(key=key).count():
|
||
raise forms.ValidationError('key：{}已存在'.format(key))
|
||
|
||
if Code.objects.filter(value=value).count():
|
||
raise forms.ValidationError('value: {}已存在'.format(value))
|
||
|
||
|
||
class CodeUpdateForm(CodeCreateForm):
|
||
|
||
def clean(self):
|
||
cleaned_data = self.cleaned_data
|
||
key = cleaned_data.get('key')
|
||
value = cleaned_data.get('value')
|
||
|
||
if self.instance:
|
||
matching_code = Code.objects.exclude(pk=self.instance.pk)
|
||
if matching_code.filter(key=key).exists():
|
||
msg = 'key：{} 已经存在'.format(key)
|
||
raise forms.ValidationError(msg)
|
||
if matching_code.filter(value=value).exists():
|
||
msg = 'value：{} 已经存在'.format(value)
|
||
raise forms.ValidationError(msg)
|
||
|
||
|
||
class DeviceCreateForm(forms.ModelForm):
|
||
class Meta:
|
||
model = DeviceInfo
|
||
exclude = ['dev_connection']
|
||
error_messages = {
|
||
'hostname': {'required': '请填写设备地址'},
|
||
'buyDate': {'required': '请填写购买日期'},
|
||
'warrantyDate': {'required': '请填写到保日期'}
|
||
}
|
||
|
||
def clean(self):
|
||
cleaned_data = super(DeviceCreateForm, self).clean()
|
||
hostname = cleaned_data.get('hostname')
|
||
|
||
if DeviceInfo.objects.filter(hostname=hostname).count():
|
||
raise forms.ValidationError('设备地址：{}已存在'.format(hostname))
|
||
|
||
|
||
class DeviceUpdateForm(DeviceCreateForm):
|
||
def clean(self):
|
||
cleaned_data = self.cleaned_data
|
||
hostname = cleaned_data.get('hostname')
|
||
|
||
if self.instance:
|
||
matching_device = DeviceInfo.objects.exclude(pk=self.instance.pk)
|
||
if matching_device.filter(hostname=hostname).exists():
|
||
raise forms.ValidationError('设备地址：{}已存在'.format(hostname))
|
||
|
||
|
||
class ConnectionInfoForm(forms.ModelForm):
|
||
|
||
class Meta:
|
||
model = ConnectionInfo
|
||
fields = '__all__'
|
||
|
||
error_messages = {
|
||
'port': {'required': '端口不能为空'},
|
||
}
|
||
|
||
def clean(self):
|
||
cleaned_data = self.cleaned_data
|
||
username = cleaned_data.get('username')
|
||
password = cleaned_data.get('password')
|
||
private_key = cleaned_data.get('private_key')
|
||
auth_type = cleaned_data.get('auth_type')
|
||
if len(username) == 0:
|
||
raise forms.ValidationError('用户名不能为空！')
|
||
if auth_type == 'password' and len(password) == 0:
|
||
raise forms.ValidationError('认证类型为[密码]时，必须设置密码信息！')
|
||
if auth_type == 'private_key' and len(private_key) == 0:
|
||
raise forms.ValidationError('认证类型为[密钥]时，必须设置密钥信息！')
|
||
|
||
|
||
class DeviceFileUploadForm(forms.ModelForm):
|
||
class Meta:
|
||
model = DeviceFile
|
||
fields = '__all__' |