1

注册后,我想要求用户:

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

我在 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)

资料来源:在 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)

现在,它不存储公司名称。我怎么做?

4

1 回答 1

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

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

于 2012-04-20T08:05:49.440 回答