?DRF?-----三大认证组件--认证组件

Posted 2020-11-16 lddragon

认证组件

铺垫:

源码分析

　　入口: restframework 框架内的 views 下的 APIView 的 dispatch方法

　　

 　　

 　　组件的最下面 有三个方法   分别是 认证组件  权限组件 和 频率组件

　　

perform_authentication (认证组件)　　

　　校验用户 - 游客 合法用户 非法用户

　　游客: 代表校验通过 进入下一步校验 (权限校验)

　　合法用户: 代表校验通过 将用户 存储在 request.user 中 再进行下一步校验

　　非法用户: 代表校验失败 抛出异常 返回403 权限异常

check_permissions(权限组件)

　　权限组件: 校验用户权限 -必须登录 所有用户 登录读写 游客只读 自定义用户角色

　　认证通过: 可以进行下一步的校验 

　　认证失败: 抛出异常 返回403权限异常结果

check_throttles(频率组件)

　　频率组件: 限制视图接口被访问的频率次数 - 限制的条件(IP、id、唯一键)、频率周期时间(s、m、h)、频率的次数（3/s）

　　没有达到限次:正常访问接口

　　达到限次: 限制时间内不能访问, 限制时间到达后,可以重新访问

RBAC(基于用户权限访问控制的认证)

　　(自己了解：基于auth的认证规则)
　　Django框架采用的是RBAC认证规则，RBAC认证规则通常会分为 三表规则、五表规则，Django采用的是六表规则
　　三表：用户表、角色表、权限表
　　五表：用户表、角色表、权限表、用户角色关系表、角色权限关系表
　　六表：用户表、角色表、权限表、用户角色关系表、角色权限关系表、用户权限关系表

　　

　　

正向反向查: 
　　用户表：角色groups，权限user_permissions
　　角色表：用户user_set，权限permissions
　　权限表：用户user_set，角色group_set

　　

 

 　　

(******)

from django.test import TestCase

# Create your tests here.
import os
import sys

if __name__ == "__main__":
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "三大认证组价.settings")
    import django
    django.setup()
    from django.contrib.auth.models import AbstractUser

    from app01 import models
    user_obj = models.User.objects.first() # type: models.User  # type 可以设置提示
    print(user_obj.username)
    # 用户表查 权限  有多个 first() 拿第一个 .name
    print(user_obj.user_permissions.first().name)  # Can add 用户表
    # 用户表 查 角色 可以有多个 first() 拿第一个
    print(user_obj.groups.first().name)  #管理员

    # 权限表 查询 导入 权限表
    from django.contrib.auth.models import Permission
    permission_obj = Permission.objects.filter(pk=16).first()
    # 权限查 自己
    print(permission_obj.name) # Can change 用户表
    # 权限 查 用户
    print(permission_obj.user_set.first().username)
    # 权限 查 角色
    p_16 = Permission.objects.filter(pk=17).first()
    print(p_16.group_set.first().name)

    # 角色表查 自己
    from django.contrib.auth.models import Group # type: Group
    group_obj = Group.objects.first()
    print(group_obj.name)
    # 查 用户
    print(group_obj.user_set.first().username)
    # 查 权限
    print(group_obj.permissions.first().name) # Can change 用户表

三表的互相查询

重点：如果自定义User表后，再另一个项目中采用原生User表，完成数据库迁移时，可能失败
# 1）卸载Django重新装
# 2）将django.contrib下面的admin、auth下的数据库迁移记录文件清空

 　　

自定义user表

　　

 　　settings文件中指定 使用自定义user

 　　

 部分源码加注释

Request类的 方法属性 user 的get方法 => self._authenticate() 完成认证
    
    认证的细则：
    # 做认证
    def _authenticate(self):
        # 遍历拿到一个个认证器，进行认证
        # self.authenticators配置的一堆认证类产生的认证类对象组成的 list
        for authenticator in self.authenticators:
            try:
                # 认证器(对象)调用认证方法authenticate(认证类对象self, request请求对象)
                # 返回值：登陆的用户与认证的信息组成的 tuple
                # 该方法被try包裹，代表该方法会抛异常，抛异常就代表认证失败
                user_auth_tuple = authenticator.authenticate(self)
            except exceptions.APIException:
                self._not_authenticated()
                raise

            # 返回值的处理
            if user_auth_tuple is not None:
                self._authenticator = authenticator
                # 如何有返回值，就将 登陆用户 与 登陆认证 分别保存到 request.user、request.auth
                self.user, self.auth = user_auth_tuple
                return
        # 如果返回值user_auth_tuple为空，代表认证通过，但是没有 登陆用户 与 登陆认证信息，代表游客
        self._not_authenticated()

 

其中 request 在 dispatch二次封装的时候 有一个 get _authenticators() 内部设置了 配置文件 在 APIView顶部可以看到

 

 

自定义认证

1) 创建继承BaseAuthentication的认证类
2) 实现authenticate方法
3) 实现体根据认证规则 确定游客、非法用户、合法用户
4) 进行全局或局部配置

认证规则
i.没有认证信息返回None(游客)
ii.有认证信息认证失败抛异常(非法用户)
iii.有认证信息认证成功返回用户与认证信息元组(合法用户)

