Auth和User Django的一个用户模型

Question

我使用Django身份验证功能创建用户。我还想扩展用户以添加个人资料图片和其他信息。

我的问题是我现在必须对用户模型进行管理。一种来自Auth，另一种来自我创建的模型。还是要合并为1个模型？

来自urls.py：

path('accounts/',include('django.contrib.auth.urls'))

来自models.py：

class User(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    firstName = models.CharField(max_length=200)
    lastName = models.CharField(max_length=200)
    image= models.ImageField(upload_to='profile_image',default='profile_image/SSMILE.jpg')

from views.py

class SignUp(generic.CreateView):
    form_class = UserCreationForm
    success_url = reverse_lazy('login')
    template_name = 'signup.html'

如您所见，我有2个用户模型。我必须最终通过注册创建auth用户，然后在我的主页视图中执行以下操作：

if str(request.user) != "AnonymousUser":
    try:
        getuser = User.objects.get(user=request.user)
        print(getuser)
    except User.DoesNotExist:
        newUser= User(user=request.user,firstName=request.user,lastName="change")
        newUser.save()

这是我想避免的黑客。

谢谢

Answer 1

您要添加另一个具有相同名称User的模型。它已经在那里。如果要扩展用户模型，则无需再添加first_name或last_name，

对此进行检查，

class Profile(models.Model):
   profile_picture = models.ImageField(upload_to=#add your path)
   user = models.OneToOneField(User, on_delete=models.CASCADE)


@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
      Profile.objects.create(user=instance)

@receiver(post_save, sender=User)
def save_user_profile(sender, instance, **kwargs):
     instance.profile.save()

这将扩展您现有的用户模型。