简单实例
settings.py
INSTALLED_APPS = [
'rest_framework',
]
urls.py
urlpatterns = [
path('admin/', admin.site.urls),
path('api/v1/auth/',views.AuthView.as_view()),
]
models.py
from django.db import models
class UserInfo(models.Model):
USER_TYPE = (
(1,'普通用户'),
(2,'VIP'),
(3,'SVIP')
)
user_type = models.IntegerField(choices=USER_TYPE)
username = models.CharField(max_length=32)
password = models.CharField(max_length=64)
class UserToken(models.Model):
user = models.OneToOneField(UserInfo,on_delete=models.CASCADE)
token = models.CharField(max_length=64)
views.py
from django.shortcuts import render,HttpResponse
from django.http import JsonResponse
from rest_framework.views import APIView
from API import models
from rest_framework.request import Request
from rest_framework import exceptions
from rest_framework.authentication import BasicAuthentication
ORDER_DICT = {
1:{
'name':'apple',
'price':15
},
2:{
'name':'dog',
'price':100
}
}
def md5(user):
import hashlib
import time
#当前时间，相当于生成一个随机的字符串
ctime = str(time.time())
m = hashlib.md5(bytes(user,encoding='utf-8'))
m.update(bytes(ctime,encoding='utf-8'))
return m.hexdigest()
class AuthView(object):
'''用于用户登录验证'''
def post(self,request,*args,**kwargs):
ret = {'code':1000,'msg':None}
try:
user = request._request.POST.get('username')
pwd = request._request.POST.get('password')
obj = models.UserInfo.objects.filter(username=user,password=pwd).first()
if not obj:
ret['code'] = 1001
ret['msg'] = '用户名或密码错误'
#为用户创建token
token = md5(user)
#存在就更新，不存在就创建
models.UserToken.objects.update_or_create(user=obj,defaults={'token':token})
ret['token'] = token
except Exception as e:
ret['code'] = 1002
ret['msg'] = '请求异常'
return JsonResponse(ret)
class Authentication(APIView):
'''认证'''
def authenticate(self,request):
token = request._request.GET.get('token')
token_obj = models.UserToken.objects.filter(token=token).first()
if not token_obj:
raise exceptions.AuthenticationFailed('用户认证失败')
#在rest framework内部会将这两个字段赋值给request，以供后续操作使用
return (token_obj.user,token_obj)
def authenticate_header(self, request):
pass
class OrderView(APIView):
'''订单相关业务'''
authentication_classes = [Authentication,] #添加认证
def get(self,request,*args,**kwargs):
#request.user
#request.auth
ret = {'code':1000,'msg':None,'data':None}
try:
ret['data'] = ORDER_DICT
except Exception as e:
pass
return JsonResponse(ret)
总结：
-写一个认证类（继承BaseAuthentication）
-重写authenticate方法，把request对象传入
-能从request对象中取出用户携带的token，根据token判断是否登录过
-如果登录过，返回两个值 user对象，token对象
-如果没有登录过抛异常
-全局使用
-在settings中配置
REST_FRAMEWORK={
"DEFAULT_AUTHENTICATION_CLASSES":["app01.service.auth.Authentication",]
}
-局部使用
-在视图类中配置：
authentication_classes=[Authentication,]
-局部禁用：
-在视图类中配置：
authentication_classes=[]
认证源码分析
-APIView中的dispatch ---> self.initial（认证，频率，权限） ---> self.perform_authentication（认证） ---> 本质又调用了新的request对象的user方法
--> request.user内部执行了：self._authenticate(注意self是新的request对象) ---> 循环拿到一个个认证类的对象，执行对象的authenticate方法，传入request对象
---> 一个个认证类的对象是在reuqest对象实例化的时候传入的 ---> APIView中的get_authenticators，通过列表推导式生成一个个的认证类对象，然后传入request对象中
