django rest framework之部分组件

认证

  在这里,笔者将登录与认证归一起,具体代码如下

  视图函数

def md5(user):#用账户当参数
    '''生成随机码'''
    import time
    import hashlib

    c_time = str(time.time())
    m = hashlib.md5(bytes(user,encoding='utf-8'))
    m.update(bytes(c_time,encoding='utf-8'))
    return m.hexdigest()


class AuthView(APIView):
    authentication_classes = [ ]

    def post(self,request,*args,**kwargs):#选择method请求方式
        ret = {'code':1000,'msg':None}
        try:
            username = request._request.POST.get('username')
            password = request._request.POST.get('password')
            obj = UserInfo.objects.filter(username=username,password=password).first()
            if not obj:
                ret['code'] = 1001
                ret['msg'] = '账号或者密码错误!'
            token = md5(username)
            print(token)
            models.UserToken.objects.update_or_create(user=obj,defaults={'token':token})#新建或者修改数据
            ret['msg'] = '欢迎登陆!'
            ret['token'] = token
        except Exception as e:
            pass
        return JsonResponse(ret)


class OrderView(APIView):

    # authentication_classes = [Authtication,]#局部使用
    def get(self,request,*args,**kwargs):
        print(request.user)
        print(request.auth)
        ret = {'code':1000,"msg":'ok','data':None}
        try:
            ret['data'] = ORDER_DICT
        except Exception as e:
            pass
        return JsonResponse(ret)

  为了规范,将认证放在单独的文件下:

  

  代码如下:

class Authtication(BaseAuthentication):#继承父类header就可以省略,因为父类里面有
    def authenticate(self,request):
        token = request._request.GET.get('token')
        token_obj = UserToken.objects.filter(token=token).first()
        if not token_obj:
            raise exceptions.AuthenticationFailed('用户认证失败!')
        #drf会将两个字段的值赋值给request,以供后续使用
        return (token_obj.user,token_obj)

    # def authenticate_header(self,request):
    #     pass

  由于drf的机制,主要是列表生成式:在全局进行认证我们可以在配置文件中这么写:

REST_FRAMEWORK = {
    "DEFAULT_AUTHENTICATION_CLASSES":["api.utils.auth.Authtication",]    #这里必须是列表,否则会报错'type' object is not iterable
}

  接下来是源码的个人解读:

用户认证源码流程

1.请求进来之后,先在自己的类中寻找dispatch方法,由于我们没有自己定义,所以找到父类ApiView中的dispatch开始执行方法

2.进入dispatch方法之后,对request进行封装,request不再是原生的request(如果要使用可以用request._request调用)
    封装的时候执行了一个authenticators=self.get_authenticators()方法,通过列表生成式实例化对象

3.封装完request之后,开始执行initial方法,其中我们的认证是self.perform_authentication(request)这个方法实现的

4.在三中的方法里面执行了request.user

5.然后执行self._authenticate()方法,主要用途是循环所有的authentication对象,执行所有authentication()方法

6.其实在这一步的时候我们已经可以用我们自己定义的类进行认证了
    ,执行self._authenticate()方法时,
  1.如果authenticate方法抛出异常,则执行self._not_authenticated()执行
  2.如果有返回值,必须是元祖:(request.user,request.auth)
  3.返回none,则交给下一个认证来处理 如果抛出错误,会继续抛出错误,也就是意味着会继续执行列表内的认证方法
7.通过认证之后,开始执行我们自己的get post等方法

梳理:

 权限

部分代码

单独目录下的权限

class MyPermission(BasePermission):    #继承内置权限
    message = '您无权访问!'    #修改通知
    def has_permission(self,request,view):
        if request.user.user_type != 3:
            return False
            #提示没有权限
        return True
        #有权限

配置文件与认证大体相同

 权限源码流程

1.同认证一样,请求进来之后先执行dispatch方法

2.对request进行封装

3.继续执行initial方法

4.对于权限继续执行check_permission方法,循环权限对象

5.循环对象之后,执行我们自己定义的has_permission方法:通过为True,否则为False,且返回错误信息,可自己修改

 节流

  节流是为了限制用户访问次数而出现的组件

示例代码如下:

import time
VISIT_RECORD = {}
class VisitThrottle(object):
    '''节流'''
    def allow_request(self,request,view):
        remote_addr = request.META.get('REMOTE_ADDR')    #获取用户IP
        c_time = time.time()
        if remote_addr not in VISIT_RECORD:
            VISIT_RECORD[remote_addr] = [c_time,]
            return True
        history = VISIT_RECORD.get(remote_addr)
        self.history = history    #将现在的history放在self.history中,以供其他地方调用
        while history and history[-1] < c_time - 60:#大于60s的就从列表中删除
            history.pop()

        if len(history) < 3:
            history.insert(0,c_time)
            return True

    def wait(self):
        c_time = time.time()
        return 60-(c_time - self.history[-1])  #显示还有多少时间可以继续访问

节流源码流程:

1.请求进来之后,依旧先执行dispatch()方法

2.对request进行封装

3.继续执行initial方法

4.get_throttle继续通过列表生成式实例化对象

5.执行我们自己写的allow_request(),True则继续,False则限制

节流的基于内置类实现的访问控制

版本控制

demo

配置文件

REST_FRAMEWORK = {
    "DEFAULT_VERSIONING_CLASS":'rest_framework.versioning.URLPathVersioning',
    "DEFAULT_VERSION":'v1',
    'ALLOWED_VERSIONS':['v1','v2'],
    "VERSION_PARAM":'version',
}

路由控制

urlpatterns = [
    re_path(r'^api/', include('app01.urls')),
]

urlpatterns = [
    re_path(r'^(?P<version>[v1|v2]+)/users/$', views.UsersView.as_view(),name='uuu'),
]

视图函数

class UsersView(APIView):
    # versioning_class = ParamVersion
    def get(self,request,*args,**kwargs):
        # version = request._request.GET.get('version') 手动获取版本号
        # print(version)
        # version = request.query_params.get('version')    #通过request的方法获取版本号

        print(request.version)    #获取版本
        print(request.versioning_scheme)   #获取处理版本的对象
        u1 = request.versioning_scheme.reverse(viewname='uuu',request=request)    #反向生成url
        print(u1)
        return HttpResponse('Marvelous')

  

源码解读

1.请求进来之后,依旧先执行dispatch()方法

2.对request进行封装

3.继续执行initial方法

4.执行determine_version方法

5.返回版本号以及版本处理对象

解析器

demo

源码解读

1.

原文地址:https://www.cnblogs.com/swearBM/p/10447638.html