我有常规的Django User
模型和UserDetails
模型(OneToOneField
带User
),它是User
模型的扩展。 (我尝试了Django 1.5的功能,这是令人头疼的奇怪可怕的文档,所以我坚持使用OneToOneField
选项)
因此,在我构建一个自定义注册页面的过程中,该页面将包含由User
字段和UserDetails
字段组成的注册表单,我想知道是否有办法自动生成表单(包括所有验证)这两个相关模型。我知道这适用于由一个模型组成的表单:
class Meta:
model = MyModel
但是,无论如何要为包含两个相关模型的表单获得类似的功能吗?
答案 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)