/* 看板娘 */ /* 粒子吸附*/

drf之认证,频率和权限

1 认证

1
2
3
4
5
6
# 登陆接口---》登陆成功,只要给前端返回json格式字符串---》字符串中带一个随机字符串
# 认证---》登陆认证---》判断用户是否登录了---》前后端分离---》用不到cookie---》携带随机字符串过来---》通过判断随机字符串判断这个人是否登录了
 
# 登陆接口编写步骤
    -创建表:User表,一对一UserToken表
  -前端传入用户名,密码---》User表中查,如果能查到---》让他登陆成功---》在UserToken中存一条记录---》返回给前端json格式字符串---》字符串中带一个随机字符串

  

1.1 登陆接口

复制代码
# class UserView(ViewSetMixin,APIView):
class UserView(ViewSet):
    @action(methods=['POST'], detail=False)
    def login(self, request):
        username = request.data.get('username')
        password = request.data.get('password')
        user = User.objects.filter(username=username, password=password).first()
        if user:
            # 登陆成功--》生成一个随机字符串-->存到token表中(如果之前有记录,更新,如果没有新增)
            # uuid生成不重复的串---》理论上不重复
            token = str(uuid.uuid4())  # 伪随机数生成,没有参数,重复概率,比其他的高一些
            UserToken.objects.update_or_create(user=user, defaults={'token': token})  # 如果存在就更新,如果不存在就新增
            return Response({'code': 100, 'msg': '登陆成功', 'token': token})
        else:
            return Response({'code': 101, 'msg': '用户名或密码错误'})
复制代码

urls.py

复制代码
#自动生成路由


from app01 import views
from rest_framework.routers import SimpleRouter
router=SimpleRouter()
router.register('user',views.UserView,'user')

urlpatterns = [
    path('admin/', admin.site.urls),
    path('', include(router.urls)),
]
复制代码

models.py

复制代码
class User(models.Model):
    username = models.CharField(max_length=32)
    password = models.CharField(max_length=32)
    user_type = models.IntegerField(choices=((1, '超级管理员'), (2, '普通管理员'), (3, '普通用户')))
    # gender=models.IntegerField(choices=((1,"男"),(2,"女"),(0,"未知"))) # 1:男  2:女 0:未知
    # user对象.get_字段名_display()  快速取出gender数字对应的中文


class UserToken(models.Model):
    user = models.OneToOneField(to=User,on_delete=models.CASCADE)
    token = models.CharField(max_length=32)
复制代码

1.2 认证类

复制代码
# 认证类:用来校验用户是否登录,如果登录了,继续往下走,如果没有登录,直接返回
# 编写步骤:
    -第一步:写一个类,继承BaseAuthentication,重写authenticate,在方法中做校验,校验是否登录,返回两个值,没有登录抛异常
  -第二步:全局配置,局部配置
      -全局配置:配置文件中
      REST_FRAMEWORK={
      "DEFAULT_AUTHENTICATION_CLASSES":["app01.auth.LoginAuth",]
          }
    -局部配置:在视图类中
      class UserView(ViewSet):
        authentication_classes = [LoginAuth] 
        
    -局部禁用:
       class UserView(ViewSet):
            authentication_classes = [] 

# 认证类中返回的两个变量,干啥用了
    -返回的第一个,给了request.user,就是当前登录用户
  -返回的第二个,给了request.auth,就是token串
复制代码

2 频率

复制代码
# 认证,权限都通过以后,现在某个接口的访问频率---》一般根据ip或者用户限制

# 使用步骤
    -第一步:写一个类,继承SimpleRateThrottle,重写类属性:scope,和get_cache_key方法
      get_cache_key返回什么,就以什么做现在,scope配置文件中要用
    
  -第二步:在配置文件中配置
        'DEFAULT_THROTTLE_RATES': {
        'minute_3': '3/m'  # minute_3是scope的字符串,一分钟访问3次
        'minute_5''5/m'
    }
  -局部使用--》视图类中
  class BookView(GenericViewSet, ListModelMixin, RetrieveModelMixin):
    throttle_classes = [IPThrottle]
  -全局使用--配置文件中
     'DEFAULT_THROTTLE_CLASSES': (  # 全局配置频率类
        'app01.auth.IPThrottle'
    ),
  
复制代码

3 权限

复制代码
# 登录成功----》所有必须登录能访问---》每个视图类上加认证类
# 用户是普通用户---》普通用户可以访问所有和单条
# 普通管理员和超级用户可以操作所有,除了访问单条和所有的那个视图类,加上认证类
# books:查看一条,和所有
# booksdetail路由下有:删除,新增,修改---》权限类加在这里

# book 5个接口,必须登录才能访问
# 5个接口分成了俩视图写:
        -BookView:获取所有,获取单条
      -BookDetailView:删除,修改,新增
        -这俩视图都需要登录:authentication_classes = [LoginAuth, ]
    -BookView只要登陆就可以操作
    -BookDetailView必须有权限才能,加了一个权限,permission_classes = [UserPermission, ]
    
# 跟写认证类步骤差不多
第一步:写一个类,继承BasePermission,重写has_permission,判断如果有权限,返回True,如果没有权限,返回False
第二步:局部使用和全局使用
    -局部使用
    class BookDetailView(GenericViewSet, CreateModelMixin, DestroyModelMixin, UpdateModelMixin):
    permission_classes = [UserPermission, ]
    
  -全使用
      REST_FRAMEWORK={
            "DEFAULT_PERMISSION_CLASSES":["app01.auth.UserPermission",]
        }
复制代码

 

复制代码
autho.py

from rest_framework.permissions import BasePermission

class UserPermission(BasePermission):
    def has_permission(self, request, view):
        self.message='您是:%s,没有权限'%request.user.get_user_type_display()  # 没有权限的提示信息
        # 如果有权限,返回True,没有权限返回False
        # 权限类,在认证类之后,request.user有了当前登录用户
        user_type = request.user.user_type
        if user_type < 3:  # 只要不是1,2,就没有权限
            return True
        else:
            return False
复制代码

 

posted @   红绿灯的黄呀  阅读(40)  评论(2编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
点击右上角即可分享
微信分享提示