如何在創建用戶類型時自動設置超級用戶

Question

我在我的應用程序中使用了一些 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 狀態

Answer 1

您可以在create_superuser function 中自動構造一個user_type model：

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 class，而不僅僅是向您的User model 添加額外的字段。 這將提高效率，因為當您獲取登錄用戶時，這些將在同一個查詢中獲取。 僅從用戶 model 訪問.is_landlord屬性也更方便。

---

注意： Django 中的模型是用PerlCase而不是snake_case編寫的，因此您可能需要將 model 從user_type重命名為UserType 。

Answer 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，那么他們應該有權訪問儀表板，否則他們將無權訪問它。