python - 使用自定义字段扩展用户模型表单

Question

注册后，我想请求用户:

• 全名(不过我想将其保存为名字和姓氏)
• 公司名称
• 电子邮件
• 密码

我在 StackOverflow 上阅读了数十个类似的情况。在 models.py 中，我像这样扩展了 User 模型:

# models.py
class UserProfile(models.Model):
  company = models.CharField(max_length = 50)
  user = models.OneToOneField(User)

def create_user_profile(sender, instance, created, **kwargs):
  if created:
    profile, created = UserProfile.objects.get_or_create(user=instance)

post_save.connect(create_user_profile, sender=User)

来源:Extending the User model with custom fields in Django

我还添加了:

# models.py

class SignupForm(UserCreationForm):
  fullname = forms.CharField(label = "Full name")
  company = forms.CharField(max_length = 50)
  email = forms.EmailField(label = "Email")
  password = forms.CharField(widget = forms.PasswordInput)

class Meta:
  model = User
  fields = ("fullname", "company", "email", "password")

def save(self, commit=True):
  user = super(SignupForm, self).save(commit=False)
  first_name, last_name = self.cleaned_data["fullname"].split()
  user.first_name = first_name
  user.last_name = last_name
  user.email = self.cleaned_data["email"]
  if commit:
    user.save()
  return user

在views.py中:

# views.py

@csrf_exempt
def signup(request):
  if request.method == 'POST':
    form = SignupForm(request.POST)
    if form.is_valid():
      new_user = form.save()
      first_name, last_name = request.POST['fullname'].split()
      email = request.POST['email']
      company = request.POST['company'],
      new_user = authenticate(
        username = email,
        password = request.POST['password']
      )
      # Log the user in automatically.
      login(request, new_user)

目前，它不存储公司名称。我该怎么做？

Answer 1

user_profile = new_user.get_profile()
user_profile.company = company
user_profile.save()

不要忘记在设置中配置您的 UserProfile 类，以便 Django 知道在 user.get_profile() 上返回什么