rest_framework认证,频率,权限

在views中

from rest_framework.views import APIView
from rest_framework.viewsets import ViewSetMixin
import hashlib
import time
from app01 import models
from rest_framework.response import Response
from django.http import JsonResponse
class Login(ViewSetMixin,APIView):
    def get_token(self):
        h6=hashlib.md5()
        ctime=time.time()
        h6.update(bytes(str(ctime),encoding='utf-8'))
        token=h6.hexdigest()
        return token

    def ulogin(self,request):

        response={'status':100,'msg':'登录成功'}
        name=request.data.get('name')
        pwd=request.data.get('pwd')
        user=models.UserInfo.objects.all().filter(name=name,pwd=pwd).first()
        if user:
            ret=models.UserToken.objects.all().create(token=self.get_token(),user=user)#登录成功就去数据库中写token
            response['token']=self.get_token()

        else:
            response['status']=101
            response['msg']='用户名或密码错误'
        return Response(response)




class Author(ViewSetMixin,APIView):
#局部使用登录认证,频率,权限
    throttle_classes = [MyThrottle, ]
    authentication_classes = [LoginAuth, ]
    permission_classes = [UserPermission,]
    def get_authors(self,request):
        author=models.Author.objects.all()
        ser=Myser.Authorser(author,many=True)
        return Response(ser.data)

#频率错误信息显示
    def throttled(self, request, wait):
        class MyThrottled(exceptions.Throttled):
            default_detail = '×××'
            extra_detail_singular = '还剩 {wait} 秒.'
            extra_detail_plural = '还剩 {wait} 秒'

在MyAuth中----------------------------------------------------------------------------------------------------


from app01 import models
from rest_framework import exceptions


class LoginAuth():
    def authenticate(self, request):
        token = request.query_params.get('token')
        ret = models.UserToken.objects.all().filter(token=token).first()
        if ret:
            return ret.user, ret
        else:
            raise exceptions.APIException('认证失败')


class UserPermission():
    # message是出错显示的中文
    message = '您没有权限查看'

    def has_permission(self, request, view):
        user_type = request.user.user_type
        # 取出用户类型对应的文字
        # 固定用法:get_字段名字_display()
        user_type_name = request.user.get_user_type_display()
        print(user_type_name)
        if user_type == 2:
            return True
        else:
            return False

from rest_framework.throttling import SimpleRateThrottle
class MyThrottle(SimpleRateThrottle):
    scope = 'pinglv'

    def get_cache_key(self, request, view):
        return self.get_ident(request)




在Myser中---------------------------------------------------------------------------------

from rest_framework import serializers
from app01 import models
class Bookser(serializers.ModelSerializer):
    class Meta:
        model=models.Book
        fields='__all__'



class Authorser(serializers.ModelSerializer):
    class Meta:
        model=models.Author
        fields='__all__'
        
        
        
        
在setting中-----------------------------------------------------------------------------------------



REST_FRAMEWORK = {
    # 'DEFAULT_AUTHENTICATION_CLASSES':['app01.MyAuth.LoginAuth',],#认证全局使用
    # 'DEFAULT_PERMISSION_CLASSES':['app01.MyAuth.UserPermission',],#权限全局使用
    # 'DEFAULT_THROTTLE_CLASSES': ['app01.MyAuth.MyThrottle', ],#频率全局使用
    # 每分钟访问10次
    'DEFAULT_THROTTLE_RATES': {
        'pinglv': '10/m'
    },
}

当前文章:rest_framework认证,频率,权限
转载来于:http://csdahua.cn/article/pijcgs.html
扫二维码与项目经理沟通

我们在微信上24小时期待你的声音

解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流