如何使用单个表单填充两个模型

Question

  class CustomAccount(models.Model):
        user = models.OneToOneField("auth.User")
        role =  models.CharField(max_length = 50,   default = 'student',    choices=APPROVAL_CHOICES)
        balance = models.FloatField( default = 0 )
        timezone = models.CharField(max_length = 200)
        def __str__(self):

            return self.user.username +" ["+ self.role + "]"

    class CustomAccountForm(forms.ModelForm):
        username = forms.CharField(max_length=30 )
        email = forms.EmailField(max_length=255 )
        password1 = forms.CharField(label= "Password",widget=forms.PasswordInput())
        password2 = forms.CharField(label="Password confirmation", widget=forms.PasswordInput , help_text="Enter the same password as above, for verification.")    

        def save(self, commit= True):
            user = User.objects.create_user(username = self.cleaned_data['username'], email = self.cleaned_data['email'] , password = self.cleaned_data['password1'])
            user.save()
            self.user = user
            return super(CustomAccountForm, self).save(commit=commit)

        def clean_username(self):
            username = self.cleaned_data["username"]
            try:
                User.objects.get(username=username)
            except User.DoesNotExist:
                return username
            raise forms.ValidationError("A user with that username already exists.")

        def clean_password2(self):
            password1 = self.cleaned_data.get("password1", "")
            password2 = self.cleaned_data["password2"]
            if password1 != password2:
                raise forms.ValidationError("The two password fields didn't match.")
            return password2
        def clean_email(self):
            email = self.cleaned_data["email"]
            try:
                User.objects.get(email=email)
            except User.DoesNotExist:
                return email
            raise forms.ValidationError("A user with that emailaddress already exists.")    
        class Meta:
            model = CustomAccount       
            exclude = ['balance','user']

我想在 Django Admin 部分使用具有 auth.User 和 CustomAccount 模型字段的单一表单创建自定义帐户。我在 /admin/mylogin/customaccount/add/ 收到错误 IntegrityError NOT NULL 约束失败：mylogin_customaccount.user_id

Answer 1

由于您收到非空错误，我认为您需要指定该字段不是必需的。这是我的一个应用程序中的一个示例。这表示表单字段不是必需的。

class arunModelForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        myuser = kwargs.pop('user', None)
        super(arunModelForm, self).__init__(*args, **kwargs)

        self.fields['nanoadded'].required = False
        self.fields['response'].required = False

我还将它包含在我的模型中，以便我的模型字段可以为空，如下所示：

class arun(models.Model):
    auser = models.CharField(max_length=50, null=True)
    adate = models.DateField(null=True)
    atime = models.TimeField(null=True)

如果我在完成这两项操作后仍然收到非空错误，我会在该字段中填写一个占位符值，以便我进一步调查问题。

Answer 2

您已经从表单中删除了用户，并且在 save 方法中，您在 save 方法中分配了 self.user。所以保存表单时，CustomAccountForm 不使用 user 属性，CustomAccount 的 user 字段为 None。

您的保存方法应如下所示：

def save(self, commit= True):
    user = User.objects.create_user(username =self.cleaned_data['username'], email = self.cleaned_data['email'] ,     password = self.cleaned_data['password1'])
    user.save()
    obj = super(CustomAccountForm, self).save(commit=False)
    obj.user = user
    obj.save()