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

Ответ

В 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 модель для всех операций, связанных с пользователями.

Ответ 18+ 🔞

А, ну вот, опять про пользователей в Джанге, да? Слушай, тут всё просто, как три копейки, но если накосячить в начале — потом будешь плакать кровавыми слезами, как тот Герасим, когда Муму топил, блядь.

Короче, есть два пути, как в сказке: налево пойдёшь — AbstractUser найдёшь, направо пойдёшь — AbstractBaseUser получишь. И оба, сука, ведут в итоге к своей кастомной модели.

AbstractUser — это как взять готовый торт и добавить сверху свою конфетку. Все стандартные поля (username, first_name, эти ваши) остаются, ты просто прилепляешь свои. Для ленивых, в общем.

AbstractBaseUser — это когда ты говоришь: «Да пошёл этот торт нахуй, я сам испеку!». Полная свобода, но и возни, блядь, овердохуища. Сам определяешь, что будет логином, что паролем, что ещё чем. И менеджера своего писать надо, а то нихуя не заработает.

Вот смотри, как на AbstractBaseUser забабахать, чтобы email вместо логина был. Это сейчас модно, все так делают.

Шаг 1: Лепим модель и менеджера в models.py

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) # Работничек админки

    # Самое важное! Говорим Джанге: "Смотри, сука, вот это поле — твой новый логин!"
    USERNAME_FIELD = 'email'
    # А тут поля, которые будут спрашивать при создании суперпользователя через createsuperuser
    REQUIRED_FIELDS = [] # У нас только email, остальное похуй

    objects = CustomUserManager() # Привязываем нашего менеджера

    def __str__(self):
        return self.email # Чтоб в админке красиво отображалось, а не "Объект №1488"

Шаг 2: Критически важный пиздец в settings.py

Вот тут, блядь, внимание! Это надо сделать ДО ПЕРВОЙ МИГРАЦИИ, а то потом будет больно, как от хуя в жопе вертолётом. Просто меняешь одну строчку:

# settings.py
AUTH_USER_MODEL = 'myapp.CustomUser' # Где 'myapp' — имя ТВОЕГО приложения, ёпта!

Шаг 3: Ну и стандартная магия миграций

python manage.py makemigrations
python manage.py migrate

Всё, пиздец. Теперь Джанга будет смотреть на твою CustomUser как на единственную и неповторимую. Все request.user, все связи — всё будет работать с ней. Главное — не проёбывай шаг с настройками, а то придётся базу сносить и начинать с чистого листа, и волнение ебать будет на максимум.