drf-认证权限频率:自定义类、内置类、全局使用、局部使用、源码分析

在这里插入图片描述

一、 认证Authentication

1.1 自定义认证类

1.1.1 表模型

# 表模型中两个表
class User(models.Model):
    username=models.CharField(max_length=32)
    password=models.CharField(max_length=32)

class UserToken(models.Model):
    user=models.OneToOneField(to='User',on_delete=models.CASCADE)
    token=models.CharField(max_length=64)

1.1.2 登录视图

# 登录视图类
class UserView(APIView):
    def post(self, request, *args, **kwargs):
        res = {
   'code': 100, 'msg': None}
        username = request.data.get('username')
        password = request.data.get('password')
        user = models.User.objects.filter(username=username, password=password).first()

        if user:  # 登录成功再生成token
            token = str(uuid.uuid4())  # 生成一个不重复的随机字符串
            # 原来有,就更新,原来没有就新增
            # models.UserToken.objects.create(user=user,token=token)
            # 根据user=user去查询,如果有,使用defaults更新,如果没有,新增一条
            models.UserToken.objects.update_or_create(defaults={
   'token': token}, user=user)
            res['msg'] = '登录成功'
            res['token'] = token
        else:
            res['code'] = '101'
            res['msg'] = '用户名或密码错误'
        return Response(res)

1.1.3 路由

# 路由配置
urlpatterns = [
    path('login/', views.UserView.as_view()),]

1.1.4 认证类的编写

# 新建一个py文件auth.py
from rest_framework.authentication import BaseAuthentication
from rest_framework.exceptions import AuthenticationFailed
from . import models
class LoginAuth(BaseAuthentication):
    def authenticate(self, request):
        # 写认证规则
        # 取出用户携带的token
        token=request.query_params.get('token')
        # 去数据库查询,token是否存在,如果存在,说明这个人登录了,如果不存在,表示没有登录
        user_token=models.UserToken.objects.filter(token=token).first()
        if user_token:
            # 登录了,认证通过,继续往后走
            '''
            如果返回user,token,后面视图类中通过request对象,可以取到当前登录用户
            '''
            return
        else:
            # 没有登录,不能继续往后走,抛认证失败异常
            raise AuthenticationFailed('您没有登录,认证失败')

            
            
            
# 对某个接口生效
class BookView(ViewSetMixin,ListAPIView):
    # 以后,必须登录以后才能访问这个视图类
    authentication_classes = [LoginAuth,]

1.1.5 注意

如需自定义认证,需继承一个rest_framework.authentication.BaseAuthentication,重写authenticate,如果认证通过,返回空或两个值,第一个值给request.user

def authenticate
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值