源码怎么找之rest_framework的用户认证

首先得有一点常识，比如用户认证，就是authenticate

比如一个函数，应该有返回值，

比如一个类里面的self，真的是代表本身这个类吗

再比如看到一个东西加括号，就两种情况，一种是函数，一种是类，区分函数和类，就看加括号之后的东西，是否还调用属性或者方法

还有如果你用pycharm，看源码就很方便了，看到一个属性或方法，你需要做的就是按住ctrl点进去就行了

 

假设你已经知道，dispatch函数是源码的入口，然后我们就可以愉快的交流了

下面演示如何使用rest_framework的用户认证功能，以及为什么要这样做

1.写路由匹配

　　因为是cbv视图，所以用as_view()　　

from django.conf.urls import url
from django.contrib import admin
from app01 import views

urlpatterns = [
    url(r'^admin/', admin.site.urls),
    url(r'^test/', views.TestView.as_view()),
]

2.CBV风格视图

from django.shortcuts import HttpResponse

from rest_framework.views import APIView#导入rest_framework的试图类

class TestView(APIView):
　　self.dispatch　　　　　　#只是为了进源码
    def get(self,request,*args,**kwargs):
        return HttpResponse('ok')

好了，最基本的框架就这样搭好了，我们开始看源码，源码的入口是在   APIView 的dispatch方法，所以我们点进去

 

dispatch函数：

    def dispatch(self, request, *args, **kwargs):#这是APIView类的入口
        """
        `.dispatch()` is pretty much the same as Django's regular dispatch,
        but with extra hooks for startup, finalize, and exception handling.
        和Django的dispatch类似，但是增加了额外的钩子，比如开始，结束，以及异常的处理
        """
        self.args = args    #封装参数
        self.kwargs = kwargs
        request = self.initialize_request(request, *args, **kwargs)     #重新封装了request
        self.request = request
        self.headers = self.default_response_headers  # deprecate?

        try:
            self.initial(request, *args, **kwargs)

            # Get the appropriate handler method
            if request.method.lower() in self.http_method_names:
                handler = getattr(self, request.method.lower(),
                                  self.http_method_not_allowed)
            else:
                handler = self.http_method_not_allowed

            response = handler(request, *args, **kwargs)

        except Exception as exc:
            response = self.handle_exception(exc)

        self.response = self.finalize_response(request, response, *args, **kwargs)
        return self.response

经过观察，这个函数里面，能让我们感觉有东西的地方就是重新封装request和self.initial初识化这两个地方

1.1先看重新封装request这个

    def initialize_request(self, request, *args, **kwargs):
        """
        Returns the initial request object.
        返回初始化后的request
        """
        parser_context = self.get_parser_context(request)

        return Request(
            request,
            parsers=self.get_parsers(),
            authenticators=self.get_authenticators(),       #重点关注这个，看名字就应该找到这个，继续点进去
            negotiator=self.get_content_negotiator(),
            parser_context=parser_context          
        )  

1.2

def get_authenticators(self):
        """
        Instantiates and returns the list of authenticators that this view can use.
        """
        return [auth() for auth in self.authentication_classes] #从配置读取用户认证配置，然后实例化，点进这个self.authentication_classes

这个函数返回了一个列表，通过列表生成式，循环    self.authentication_classes   ，然后拿到的东西实例化

tips：看到一个东西加括号，就两种情况，一种是函数，一种是类，区分函数和类，就看加括号之后的东西，是否还调用属性或者方法

1.3

class APIView(View):

    # The following policies may be set at either globally, or per-view.
    renderer_classes = api_settings.DEFAULT_RENDERER_CLASSES        
    parser_classes = api_settings.DEFAULT_PARSER_CLASSES
    authentication_classes = api_settings.DEFAULT_AUTHENTICATION_CLASSES　　　　　　#
    throttle_classes = api_settings.DEFAULT_THROTTLE_CLASSES

坏了，这下不能点了，看settings，说明这是配置信息，好了，看request看到这就走不下去了，因为不能点了，现在我们拿到的信息是：

　　在封装request的时候， 把一个auth()列表传到了request的参数里面，然后这个auth（）是从这个self.authentication_classes 里面拿到的

现在我们可以打印一下这个  self.authentication_classes，self就是APIview 嘛，我门的视图函数就是继承了这个类啊，所以，打印一下：

拿到了这个：

[<class 'rest_framework.authentication.SessionAuthentication'>, <class 'rest_framework.authentication.BasicAuthentication'>]

这种情况应该兴奋，因为真的有东西，拿到的是个路径，然后我们就可以用这个路径拿到这两个类

并且，验证了self.authentication_classes  就是个列表，然后里面是一个一个类

如果我们自己写这个authentication_classes的话，也要写成一个列表，里面是一个个类，下面看一下这个类要怎么写

1.4

然后我们先导入一下这个路径，看一下里面写的啥，

from rest_framework.authentication import SessionAuthentication

点进去

是一个新的文件

 

 总体看一下，这个文件有这样几个类

然后都继承了BaseAuthentication类，先看一下这个类

1.5

class BaseAuthentication(object):
    """
    All authentication classes should extend BaseAuthentication.
    """

    def authenticate(self, request):
        """
        Authenticate the request and return a two-tuple of (user, token).
        """
        raise NotImplementedError(".authenticate() must be overridden.")

    def authenticate_header(self, request):
        """
        Return a string to be used as the value of the `WWW-Authenticate`
        header in a `401 Unauthenticated` response, or `None` if the
        authentication scheme should return `403 Permission Denied` responses.
        """
        pass

没有什么实质性的东西，但是规定了继承这个类的子类，必须要有这个   authenticate   方法，否则就会报错，所以我们就知道了应该怎么自己写这个类了

