如何使用UserProfile中的字段擴展UserCreationForm

Question

我發現此 post關於如何使用諸如「email」等額外字段擴展UserCreationForm。但是，電子郵件字段已在預建的用戶模型中定義。

我創建了一個額外的模型（稱爲UserProfile），它進一步擴展了Django的預構建User類。如何讓我在UserProfile中定義的這些字段出現在我的UserCreationForm中？

Answer 1

添加字段以適合您的UserProfile模型（這不是太容易使用ModelForm避免重複自己，遺憾的是），然後創建並保存新UserProfile例如在過纏身save()功能。改編自您鏈接到的帖子：

from django import forms 
from django.contrib.auth.models import User 
from django.contrib.auth.forms import UserCreationForm 


class UserCreateForm(UserCreationForm): 
    job_title = forms.CharField(max_length=100, required=True) 
    age = forms.IntegerField(required=True) 

    class Meta: 
     model = User 

    def save(self, commit=True): 
     if not commit: 
      raise NotImplementedError("Can't create User and UserProfile without database save") 
     user = super(UserCreateForm, self).save(commit=True) 
     user_profile = UserProfile(user=user, job_title=self.cleaned_data['job_title'], 
      age=self.cleaned_data['age']) 
     user_profile.save() 
     return user, user_profile