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)

django实现自定义登陆验证的相关教程结束。