mirror of
https://github.com/insistence/RuoYi-Vue3-FastAPI.git
synced 2026-09-25 13:51:05 +00:00
95 lines
4.1 KiB
Python
95 lines
4.1 KiB
Python
from fastapi import Depends, Request, params
|
|
|
|
from common.context import RequestContext
|
|
from exceptions.exception import PermissionException
|
|
from utils.dependency_util import DependencyUtil
|
|
|
|
|
|
class CheckUserInterfaceAuth:
|
|
"""
|
|
校验当前用户是否具有相应的接口权限
|
|
"""
|
|
|
|
def __init__(self, perm: str | list, is_strict: bool = False) -> None:
|
|
"""
|
|
校验当前用户是否具有相应的接口权限
|
|
|
|
:param perm: 权限标识
|
|
:param is_strict: 当传入的权限标识是list类型时，是否开启严格模式，开启表示会校验列表中的每一个权限标识，所有的校验结果都需要为True才会通过
|
|
"""
|
|
self.perm = perm
|
|
self.is_strict = is_strict
|
|
|
|
def __call__(self, request: Request) -> bool:
|
|
DependencyUtil.check_exclude_routes(
|
|
request, err_msg='当前路由不在认证规则内，不可使用CheckUserInterfaceAuth依赖项'
|
|
)
|
|
current_user = RequestContext.get_current_user()
|
|
user_auth_list = current_user.permissions
|
|
if '*:*:*' in user_auth_list:
|
|
return True
|
|
if isinstance(self.perm, str) and self.perm in user_auth_list:
|
|
return True
|
|
if isinstance(self.perm, list):
|
|
if self.is_strict:
|
|
if all(perm_str in user_auth_list for perm_str in self.perm):
|
|
return True
|
|
elif any(perm_str in user_auth_list for perm_str in self.perm):
|
|
return True
|
|
raise PermissionException(data='', message='该用户无此接口权限')
|
|
|
|
|
|
class CheckRoleInterfaceAuth:
|
|
"""
|
|
根据角色校验当前用户是否具有相应的接口权限
|
|
"""
|
|
|
|
def __init__(self, role_key: str | list, is_strict: bool = False) -> None:
|
|
"""
|
|
根据角色校验当前用户是否具有相应的接口权限
|
|
|
|
:param role_key: 角色标识
|
|
:param is_strict: 当传入的角色标识是list类型时，是否开启严格模式，开启表示会校验列表中的每一个角色标识，所有的校验结果都需要为True才会通过
|
|
"""
|
|
self.role_key = role_key
|
|
self.is_strict = is_strict
|
|
|
|
def __call__(self, request: Request) -> bool:
|
|
DependencyUtil.check_exclude_routes(
|
|
request, err_msg='当前路由不在认证规则内，不可使用CheckRoleInterfaceAuth依赖项'
|
|
)
|
|
current_user = RequestContext.get_current_user()
|
|
user_role_list = current_user.user.role
|
|
user_role_key_list = [role.role_key for role in user_role_list]
|
|
if isinstance(self.role_key, str) and self.role_key in user_role_key_list:
|
|
return True
|
|
if isinstance(self.role_key, list):
|
|
if self.is_strict:
|
|
if all(role_key_str in user_role_key_list for role_key_str in self.role_key):
|
|
return True
|
|
elif any(role_key_str in user_role_key_list for role_key_str in self.role_key):
|
|
return True
|
|
raise PermissionException(data='', message='该用户无此接口权限')
|
|
|
|
|
|
def UserInterfaceAuthDependency(perm: str | list, is_strict: bool = False) -> params.Depends: # noqa: N802
|
|
"""
|
|
根据权限标识校验当前用户接口权限依赖
|
|
|
|
:param perm: 权限标识
|
|
:param is_strict: 当传入的权限标识是list类型时，是否开启严格模式，开启表示会校验列表中的每一个权限标识，所有的校验结果都需要为True才会通过
|
|
:return: 根据权限标识校验当前用户接口权限依赖
|
|
"""
|
|
return Depends(CheckUserInterfaceAuth(perm, is_strict))
|
|
|
|
|
|
def RoleInterfaceAuthDependency(role_key: str | list, is_strict: bool = False) -> params.Depends: # noqa: N802
|
|
"""
|
|
根据角色校验当前用户接口权限依赖
|
|
|
|
:param role_key: 角色标识
|
|
:param is_strict: 当传入的角色标识是list类型时，是否开启严格模式，开启表示会校验列表中的每一个角色标识，所有的校验结果都需要为True才会通过
|
|
:return: 根据角色校验当前用户接口权限依赖
|
|
"""
|
|
return Depends(CheckRoleInterfaceAuth(role_key, is_strict))
|