自定义外键属性

Question

我有一个名为Personnel 的模型，我将其用作User 模型的配置文件模型。这是我的模型

class Personnel(models.Model):
    user = models.OneToOneField(
        User
    )
    phone = models.CharField(
        null=False, max_length=50, verbose_name="Phone"
    )

我需要在 ModelForm 中使用上述模型。这是我的ModelForm：

class PersonnelForm(forms.ModelForm):
    class Meta:
        model = Personnel
        exclude = ('user')

    def __init__(self, *args, **kwargs):
        personnel = Personnel(user=User.objects.create(username=(''.join(choice(ascii_uppercase + digits) for x in range(30)))))
        super(PersonnelForm, self).__init__(
             instance=personnel, *args, **kwargs
        )

我不需要将user 字段显示在网络上，因此我将其排除在外。但是我仍然需要为它传递一个值，所以通过传递一个Personnel 实例来启动我的PersonnelForm。 User 模式有一个名为 email 的字段应该由用户输入，但我无法显示此字段，因为它不是我的 Personnel 模型上的字段。

我可以创建一个名为email 的虚拟字段，它在设置时设置user.email 字段的值，在获取时获取相同的值。像这样：

@property
def email():
    def fget(self):
        return 'fff'

    def fset(self, email):
        self.user.email = email

我需要以类似于上面的 sn-p 的方式使用它，因为我将使用 PersonnelForm 和 Django's generic CRUD views。

谢谢。

（请随时编辑并为标题建议一个更好的名称，以便其他人也能从中受益。）

---

使用丹尼尔的建议：

class PersonnelForm(forms.ModelForm):

    email = forms.EmailField(
        required=True, label=_("Email")
    )

    class Meta:
        model = Personnel
        exclude = ('user')

    def save(self, *args, **kwargs):
        commit = kwargs.get('commit', True)
        kwargs['commit'] = False
        personnel = super(PersonnelForm, self).save(*args, **kwargs)
        if not personnel.user: #To prevent creating a new user when updating.
            user = User.objects.create_user(
                email=self.cleaned_data['email'],
                username=(''.join(choice(ascii_uppercase + digits) for x in range(30))),
            )
        if commit:
            personnel.save()
        return personnel

这适用于使用 ModelForm 创建新记录，但我似乎无法更新电子邮件或在更新时查看电子邮件。

Answer 1

您的问题有两个误解。首先，即使您已将其排除在外，您也需要为该字段传递一个值；其次，您不能包含不在模型中的字段。这些都不是真的。

绝对可以定义一个单独的email 字段，将其与真实模型字段一起显示在表单上，​​然后使用该值创建User 对象。

鉴于您不想修改视图来执行此逻辑（IMO 是错误的决定，但这是您的代码），您可以在覆盖的 save 方法中执行此操作：

class PersonnelForm(forms.ModelForm):
    email = forms.CharField()

    class Meta:
        model = Personnel
        exclude = ('user',)

    def save(self, *args, **kwargs):
        # call super method with commit=False, but remember original commit arg
        commit = kwargs.get('commit', True)
        kwargs['commit'] = False
        new_object = super(PersonnelForm, self).save(*args, **kwargs)
        user = User.objects.create_user(
            email=self.cleaned_data['user'],
            username=get_random_username(),
            password=User.objects.make_random_password()
        )
        new_object.user = user
        # now save if commit was not explicitly set to False
        if commit:
            new_object.save()
        return new_object