mirror of
https://github.com/RobbieHan/sandboxMP.git
synced 2026-02-03 02:43:14 +08:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
# @Time : 2018/12/19 16:13
|
||
# @Author : RobbieHan
|
||
# @File : forms.py
|
||
|
||
from django import forms
|
||
|
||
from .models import Code
|
||
|
||
|
||
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) |