Можно ли добавить метод create_user_with_permission так же, как create_user в django?

Я делаю сайт на django и я создал группы для доступа к странице администратора (Admin, Content Writer, Users). Я хочу, чтобы когда пользователь регистрируется, ему присваивалась определенная группа в зависимости от того, с какой формы он входит/регистрируется.

Вот мой models.py файл для accounts приложения

models.py

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


# Custom User Manager
class CustomUserManager(BaseUserManager):
    def _create_user(self, email, password, first_name, last_name=None, **extra_fields):
        if (not email):
            raise ValueError("Email Must Be Provided")
        if (not password):
            raise ValueError("Password is not Provided")

        user = self.model(
            email=self.normalize_email(email),
            first_name=first_name,
            last_name=last_name,
            **extra_fields
        )

        user.set_password(password)
        user.save(using=self._db)

        return user

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

    def create_superuser(self, email, password, first_name, last_name=None, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_active', 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, first_name, last_name, **extra_fields)


# Custom user Model
class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(db_index=True, unique=True, max_length=254)
    first_name = models.CharField(max_length=255)
    last_name = models.CharField(max_length=255, null=True, blank=True)
    mobile = models.CharField(max_length=50, null=True, blank=True)
    address = models.CharField(max_length=250, null=True, blank=True)
    # profile_pic = models.ImageField(null=True, blank=True)

    is_staff = models.BooleanField(default=True)
    is_active = models.BooleanField(default=True)
    is_superuser = models.BooleanField(default=False)

    objects = CustomUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['first_name']

    class Meta:
        verbose_name = 'User'
        verbose_name_plural = 'Users'

вот как выглядят мои Groups в admin page

enter image description here

В моем классе CustomUserManager можно добавить еще один метод, например create_content_writer, который имеет функциональность create_user, но устанавливает is_staff в true и is_active в true и добавляет этого пользователя в группу Content Writer при создании.

Или если есть более эффективный способ сделать это, то, пожалуйста, скажите, потому что в будущем, если группы будут увеличиваться, то мне придется сделать отдельную функцию для каждой группы. Если все это может быть достигнуто с помощью одной единственной функции, это было бы действительно здорово!

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