首页 > Django

Django Auth装饰器验证用户身份与权限

我们知道在视图函数中，我们使实现对用户身份及权限的验证。Django为了给开发者提供方便， 还提供了便捷的装饰器来完成这类的校验。比如，@login_required 我们使用它来验证用户是否已经登录，只有登录的用户才可以访问视图函数，并获得响应，否则将重定向到登录界面。当然还有校验权限的装饰器 @permission_required，在本节我们将对这些装饰器使用方法进行逐一介绍。

1. 校验用户登录状态@login_required

为了分析这个装饰器，我们还是首先看一下 Django 的源码，它的定义如下文件中：

from django.contrib.auth.decorators import login_required

1) login_required函数参数说明

它的源码如下所示： 
def login_required(function=None, redirect_field_name=REDIRECT_FIELD_NAME, login_url=None):

    actual_decorator = user_passes_test(
        lambda u: u.is_authenticated,
        login_url=login_url,
        redirect_field_name=redirect_field_name
    )
    if function:
        return actual_decorator(function)
    return actual_decorator
可以看出这个函数可以传递两个参数，我们这个函数做一下简单的分析：login_url 表示若为匿名用户访问时重定向的 URL，这里一般指定的都是登录页面的 URL 路径，默认的登录页需要在配置文件通过 LOGIN_URL 指定，然后通过使用以下方式进行调用 settings.py.LOGIN_URL；redirect_field_name 默认值为 next，作为 GET 的请求参数即参训字符串的形式，它的格式如下：

127.0.0.1:8000/login/?next=/index/add_book/1

这个参数用于登录后直接跳回到原先访问的视图。上述源码中可以看出该方法的实现核心是调用了 user_passes_test 方法。它需要传递三个参数，分析它的部分源码。如下所示： 
def user_passes_test(test_func, login_url=None, redirect_field_name=REDIRECT_FIELD_NAME):

    def decorator(view_func):
        @wraps(view_func)
        def _wrapped_view(request, *args, **kwargs):
            #测试函数，通过后执行对应的视图函数
            if test_func(request.user):
                return view_func(request, *args, **kwargs)
            path = request.build_absolute_uri()  #返回请求完成的URL
             #获取登录页指定的URL
            resolved_login_url = resolve_url(login_url or settings.LOGIN_URL)
            # If the login url is the same scheme and net location then just
            # use the path as the "next" url.
            login_scheme, login_netloc = urlparse(resolved_login_url)[:2]
            current_scheme, current_netloc = urlparse(path)[:2]
            #如果登录页的 URL与path的协议，域都相同则执行下面代码
            if ((not login_scheme or login_scheme == current_scheme) and
                    (not login_netloc or login_netloc == current_netloc)):
                #获取视图的全路径，返回 HttpResponseRedirect
                path = request.get_full_path()
            from django.contrib.auth.views import redirect_to_login
            return redirect_to_login(
                path, resolved_login_url, redirect_field_name)
        return _wrapped_view
    return decorator
从 user_passes_test 的实现可以看出，它首先会判断 request.user.is_authenticated 是否会返回 True，如果成立，则会执行视图函数。否则，将重定向到登录页面。

2) login_required应用方式

它的使用方式也非常的简单，只需要在视图函数加上方稍加改动即可，如下所示： 
from django.contrib.auth.decorators import login_required
@login_required
def search_title_views(request):
    pass
如果在用户未登录的情况下访问这个视图的话，那么它将会跳转到登录页，需要注意的是由于这里没有指定 login_url，因此在配置文件中的 LOGIN_URL 要设置正确。

2. 校验用户权限@permission_required

1) @permission_required的源码分析

理解了如何校验登录状态的装饰器 @login_required， 下面我们对 @permission_required 进行讲解，这样大家理解起来会更加简单。我们还是看一下它的实现源码，如下所示： 
def permission_required(perm, login_url=None, raise_exception=False):
    """
   用于检查用户是否启用了特定权限的视图的装饰器，必要时可重定向到登录页。
   如果给定了raise_exception参数，则会引发PermissionDenied异常。
    """
    def check_perms(user):
         #如果指定权限是字符串，则将其放在元组中
        if isinstance(perm, str):
            perms = (perm,)
        else:
            perms = perm
        #校验用户是否具有指定的权限
        if user.has_perms(perms):
            return True
        # In case the 403 handler should be called raise the exception
        if raise_exception:
            raise PermissionDenied
        #最终没有通过校验返回 False
        return False
    #check_perms 即为 user_passes_test中的测试函数
    return user_passes_test(check_perms, login_url=login_url)
这个函数接受三个参数，它们的介绍如下：
它的定义格式和 @login_required 是非常类似的。但是也有一点小小的区别，如下所示：
@permission_required("index.can_view_book")#也可校验多个权限，在方法内添加即可
def book_add_views(request):
    pass
我们可以这样理解，如果访问用户没有被授予 index.can_view_book 权限，就会跳转到登录页。这样不仅需要当前用户是已登庄状态，还需要用户拥有 can_view_book 的权限。

Django 的用户认证系统的到这里就介绍完了。任何一个服务型的 Web 站点都离不开用户与权限的概念，这一点我们也做了反复的强调。Django 对这两大功能提供了很好的支持。作为开发者在实际的业务开发中，应该尽量使用 Django 的用户认证系统，特别是 User 模型，它会给我们带来极大的便利。对于特定的需求，也可以考虑对其进行扩展，例如自定义权限、自定义认证后端等等。

相关文章