django添加装饰器进行登录角色验证

from django.http import JsonResponse

from functools import wrapsfrom utils.token import get_userid
from yshop.models import MyUser


def check_role(view_func):
    @wraps(view_func)
    def wrapper(request, *args, **kwargs):
        user_token = args[0].META.get('HTTP_AUTHORIZATION')
        user_id = get_userid(user_token)
        try:
            user_info = MyUser.objects.get(user_id=user_id)
            if user_info.role != 'user':
                return view_func(request, *args, **kwargs)
            else:
                return JsonResponse({'code': 403, 'msg': '权限错误！'})
        except Exception as e:
            return JsonResponse({'code': 405, 'msg': '未知错误，请联系管理员！'})

    return wrapper

class DataStatistics(APIView):

    @check_role
    def post(self, request):
        ''' 其他代码 '''
        return Response({'code': 200, 'msg': "数据查询成功!"})