Django中的自定义用户模型和超级用户

vyswwuz2  于 2023-01-21  发布在  Go
关注(0)|答案(1)|浏览(150)

我正在使用Django作为我的移动应用程序的后端。当创建用户信息时,我需要以下信息:名称,出生日期和firebaseUID。请注意,我不需要他们的电子邮件或密码。但我想创建作为管理员的超级用户,并希望超级用户使用电子邮件和密码创建。所以我不确定是否必须:
1.通过扩展AbstractBaseUser创建自定义用户,并将用户管理器连接到该自定义用户,或者
1.通过扩展AbstractBaseUser创建一个自定义用户,然后创建另一个仅针对超级用户的Django用户类。
我用下面的代码实现了选项1,但是得到了下面的错误:"类型错误:MyUserManager. create_superuser()缺少1个必需的位置参数:'email'"同样,当我创建超级用户时,Django要求输入名字,firebaseUID和DOB字段,尽管我只想创建超级用户的email和密码字段。

from django.contrib.auth.models import AbstractBaseUser
from django.db import models
from django.utils import timezone

class MyUserManager(models.Manager):

    def _create_user(self, email, password, **extra_fields):
        """
        Creates and saves a User with the given email and password.
        """
        if not email:
            raise ValueError('The given email must be set')
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_user(self, email=None, password=None, **extra_fields):
        extra_fields.setdefault('is_staff', False)
        extra_fields.setdefault('is_superuser', False)
        return self._create_user(email, password, **extra_fields)

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)

        if extra_fields.get('is_staff') is not True:
            raise ValueError('Superuser must have is_staff=True.')
        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')

        return self._create_user(email, password, **extra_fields)
    
    def get_by_natural_key(self, email):
        return self.get(email=email)
    

class CustomUser(AbstractBaseUser):
    name = models.CharField(null=True, blank=True, max_length=100)
    userName = models.CharField(
        null=True, blank=True, max_length=100, unique=True)
    firebaseUid = models.CharField(
        null=True, blank=True, max_length=100, unique=True)
    dob = models.DateField(max_length=8)
    created_at = models.DateTimeField(default=timezone.now)
    email = models.EmailField(unique=True) #used only for createing super user at this point 

    is_staff = models.BooleanField(default=False)
    is_superuser = models.BooleanField(default=False)
    # countryCode = models.CharField(null=True, blank=True, max_length=5)

    USERNAME_FIELD = 'userName'
    REQUIRED_FIELDS = ['name', 'firebaseUid', 'dob']

    objects = MyUserManager()

    def __str__(self):
        return self.userName

    def get_fBaseUid_name(self):
        return self.firebaseUid

    def has_perm(self, perm, obj=None):
        return self.is_superuser

    def has_module_perms(self, app_label):
        return self.is_superuser

如果您需要任何其他信息,请告诉我。谢谢!

oknrviil

oknrviil1#

是否已在www.example.com中指定用户模型settings.py?
如果没有,请指定它

AUTH_USER_MODEL = "yourapp.UserModel"

此外,在CustomUser模型中
这样做

# For creating a superuser using Django
REQUIRED_FIELDS = []

而且如果你想用"电子邮件"和"密码"创建用户,而不是"用户名"和"密码"
这样做

USERNAME_FIELD = "email"

您也可以将此代码用于create_superuser

def create_superuser(self, email, password=None):

   if password is None:
      raise TypeError("Password should not be none")

   user = self.create_user(email, password)
   user.is_superuser = True
   user.is_staff = True
   user.save()
   return user

希望能有所帮助

相关问题