# 自定义认证类
# 1）继承BaseAuthentication类
# 2）重新authenticate(self, request)方法，自定义认证规则
# 3）认证规则基于的条件：
#       没有认证信息返回None(游客)
#       有认证信息认证失败抛异常(非法用户)
#       有认证信息认证成功返回用户与认证信息元组(合法用户)
# 4）完全视图类的全局(settings文件中)或局部(确切的视图类)配置
from rest_framework.authentication import BaseAuthentication
from rest_framework.exceptions import AuthenticationFailed
from . import models
class MyAuthentication(BaseAuthentication):
    """
    同前台请求头拿认证信息auth（获取认证的字段要与前台约定）
    没有auth是游客，返回None
    有auth进行校验
        失败是非法用户，抛出异常
        成功是合法用户，返回 (用户, 认证信息)
    """
    def authenticate(self, request):
        # 前台在请求头携带认证信息，
        #       且默认规范用 Authorization 字段携带认证信息，
        #       后台固定在请求对象的META字段中 HTTP_AUTHORIZATION 获取
        auth = request.META.get(‘HTTP_AUTHORIZATION‘, None)

        # 处理游客
        if auth is None:
            return None

        # 设置一下认证字段小规则(两段式)："auth 认证字符串"
        auth_list = auth.split()

        # 校验合法还是非法用户
        if not (len(auth_list) == 2 and auth_list[0].lower() == ‘auth‘):
            raise AuthenticationFailed(‘认证信息有误，非法用户‘)

        # 合法的用户还需要从auth_list[1]中解析出来
        # 注：假设一种情况，信息为abc.123.xyz，就可以解析出admin用户；实际开发，该逻辑一定是校验用户的正常逻辑
        if auth_list[1] != ‘abc.123.xyz‘:  # 校验失败
            raise AuthenticationFailed(‘用户校验失败，非法用户‘)

        user = models.User.objects.filter(username=‘admin‘).first()

        if not user:
            raise AuthenticationFailed(‘用户数据有误，非法用户‘)
        return (user, None)

 

 权限校验

4) 权限组件
    self.check_permissions(request)
    认证细则：
    def check_permissions(self, request):
        # 遍历权限对象列表得到一个个权限对象(权限器)，进行权限认证
        for permission in self.get_permissions():
            # 权限类一定有一个has_permission权限方法，用来做权限认证的
            # 参数：权限对象self、请求对象request、视图类对象
            # 返回值：有权限返回True，无权限返回False
            if not permission.has_permission(request, self):
                self.permission_denied(
                    request, message=getattr(permission, ‘message‘, None)
                )

 

 系统权限类

"""
1）AllowAny：
    认证规则全部返还True：return True
        游客与登陆用户都有所有权限

2) IsAuthenticated：
    认证规则必须有登陆的合法用户：return bool(request.user and request.user.is_authenticated)
        游客没有任何权限，登陆用户才有权限
    
3) IsAdminUser：
    认证规则必须是后台管理用户：return bool(request.user and request.user.is_staff)
        游客没有任何权限，登陆用户才有权限

4) IsAuthenticatedOrReadOnly
    认证规则必须是只读请求或是合法用户：
        return bool(
            request.method in SAFE_METHODS or
            request.user and
            request.user.is_authenticated
        )
        游客只读，合法用户无限制
"""

# api/views.py
from rest_framework.permissions import IsAuthenticated
class TestAuthenticatedAPIView(APIView):
    permission_classes = [IsAuthenticated]
    def get(self, request, *args, **kwargs):
        return APIResponse(0, ‘test 登录才能访问的接口 ok‘)
    
    
# 因为默认全局配置的权限类是AllowAny
# settings.py
REST_FRAMEWORK = {
    # 权限类配置
    ‘DEFAULT_PERMISSION_CLASSES‘: [
        ‘rest_framework.permissions.AllowAny‘,
    ],
}
    

 

 自定义权限类

1) 创建继承BasePermission的权限类
2) 实现has_permission方法
3) 实现体根据权限规则 确定有无权限
4) 进行全局或局部配置

认证规则
i.满足设置的用户条件，代表有权限，返回True
ii.不满足设置的用户条件，代表有权限，返回False

# utils/permissions.py
from rest_framework.permissions import BasePermission
from django.contrib.auth.models import Group
class MyPermission(BasePermission):
    def has_permission(self, request, view):
        # 只读接口判断
        r1 = request.method in (‘GET‘, ‘HEAD‘, ‘OPTIONS‘)
        # group为有权限的分组
        group = Group.objects.filter(name=‘管理员‘).first()
        # groups为当前用户所属的所有分组
        groups = request.user.groups.all()
        r2 = group and groups
        r3 = group in groups
        # 读接口大家都有权限，写接口必须为指定分组下的登陆用户
        return r1 or (r2 and r3)
    
    
# 游客只读，登录用户只读，只有登录用户属于 管理员 分组，才可以增删改
from utils.permissions import MyPermission
class TestAdminOrReadOnlyAPIView(APIView):
    permission_classes = [MyPermission]
    # 所有用户都可以访问
    def get(self, request, *args, **kwargs):
        return APIResponse(0, ‘自定义读 OK‘)
    # 必须是 自定义“管理员”分组 下的用户
    def post(self, request, *args, **kwargs):
        return APIResponse(0, ‘自定义写 OK‘)