那就可以猜到，下面继承了这个基类的几个子类就是不同的认证方法了

看了这个基类，大概知道自己定义的类里面必须要写这样一个authenticate   方法，不写会报错，但是应该怎么写呢？需要返回什么呢，其实重点是要返回什么

 

我们现在才走了一半，只看了封装request这一部分，还有初始化呢，重新点进dispatch函数，找到

2.1

    def initial(self, request, *args, **kwargs):
        """
        Runs anything that needs to occur prior to calling the method handler.
        """
        self.format_kwarg = self.get_format_suffix(**kwargs)

        # Perform content negotiation and store the accepted info on the request
       

        neg = self.perform_content_negotiation(request) 
        request.accepted_renderer, request.accepted_media_type = neg

        # Determine the API version, if versioning is in use.
     
        version, scheme = self.determine_version(request, *args, **kwargs)
        request.version, request.versioning_scheme = version, scheme 

        # Ensure that the incoming request is permitted
        #用户认证
        self.perform_authentication(request)
      
        self.check_permissions(request)
      
        self.check_throttles(request)

 

 我们雄厚的英文实力告诉我们，这个东西是我们要的       self.perform_authentication(request)　　　　　　-------执行认证

注意这里传了参数，是封装后的request

2.2

    def perform_authentication(self, request):
        """
        Perform authentication on the incoming request.

        Note that if you override this and simply 'pass', then authentication
        will instead be performed lazily, the first time either
        `request.user` or `request.auth` is accessed.
        """
        request.user

就执行了一个request.user，这个user应该是个方法，才能直接这么用，所以再找request里面的user方法，

2.3

    @property
    def user(self):
        """
        Returns the user associated with the current request, as authenticated
        by the authentication classes provided to the request.
        """
        if not hasattr(self, '_user'):
            self._authenticate()
        return self._user

果不其然，是一个加了装饰器的方法，我们的request里面没有这个_user属性，所以执行的是   self._authenticate()  这个方法

2.4

    def _authenticate(self):
        """
        Attempt to authenticate the request using each authentication instance
        in turn.
        """
        for authenticator in self.authenticators:　　　　#
            try:
                user_auth_tuple = authenticator.authenticate(self)
            except exceptions.APIException:
                self._not_authenticated()
                raise

            if user_auth_tuple is not None:
                self._authenticator = authenticator
                self.user, self.auth = user_auth_tuple
                return

        self._not_authenticated()

循环   self.authenticators，拿到   authenticator 并执行 authenticator 里面的  authenticate 方法，参数为request

 

注意， self.authenticators  这个东西在1.1中就已经拿到了，这个就是我们要写的验证的类，然后跟我们得到的结论也一样，有一个authentic 方法

 

2.5 看看这个方法，authentic 

    def authenticate(self, request):
        return (self.force_user, self.force_token)

 

哦，返回一个元组，大概就是  用户和token，现在这两行代码是如果有指定值的时候的情况，但是可以说明问题

再看看1.4里面那几个验证的子类，基本上就确认是返回一个已通过认证的用户和对应的token

这是通过认证的情况

 

2.6再看2.4里面，最后那个     self._not_authenticated()   ，是表示未通过验证的情况

点进去

    def _not_authenticated(self):
        """
        Set authenticator, user & authtoken representing an unauthenticated request.

        Defaults are None, AnonymousUser & None.
        """
        self._authenticator = None

        if api_settings.UNAUTHENTICATED_USER:
            self.user = api_settings.UNAUTHENTICATED_USER()
        else:
            self.user = None

        if api_settings.UNAUTHENTICATED_TOKEN:
            self.auth = api_settings.UNAUTHENTICATED_TOKEN()
        else:
            self.auth = None

没有默认值的话，是返回None的，但是是有默认值的

点进去api_settings ,是个类

api_settings = APISettings(None, DEFAULTS, IMPORT_STRINGS)

再点进去，就是rest_framework的配置文件

搜一下，就可以看到这条配置

'UNAUTHENTICATED_USER': 'django.contrib.auth.models.AnonymousUser',
 'UNAUTHENTICATED_TOKEN': None,

默认是调用Django的匿名用户类，所以默认如果不登录，是匿名用户

 

还能说明一个问题，不登录也能访问，即匿名用户访问是否被允许，就可以在认证类中配置了，返回None，就是允许，抛出异常就是不允许

 

贴一个完整的代码

from django.shortcuts import HttpResponse

from rest_framework.views import APIView#导入rest_framework的试图类

from rest_framework.authentication import BaseAuthentication
from rest_framework import exceptions
from . import models
class CustomAuthenticate(BaseAuthentication):
    """
    All authentication classes should extend BaseAuthentication.
    """

    def authenticate(self, request):
        """
        Authenticate the request and return a two-tuple of (user, token).
        """
        tk = request.query_param.get('tk')      #获取用户输入的token   简化写法，就直接在url中传参了
        token_obj = models.Token.objects.filter(token=tk).first()
        if token_obj:
            return (token_obj.user,token_obj)


        raise exceptions.AuthenticationFailed('请登录')    #抛出异常表示不允许匿名用户访问

        # else:
        #     return (None,None)
    def authenticate_header(self, request):
        """
        Return a string to be used as the value of the `WWW-Authenticate`
        header in a `401 Unauthenticated` response, or `None` if the
        authentication scheme should return `403 Permission Denied` responses.
        """
        pass
class TestView(APIView):
    authentication_classes = [CustomAuthenticate,]          #应用认证配置类
    def get(self,request,*args,**kwargs):
        return HttpResponse('ok')