Как автоматически установить суперпользователя при создании типа пользователя

Я использую в своем приложении некоторые user_types, такие как is_admin и is_landlord. Я хочу, чтобы всякий раз, когда пользователь создается с помощью python manage.py createsuperuser. ему автоматически назначается is_admin.

модели.py

class UserManager(BaseUserManager):

    def _create_user(self, email, password, is_staff, is_superuser, **extra_fields):
        if not email:
            raise ValueError('Users must have an email address')
        now = datetime.datetime.now(pytz.utc)
        email = self.normalize_email(email)
        user = self.model(
            email=email,
            is_staff=is_staff,
            is_active=True,
            is_superuser=is_superuser,
            last_login=now,
            date_joined=now,
            **extra_fields
        )
        user.set_password(password)
        user.save(using=self._db)
        return user

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

    def create_superuser(self, email, password, **extra_fields):
        user = self._create_user(email, password, True, True, **extra_fields)
        user.save(using=self._db)
        return user

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(max_length=254, unique=True)
    is_staff = models.BooleanField(default=False)
    is_superuser = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    last_login = models.DateTimeField(null=True, blank=True)
    date_joined = models.DateTimeField(auto_now_add=True)

    # CUSTOM USER FIELDS
    name = models.CharField(max_length=30, blank=True, null=True)
    telephone = models.IntegerField(blank=True, null=True)

    USERNAME_FIELD = 'email'
    EMAIL_FIELD = 'email'
    REQUIRED_FIELDS = []

    objects = UserManager()

    def get_absolute_url(self):
        return "/users/%i/" % (self.pk)
        
    def get_email(self):
        return self.email

class user_type(models.Model):
    is_admin = models.BooleanField(default=False)
    is_landlord = models.BooleanField(default=False)
    user = models.OneToOneField(User, on_delete=models.CASCADE)

    def __str__(self):
        if self.is_landlord == True:
            return User.get_email(self.user) + " - is_landlord"
        else:
            return User.get_email(self.user) + " - is_admin"

Примечание. Я хочу, чтобы созданный суперпользователь автоматически имел статус is_admin.

Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Коллекции в Laravel более простым способом
Коллекции в Laravel более простым способом
Привет, читатели, сегодня мы узнаем о коллекциях. В Laravel коллекции - это способ манипулировать массивами и играть с массивами данных. Благодаря...
JavaScript Вопросы с множественным выбором и ответы
JavaScript Вопросы с множественным выбором и ответы
Если вы ищете платформу, которая предоставляет вам бесплатный тест JavaScript MCQ (Multiple Choice Questions With Answers) для оценки ваших знаний,...
Массив зависимостей в React
Массив зависимостей в React
Все о массиве Dependency и его связи с useEffect.
Toor - Ангулярный шаблон для бронирования путешествий
Toor - Ангулярный шаблон для бронирования путешествий
Toor - Travel Booking Angular Template один из лучших Travel & Tour booking template in the world. 30+ валидированных HTML5 страниц, которые помогут...
0
0
115
2
Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Поэтому я попробовал метод, который работает не так эффективно, но управляемо.

То, что я делаю, это простой трюк.

просмотры.py

def Dashboard(request, pk):
    
        user = User.objects.get(id=request.user.pk)
        landlord = user.is_staff & user.is_active
        if not landlord:
            reservations = Tables.objects.all()
            available_ = Tables.objects.filter(status = "available")
            unavailable_ = Tables.objects.filter(status = "unavailable")
            bar_owners_ = user_type.objects.filter(is_landlord=True)
            context = {"reservations":reservations, "available_":available_, "unavailable_":unavailable_, "bar_owners_":bar_owners_}
            return render(request, "dashboard/super/admin/dashboard.html", context)
        else:
            """Admin dashboard view."""
            user = User.objects.get(pk=pk)
            reservations = Tables.objects.filter(bar__user_id=user.id)
            available =Tables.objects.filter(bar__user_id=user.id, status = "available")
            unavailable =Tables.objects.filter(bar__user_id=user.id, status = "unavailable")

            context = {"reservations":reservations, "available":available, "unavailable":unavailable}

            return render(request, "dashboard/super/landlord/dashboard.html", context)

Поэтому я сообщаю Django, что если пользователь активен или для is_active установлено значение true, только тогда у него должен быть доступ к панели инструментов, иначе у него не будет к ней доступа.

Ответ принят как подходящий

Вы можете автоматически построить модель user_type в функции create_superuser:

class UserManager(BaseUserManager):
    
    # …
    
    def create_superuser(self, email, password, **extra_fields):
        user = self._create_user(email, password, True, True, **extra_fields)
        user.save(using=self._db)
        user_type.objects.update_or_create(
            user=user,
            defaults = {
                'is_admin': True
            }
        )
        return user

Однако мне непонятно, почему вы используете класс user_type, а не просто добавляете дополнительные поля в свою модель User. Это сделает его более эффективным, поскольку они будут извлекаться в том же запросе, когда вы извлекаете пользователя, вошедшего в систему. Также удобнее просто получить доступ к атрибуту .is_landlord из пользовательской модели.


Примечание. Модели в Django записываются в PerlCase, а не в змеином регистре. поэтому вы можете переименовать модель с user_type на UserType.

Спасибо за помощь. Я заметил, что вы использовали user_type..objects. Это правильно или должно быть user_type.objects с одной точкой

coderboy 18.12.2020 21:05

@coderboy: спасибо, это действительно опечатка. Зафиксированный.

Willem Van Onsem 18.12.2020 21:05

Другие вопросы по теме