创建一个Django表单以保存两个模型

时间:2013-04-08 22:00:22

标签: python django django-models django-forms

我有常规的Django User模型和UserDetails模型(OneToOneFieldUser),它是User模型的扩展。 (我尝试了Django 1.5的功能,这是令人头疼的奇怪可怕的文档,所以我坚持使用OneToOneField选项)

因此,在我构建一个自定义注册页面的过程中,该页面将包含由User字段和UserDetails字段组成的注册表单,我想知道是否有办法自动生成表单(包括所有验证)这两个相关模型。我知道这适用于由一个模型组成的表单:

class Meta:
    model = MyModel

但是,无论如何要为包含两个相关模型的表单获得类似的功能吗?

2 个答案:

答案 0 :(得分:14)

from django.forms.models import model_to_dict, fields_for_model


class UserDetailsForm(ModelForm):
    def __init__(self, instance=None, *args, **kwargs):
        _fields = ('first_name', 'last_name', 'email',)
        _initial = model_to_dict(instance.user, _fields) if instance is not None else {}
        super(UserDetailsForm, self).__init__(initial=_initial, instance=instance, *args, **kwargs)
        self.fields.update(fields_for_model(User, _fields))

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

    def save(self, *args, **kwargs):
        u = self.instance.user
        u.first_name = self.cleaned_data['first_name']
        u.last_name = self.cleaned_data['last_name']
        u.email = self.cleaned_data['email']
        u.save()
        profile = super(UserDetailsForm, self).save(*args,**kwargs)
        return profile

答案 1 :(得分:5)

你可以实现这一目标的一种方法,如果你想让ModelForm for User和UserDetails分开将两个表单都返回给前端,并确保它们都在同一个html表单元素中(即所有字段)将在发布数据时返回。)

如果User和UserDetails没有任何具有相同名称的字段,则此方法有效。要在每个ModelForm实例中输入数据,请使用常用方法:

    form_user = UserForm(request.POST, instance=request.user)
    form_user_details = UserDetailsForm(request.POST, instance=request.user.userdetails)