详解python如何在django中为用户模型添加自定义权限

下面是详解如何在 Django 中为用户模型添加自定义权限的攻略。

1. 概述

在 Django 中，我们可以使用自带的权限系统控制用户对资源的访问，但是这些权限可能不足以满足我们的需求，我们需要自定义权限。本文将介绍如何在 Django 中为用户模型添加自定义权限。

2. 实现步骤

2.1. 定义权限

在 Django 中，权限在 django.contrib.auth.models 模块中的 Permission 模型中进行定义。为了添加自定义权限，我们需要在 Permission 模型的 codename 中添加自定义权限字符串。

我们可以将自定义权限分组，并定义权限的名称和描述。

from django.contrib.auth.models import Permission, Group
from django.db.models.signals import post_migrate
from django.dispatch import receiver
from django.utils.translation import ugettext_lazy as _

# 自定义权限分组名称
GROUP_NAME = 'example_group'

# 定义权限
PERMISSIONS = [
    ('example_permission_1', 'Example Permission 1'),
    ('example_permission_2', 'Example Permission 2')
]

# 创建自定义权限分组
@receiver(post_migrate)
def create_groups_and_permissions(app_config, **kwargs):
    example_group, created = Group.objects.get_or_create(name=GROUP_NAME)
    if created:
        for codename, name in PERMISSIONS:
            permission, created = Permission.objects.get_or_create(codename=codename, name=name)
            if created:
                example_group.permissions.add(permission)
                permission.save()
                print(f"Added permission {codename} - {name}")

在上面的代码中，我们通过 Django 的 post_migrate 信号来创建自定义权限分组和权限，并将权限添加到分组中。

2.2. 添加权限到用户模型

接下来，我们需要将自定义权限添加到 Django 自带的用户模型中。为此，我们需要继承 AbstractUser 模型，并添加一个多对多的权限关系字段。

from django.contrib.auth.models import AbstractUser

class CustomUser(AbstractUser):
    user_permissions = models.ManyToManyField(
        Permission,
        verbose_name=_('user permissions'),
        blank=True,
        help_text=_('Specific permissions for this user.'),
        related_name="customuser_set",
        related_query_name="user",
    )

同时，我们需要在 settings.py 中指定使用我们自定义的用户模型：

AUTH_USER_MODEL = 'yourapp.CustomUser'

2.3. 在视图中验证权限

最后，在视图中我们需要验证用户是否具有相应的权限。我们可以使用 user_has_perm 函数对用户、对象、和权限字符串进行验证。

from django.contrib.auth.decorators import user_passes_test

@user_passes_test(lambda u: u.has_perm('example_permission_1', obj=some_object))
def my_view(request):
    # your view code

上面的代码中，user_passes_test 装饰器将会检查用户是否具有 "example_permission_1" 权限，如果没有，会跳转到默认的未授权页面。

另外，在视图中我们需要传递一个对象作为权限的验证参数。在上面的代码中，我们可以通过 obj 参数传递这个对象。

3. 示例说明

3.1. 示例1

假设我们正在开发一个博客网站，我们需要为用户模型添加一个自定义权限，以便允许某些用户编辑和删除他们发布的博客文章。

我们可以使用上面的方法添加一个名为 "can_edit_own_post" 的自定义权限。然后，在博客视图中，我们可以使用 user_passes_test 装饰器验证用户是否具有该权限。

from django.contrib.auth.decorators import user_passes_test

@user_passes_test(lambda u: u.has_perm('can_edit_own_post', obj=post))
def edit_post(request, post_id):
    # your view code

这样，只有发布该博客文章的用户才能编辑和删除它。

3.2. 示例2

假设我们正在开发一个社交网络网站，我们需要为用户模型添加一个自定义权限，以便允许某些用户查看其他用户的好友列表。

我们可以使用上面的方法添加一个名为 "can_view_friends_list" 的自定义权限。然后，在用户的好友列表视图中，我们可以使用 user_passes_test 装饰器验证用户是否具有该权限。

from django.contrib.auth.decorators import user_passes_test

@user_passes_test(lambda u: u.has_perm('can_view_friends_list', obj=user))
def friends_list(request, user_id):
    # your view code

这样，只有具有 "can_view_friends_list" 权限的用户才能查看其他用户的好友列表。

4. 总结

在以上攻略中，我们介绍了如何在 Django 中为用户模型添加自定义权限。我们首先定义了自定义权限，并将其添加到了自定义用户模型中。然后，我们在视图中使用 user_passes_test 装饰器对用户的权限进行验证。同时，我们还举了两个例子，帮助读者更好地理解如何实现这个功能。