我有两个模型,如下所示。
class Account(models.Model):
ASSET='A'
LIABILITY='L'
INCOME='I'
EXPENSE='E'
ACCOUNT_TYPE=((ASSET,'Asset'),
(LIABILITY,'Liability'),
(INCOME,'Income'),
(EXPENSE,'Expense'))
name=models.CharField(unique=True,db_index=True,max_length=70)
type=models.CharField(choices=ACCOUNT_TYPE,max_length=1)
class Person(models.Model):
first_name=models.CharField(max_length=30,)
last_name=models.CharField(max_length=30,)
account=models.OneToOneField(Account,on_delete=models.CASCADE)
Person 模型具有以下 CreateView 和模型表单。
class CreatePerson(CreateView):
model=Person
form_class=CreatePersonForm
class CreatePersonForm(forms.ModelForm):
display_name=forms.CharField()
class Meta:
model= Person
fields = ['first_name','last_name','display_name']
创建新Person 时,我需要
首先创建一个Account对象(name=display_name, type='A')并需要将其分配给Person对象。
如果具有相同 display_name 的 Account 已经存在,则再次呈现表单,并出现验证错误。
有人可以指出正确的方向来继续这里吗?
谢谢。
您可以实现clean_display_name
方法来检查 Acount 是否已经存在:
class CreatePersonForm(forms.ModelForm):
display_name=forms.CharField()
class Meta:
model= Person
fields = ['first_name','last_name','display_name']
def clean_display_name(self):
name = self.cleaned_data.get('display_name')
account = Account.objects.filter(name=name).exists()
if account:
raise forms.ValidationError('Account already exists')
return name
要将帐户添加到配置文件,您可以覆盖视图的form_valid
方法:
class CreatePerson(CreateView):
model=Person
form_class=CreatePersonForm
def form_valid(self, form):
account = Account.objects.create(name=form.cleaned_data['display_name'])
form.instance.account = account
return super().form_valid(form)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句