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)