Django Django用户组权限与装饰器

在本文中，我们将介绍Django中的用户组权限和装饰器的使用方法及其区别。Django是一个功能强大的Python Web框架，提供了多种权限验证和用户管理的机制，其中用户组和装饰器是常用的两种方式。

用户组

用户组是一种将用户分类的机制，可以将具有相似权限和功能的用户进行归类。在Django中，我们可以通过创建用户组并将用户加入到相应的组中，实现对不同用户组的权限控制。

创建用户组

首先，我们需要在Django的管理后台中创建用户组。在settings.py文件中，我们可以找到一下代码：

AUTH_USER_MODEL = 'auth.User'

我们需要在admin.py文件中进行如下设置：

from django.contrib.auth.models import Group

admin.site.unregister(Group)

然后，在models.py文件中定义你的用户组，例如：

from django.contrib.auth.models import Group

class CustomGroup(Group):
    description = models.TextField(max_length=100)

    def __str__(self):
        return self.name

最后，在你的应用中进行数据库迁移：

python manage.py makemigrations
python manage.py migrate

添加用户到用户组

创建用户组之后，我们可以将用户加入到相应的组中。可以通过在用户的save()方法中添加相应的逻辑来实现。

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

class CustomUserManager(BaseUserManager):
    def create_user(self, email, password=None, username=None):
        if not email:
            raise ValueError('Users must have an email address')

        user = self.model(
            email=self.normalize_email(email),
            username=username
        )

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

    def create_superuser(self, email, password):
        user = self.create_user(
            email=self.normalize_email(email),
            password=password,
        )
        user.is_admin = True
        user.is_staff = True
        user.is_superuser = True
        user.save(using=self._db)
        return user

class CustomUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
    )
    username = models.CharField(max_length=30, default="")
    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)
    is_staff = models.BooleanField(default=False)
    avatar = models.ImageField(upload_to='avatars/', blank=True, null=True)

    objects = CustomUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    def __str__(self):
        return self.email

    def get_short_name(self):
        return self.email

    class Meta:
        verbose_name_plural = 'users'

class CustomGroup(Group):
    description = models.TextField(max_length=100)

    def __str__(self):
        return self.name

以上代码中，我们通过继承Django内置的AbstractBaseUser和PermissionsMixin类来自定义用户模型，并在其中添加了CustomGroup模型。

一旦你的用户组和用户模型创建好了，你就可以通过如下代码将用户添加到特定的用户组中：

from django.contrib.auth.models import Group

user.groups.set([Group.objects.get(name='group_name')])

权限

与用户组类似，权限是另一种用于用户管理的机制。在Django中，权限是与用户组相关联的。我们可以将权限与用户组关联，并根据需要进行分配。

定义权限

在Django中，权限已经在django.contrib.auth模块中预定义好了。例如，定义一个可以查看用户信息的权限：

from django.contrib.auth.models import Permission

permission = Permission.objects.create(
    codename='view_user',
    name='Can view user',
    content_type=ContentType.objects.get_for_model(User)
)

将权限分配给用户组

通过以下代码，我们可以将权限分配给指定的用户组：

from django.contrib.auth.models import Permission, Group

group.permissions.add(Permission.objects.get(codename='permission_codename'))

判断用户是否拥有权限

我们可以通过以下代码判断用户是否拥有某个权限：

user.has_perm('app_label.permission_codename')

装饰器 vs 权限验证

装饰器是一种Python语言中常用的装饰函数的机制，用于在函数调用前后增加额外的逻辑。在Django中，装饰器可以用于对视图函数进行权限验证。

装饰器

Django提供了@login_required装饰器来验证用户是否已登录，以及@permission_required装饰器来验证用户是否拥有某个特定的权限。例如：

from django.contrib.auth.decorators import login_required, permission_required

@login_required
def my_view(request):
    # ...
    pass

@permission_required('polls.can_vote')
def my_view(request):
    # ...
    pass

权限验证

除了使用装饰器进行权限验证，我们还可以在视图函数中手动验证用户的权限。例如：

from django.contrib.auth.decorators import permission_required

@permission_required('polls.can_vote')
def my_view(request):
    if not request.user.has_perm('polls.can_vote'):
        # 没有权限的处理逻辑
    else:
        # 有权限的处理逻辑