Как в Django заменить стандартную модель User на кастомную?

«Как в Django заменить стандартную модель User на кастомную?» — вопрос из категории Django, который задают на 10% собеседований Python Разработчик. Ниже — развёрнутый ответ с разбором ключевых моментов.

Ответ

В Django для создания кастомной модели пользователя рекомендуется наследоваться от AbstractUser или AbstractBaseUser.

  • AbstractUser: Используйте, если вы хотите добавить поля к стандартной модели User, сохранив ее поля (username, first_name и т.д.).
  • AbstractBaseUser: Используйте, если вам нужна полная свобода в определении полей модели. Требует больше настроек, включая создание менеджера.

Процесс состоит из следующих шагов:

Шаг 1: Создание модели и менеджера (пример с AbstractBaseUser)

В файле models.py вашего приложения определите менеджер и модель. В этом примере мы используем email как основной идентификатор.

from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
from django.db import models

class CustomUserManager(BaseUserManager):
    def create_user(self, email, password=None, **extra_fields):
        if not email:
            raise ValueError('Поле Email должно быть установлено')
        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_superuser(self, email, password=None, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        return self.create_user(email, password, **extra_fields)

class CustomUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)

    # Указываем, что поле email будет использоваться для входа
    USERNAME_FIELD = 'email'
    # Список имен полей, которые будут запрашиваться при создании суперпользователя
    REQUIRED_FIELDS = []

    objects = CustomUserManager()

    def __str__(self):
        return self.email

Шаг 2: Регистрация модели в settings.py

Важно: Этот шаг необходимо сделать в самом начале проекта, до создания первой миграции.

# settings.py
AUTH_USER_MODEL = 'myapp.CustomUser' # myapp - название вашего приложения

Шаг 3: Создание и применение миграций

После настройки модели и settings.py создайте и примените миграции:

python manage.py makemigrations
python manage.py migrate

Теперь Django будет использовать вашу CustomUser модель для всех операций, связанных с пользователями.