认证组件
铺垫:
源码分析
入口: 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‘)
原文地址:https://www.cnblogs.com/lddragon/p/11717230.html