Пользовательская модель пользователя в Django и суперпользователь

Я использую Django в качестве бэкенда для своего мобильного приложения. При создании информации о пользователях мне нужна следующая информация: имя, DoB и firebaseUID. Обратите внимание, что мне не нужен их email или пароль. Но я хочу создать суперпользователей, которые будут действовать как администраторы, и хочу, чтобы суперпользователи были созданы с использованием электронной почты и пароля. Поэтому я не уверен, должен ли я:

  1. Создайте пользовательского пользователя, расширяя AbstractBaseUser, и прикрепите менеджер пользователей к этому пользовательскому пользователю ИЛИ
  2. .
  3. Создайте пользовательского пользователя, расширяя AbstractBaseUser, а затем создайте другой класс Django user, который предназначен только для суперпользователей.
  4. .

Я реализовал вариант 1 с помощью приведенного ниже кода, но получаю следующую ошибку: "TypeError: MyUserManager.create_superuser() missing 1 required positional argument: 'email'" Также при создании суперпользователя Django запрашивает поля name, firebaseUID и DOB, хотя я хочу создавать суперпользователей только с полями email и password.



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

    



Пожалуйста, дайте мне знать, если вам нужна любая другая информация. Спасибо заранее!

Указана ли модель пользователя в файле settings.py?

Если нет, укажите это

AUTH_USER_MODEL = "yourapp.UserModel"

Также, в вашей модели CustomUser

Сделайте это

# For creating a superuser using Django
REQUIRED_FIELDS = []

И если вы хотите создавать пользователей с "email" и "password", а не с "username" и "password"

Сделайте это

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

Надеюсь, это поможет

Вернуться на верх