django实现自定义登陆验证

django实现自定义登陆验证

---

• 自定义装饰器函数和类

•  from utils.http import HttpResponseUnauthorized
   from django.views import View
  
   # 登陆校验装饰器   函数视图校验
   def login_require(func):
       def wrapper(request):
  
           # 校验session中是否存在login_flag
           login_flag = request.session.get('login_flag')
  
           if not login_flag:
               return HttpResponseUnauthorized('用户未登录')
  
           result = func(request)
           return result
  
       return wrapper
  
   # 登陆校验视图类    类视图校验
   class LoginRequireMixin(View):
       @classmethod
       def as_view(cls, **initkwargs):
           # 对父类的as_view进行改写并封装
           view = super(LoginRequireMixin, cls).as_view(**initkwargs)
           return login_require(view)

• HttpResponseUnauthorized

   from django.http import HttpResponse
  
   class HttpResponseUnauthorized(HttpResponse):
       status_code = 401

• 具体使用

•  # 保存成绩
   @login_require
   def saveScore(request):
       if request.method == 'POST':
  
           open_id = request.POST.get('open_id')
           nickName = request.POST.get('nickName')
           gameType = request.POST.get('gameType')
           spendTime = request.POST.get('spendTime')
           ageType = request.POST.get('ageType')
  
           if not all([open_id, nickName, gameType, spendTime, ageType]):
               date = {'success': False, 'info': '请求数据不完整'}
               return JsonResponse(date)
  
           if request.session.get('openId') != open_id:
               data = {'success': False, 'info': '请求玩家和使用玩家不一致'}
               return JsonResponse(data)
  
           Score(open_id=open_id, nickName=nickName, gameType=gameType, spendTime=spendTime, ageType=ageType).save()
           data = {'success': True, 'info': '保存成功!'}
           return JsonResponse(data)
  
       return HttpResponseForbidden('Method not allowed')
  
   # 分类成绩查询
   class RankingQuery(LoginRequireMixin):
  
       def post(self, request):
           queryType = request.POST.get('queryType')
           offset = request.POST.get('offset')
  
           if not all([queryType, offset]):
               data = {'success': False, 'info': '数据不完整'}
               return JsonResponse(data)
  
           queryTypeRoute = {
               'byScore': self.queryByScore
           }
           if queryType not in queryTypeRoute:
               return HttpResponseBadRequest('unsupport queryType')
  
           func = queryTypeRoute['queryType']
           data = func(offset)
  
           return JsonResponse(data)