权限管理Django中间件

权限管理django中间件

from django.http import JsonResponse
from django.http.request import HttpRequest
from django.http.response import HttpResponse
from django.utils.deprecation import MiddlewareMixin
from django.apps import apps
import jwt

class MyMiddleware(MiddlewareMixin):
    def __init__(self, get_response=None):
        """
        初始化中间件，获取必要的模型引用。
        
        参数:
            get_response (callable, optional): 链中的下一个中间件。
        """
        super().__init__(get_response)
        # 获取必要的模型
        self.UserModel = apps.get_model('userlogin', 'User')
        self.UserRoleModel = apps.get_model('usermanage', 'UserRole')
        self.RolePermissionModel = apps.get_model('usermanage', 'RolePermission')
        self.RouteModel = apps.get_model('usermanage', 'Route')
    
    def __call__(self, request: HttpRequest) -> HttpResponse:
        """
        处理传入的请求。

        参数:
            request (HttpRequest): 传入的HTTP请求。

        返回:
            HttpResponse: HTTP响应。
        """
        try:
            # 从Authorization头部提取JWT令牌
            token = request.headers['Authorization'].split(' ')[0]
            # 解码令牌（注意：此处暂时禁用了签名验证）
            user = jwt.decode(token, algorithms=["RS256"], options={"verify_signature": False})
            # 使用解码后的令牌中的sub（主题）获取用户ID
            user_id = self.UserModel.objects.get(sub=user['sub']).user_id
            # 获取与用户关联的角色
            user_role = self.UserRoleModel.objects.filter(user_id=user_id).values_list('role_id', flat=True)
            # 获取与用户角色相关联的权限
            user_permission = self.RolePermissionModel.objects.filter(role_id__in=user_role).values_list('permission_id', flat=True)
            # 获取用户允许的URL
            permitted_urls = self.RouteModel.objects.filter(permission_id__in=user_permission).values_list('route_url', flat=True)
            # 判断当前请求的URL是否在用户允许的URL列表中
            if request.path in permitted_urls:
                # 如果是，则继续处理请求
                return self.get_response(request)
            else:
                # 如果不是，则返回权限不足的JSON响应
                return JsonResponse({'code': 403, 'msg': '没有权限'})
        except KeyError:
            return JsonResponse({'code': 400, 'msg': 'Authorization头部缺少令牌'})
        except jwt.DecodeError:
            return JsonResponse({'code': 401, 'msg': '令牌无效或过期'})
        except self.UserModel.DoesNotExist:
            return JsonResponse({'code': 401, 'msg': '用户不存在'})
        except Exception as e:
            return JsonResponse({'code': 500, 'msg': '服务器内部错误'})