drf三大认证

Posted 2020-11-16 zrh-960906

源码分析

　　dispath方法内 self.initial(request, *args, **kwargs) 进入三大认证

        # 认证组件:校验用户（游客，合法用户，非法用户）
        # 游客：代表校验通过直接进入下一步校验，
        # 合法用户：代表校验通过，将用户存储在request.user中，在进入下一步校验
        # 非法用户：代表校验失败，抛出异常，返回403权限异常结果
        self.perform_authentication(request)
        # 权限组件：校验用户权限，必须登录，所有用户登录读写，游客只读，自定义用户角色
        # 认证通过：可以进入下一步校验
        # 认证失败：抛出异常，返回403权限异常结果
        self.check_permissions(request)
        # 频率组件：限制视图接口被访问的频率次数  限制的条件
        # 没有达到限次：正常访问接口
        # 达到限次：限制时间内不能访问，限制时间到达后，可以重新访问
        self.check_throttles(request)    

认证组件

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()

权限组件

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)
                )

权限六表分析

models.py

from django.db import models

# Create your models here.

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

class User(AbstractUser):
    mobile = models.CharField(max_length=11,unique=True)

    class Meta:
        db_table = ‘api_user‘
        verbose_name = ‘用户表‘
        verbose_name_plural = verbose_name

    def __str__(self):
        return self.username

t_model.py

# django脚本话启动
import os, django
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "day74.settings")
django.setup()

from api import models

user = models.User.objects.first()
# print(user.username)
# print(user.groups.first().name)
# print(user.user_permissions.first().name)

from django.contrib.auth.models import Group
group = Group.objects.first()
# print(group.name)
# print(group.user_set.first().username)
# print(group.permissions.first().name)


from django.contrib.auth.models import Permission
p_16 = Permission.objects.filter(pk=16).first()
print(p_16.user_set.first().username)
p_17 = Permission.objects.filter(pk=17).first()
print(p_17.group_set.first().name)

自定义认证类

settings.py

# drf配置
REST_FRAMEWORK = {
    # 认证类配置
    ‘DEFAULT_AUTHENTICATION_CLASSES‘: [
        # ‘rest_framework.authentication.SessionAuthentication‘,
        # ‘rest_framework.authentication.BasicAuthentication‘,
        ‘api.authentications.MyAuthentication‘,
        ],
}

authentications.py

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_list = auth.split()
        # 校验合法还是非法
        if not (len(auth_list) == 2 and auth_list[0].lower() == ‘auth‘):
            raise AuthenticationFailed(‘认证信息有误，非法用户‘)

        # 合法用户需要从auth_list[1]中解析出来
        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)

系统权限类

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
        )
        游客只读，合法用户无限制

　　urls.py

from django.conf.urls import url
from . import views

urlpatterns = [
    url(r‘^test‘, views.TestAPIView.as_view()),

　　 url(r‘^test1‘, views.TestAuthenticatedAPIView.as_view()),

]

settings.py

REST_FRAMEWORK = {
    # 权限类配置
    ‘DEFAULT_PERMISSION_CLASSES‘: [
        ‘rest_framework.permissions.AllowAny‘,
    ],
}

api/views.py

from rest_framework.views import APIView
from rest_framework.generics import GenericAPIView
from rest_framework.viewsets import GenericViewSet,ViewSet
from utils.response import APIResponse
from rest_framework.permissions import IsAuthenticated

class TestAPIView(APIView):
    def get(self, request, *args, **kwargs):
        return APIResponse(0, ‘test get ok‘)


class TestAuthenticatedAPIView(APIView):
    permission_classes = [IsAuthenticated]
    def get(self, request, *args, **kwargs):
        return APIResponse(0,‘test 登录才能访问接口 ok‘)

自定义权限类

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

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

urls.py

from django.conf.urls import url
from . import views

urlpatterns = [
    url(r‘^test2‘, views.TestAdminOrReadOnlyAPIView.as_view()),
]

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 = request.user.groups.all()
        r2 = group and groups
        r3 = group in groups
　　　　# 读接口都权限，写接口必须指定分组下的登录用户
        return r1 or (r2 and r3)

views.py

# 游客只读，登录用户只读，只有登录用户属于管理员分组，才可以增删改
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‘)