django createviw-创建另一个模型对象并在当前模型中使用它

Question

我有两个模型，如下所示。

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 对象（名称=display_name，类型='A'）并需要将其分配给人对象。
如果具有相同 display_name 的帐户已经存在，则重新呈现表单，并出现验证错误。

有人可以指出正确的方向以继续这里吗？

谢谢。

Answer 1

您可以执行 clean_display_name 方法来检查帐户是否已经存在：

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)

django createviw-创建另一个模型对象并在当前模型中使用它

django createviw- create another model object and use it in current model

django

activity-oncreateview