Cheng Zhou
3630b9000f
## 核心改进
- 完全迁移到接口级权限（68个API端点）
- 实现前置权限检查机制，解决跨模块数据访问权限问题
- 添加权限管理API端点，支持前置权限查询
## 关键变更
1. 权限配置系统
- 添加OPERATION_PREREQUISITES映射表
- 所有操作配置前置权限依赖
2. 权限检查函数
- role_has_api_permission()支持前置权限检查
- get_missing_prerequisites()获取缺失权限列表
3. 权限管理API
- GET /api-permissions/operations - 获取所有操作及前置权限
- GET /api-permissions/operations/prerequisites - 获取前置权限依赖
- GET /api-permissions/{endpoint_key}/prerequisites - 检查缺失权限
4. API端点迁移
- 第1批：subjects, visits, aes, monitoring_visit_issues (23个)
- 第2批：members, sites, project_milestones (11个)
- 第3批：finance_contracts, fees_contracts, drug_shipments (15个)
- 第4批：startup endpoints (19个)
## 测试验证
- 单元测试：24个测试全部通过
- 集成测试：22个迁移端点测试通过
- 前置权限测试：12个测试全部通过
- 总计：46个测试全部通过
## 向后兼容性
- 模块级权限表保留用于历史数据
- 接口级权限未配置时自动回退到模块级权限
- 现有权限配置继续有效
Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
298 lines
11 KiB
Python
298 lines
11 KiB
Python
from typing import Annotated, AsyncGenerator, Callable, Iterable
|
|
import uuid
|
|
|
|
from fastapi import Depends, HTTPException, Request, status
|
|
from pydantic import ValidationError
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.exceptions import AppException
|
|
from app.core.security import decode_token, oauth2_scheme
|
|
from app.crud import user as user_crud
|
|
from app.crud import member as member_crud
|
|
from app.crud import site as site_crud
|
|
from app.core.project_permissions import role_has_project_permission, role_has_api_permission, get_missing_prerequisites
|
|
from app.db.session import SessionLocal
|
|
from app.schemas.user import TokenPayload
|
|
|
|
|
|
async def get_db_session() -> AsyncGenerator[AsyncSession, None]:
|
|
async with SessionLocal() as session:
|
|
yield session
|
|
|
|
|
|
async def get_current_user(
|
|
token: Annotated[str, Depends(oauth2_scheme)],
|
|
db: Annotated[AsyncSession, Depends(get_db_session)],
|
|
):
|
|
payload = decode_token(token)
|
|
try:
|
|
token_data = TokenPayload(**payload)
|
|
except ValidationError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="无法验证登录凭据",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
) from exc
|
|
|
|
user = await user_crud.get_by_id(db, uuid.UUID(str(token_data.sub)))
|
|
if not user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="账号不存在",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
if not user.is_active:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="账号已停用",
|
|
)
|
|
return user
|
|
|
|
|
|
def require_roles(roles: Iterable[str]) -> Callable:
|
|
roles_set = set(roles)
|
|
|
|
async def dependency(current_user=Depends(get_current_user)):
|
|
current_role = current_user.role.value if hasattr(current_user.role, "value") else str(current_user.role)
|
|
if current_role not in roles_set:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="权限不足",
|
|
)
|
|
return current_user
|
|
|
|
return dependency
|
|
|
|
|
|
async def get_study_member(
|
|
study_id: uuid.UUID,
|
|
current_user=Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db_session),
|
|
):
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if role_value == "ADMIN":
|
|
return None
|
|
return await member_crud.get_member(db, study_id, current_user.id)
|
|
|
|
|
|
def require_study_member():
|
|
async def dependency(
|
|
study_id: uuid.UUID,
|
|
current_user=Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db_session),
|
|
):
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if role_value == "ADMIN":
|
|
return current_user
|
|
membership = await member_crud.get_member(db, study_id, current_user.id)
|
|
if not membership or not membership.is_active:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message="不是该项目成员",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
return current_user
|
|
|
|
return dependency
|
|
|
|
|
|
def require_study_roles(roles: Iterable[str], *, allow_system_admin: bool = True):
|
|
roles_set = set(roles)
|
|
|
|
async def dependency(
|
|
study_id: uuid.UUID,
|
|
current_user=Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db_session),
|
|
):
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if allow_system_admin and role_value == "ADMIN":
|
|
return current_user
|
|
membership = await member_crud.get_member(db, study_id, current_user.id)
|
|
if not membership or not membership.is_active or membership.role_in_study not in roles_set:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message="项目权限不足",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
return current_user
|
|
|
|
return dependency
|
|
|
|
|
|
def require_study_permission(module: str, action: str, *, allow_system_admin: bool = True):
|
|
async def dependency(
|
|
study_id: uuid.UUID,
|
|
current_user=Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db_session),
|
|
):
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if allow_system_admin and role_value == "ADMIN":
|
|
return current_user
|
|
membership = await member_crud.get_member(db, study_id, current_user.id)
|
|
if not membership or not membership.is_active:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message="不是该项目成员",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
allowed = await role_has_project_permission(db, study_id, membership.role_in_study, module, action)
|
|
if not allowed:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message="项目权限不足",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
return current_user
|
|
|
|
return dependency
|
|
|
|
|
|
def require_api_permission(endpoint_key: str, *, allow_system_admin: bool = True, check_prerequisites: bool = True):
|
|
"""基于接口的权限检查（包含前置权限检查）
|
|
|
|
参数:
|
|
endpoint_key: 接口标识，格式为 "subjects:create"
|
|
allow_system_admin: 是否允许系统管理员绕过权限检查
|
|
check_prerequisites: 是否检查前置权限
|
|
"""
|
|
async def dependency(
|
|
study_id: uuid.UUID,
|
|
current_user=Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db_session),
|
|
):
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if allow_system_admin and role_value == "ADMIN":
|
|
return current_user
|
|
membership = await member_crud.get_member(db, study_id, current_user.id)
|
|
if not membership or not membership.is_active:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message="不是该项目成员",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
allowed = await role_has_api_permission(
|
|
db, study_id, membership.role_in_study, endpoint_key, check_prerequisites=check_prerequisites
|
|
)
|
|
if not allowed:
|
|
# 获取缺失的前置权限，用于错误提示
|
|
missing_prereqs = await get_missing_prerequisites(
|
|
db, study_id, membership.role_in_study, endpoint_key
|
|
)
|
|
if missing_prereqs:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message=f"缺失前置权限: {', '.join(missing_prereqs)}",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
else:
|
|
raise AppException(
|
|
code="FORBIDDEN",
|
|
message="接口权限不足",
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
)
|
|
return current_user
|
|
|
|
return dependency
|
|
|
|
|
|
async def get_cra_site_scope(
|
|
db: AsyncSession,
|
|
study_id: uuid.UUID,
|
|
current_user,
|
|
) -> tuple[set[uuid.UUID], set[str]] | None:
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if role_value == "ADMIN":
|
|
return None
|
|
membership = await member_crud.get_member(db, study_id, current_user.id)
|
|
if not membership or not membership.is_active:
|
|
return None
|
|
if membership.role_in_study != "CRA":
|
|
return None
|
|
site_ids = await site_crud.list_ids_by_contact_user(db, study_id, current_user.id)
|
|
site_names = await site_crud.list_names_by_contact_user(db, study_id, current_user.id)
|
|
return site_ids, site_names
|
|
|
|
|
|
def require_study_not_locked():
|
|
"""检查项目是否被锁定，如果锁定则拒绝所有修改请求"""
|
|
from app.crud import study as study_crud
|
|
from app.crud import faq_category as faq_category_crud
|
|
from app.crud import faq_item as faq_item_crud
|
|
|
|
async def resolve_study_id(request: Request, db: AsyncSession) -> uuid.UUID:
|
|
raw_study_id = request.path_params.get("study_id") or request.query_params.get("study_id")
|
|
if raw_study_id:
|
|
try:
|
|
return uuid.UUID(str(raw_study_id))
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="项目 ID 格式错误",
|
|
) from exc
|
|
|
|
try:
|
|
body = await request.json()
|
|
except Exception:
|
|
body = None
|
|
if isinstance(body, dict) and body.get("study_id"):
|
|
try:
|
|
return uuid.UUID(str(body["study_id"]))
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="项目 ID 格式错误",
|
|
) from exc
|
|
|
|
raw_category_id = request.path_params.get("category_id")
|
|
if raw_category_id:
|
|
try:
|
|
category_id = uuid.UUID(str(raw_category_id))
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="分类 ID 格式错误",
|
|
) from exc
|
|
category = await faq_category_crud.get_category(db, category_id)
|
|
if not category or not category.study_id:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="分类不存在")
|
|
return category.study_id
|
|
|
|
raw_item_id = request.path_params.get("item_id")
|
|
if raw_item_id:
|
|
try:
|
|
item_id = uuid.UUID(str(raw_item_id))
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="FAQ ID 格式错误",
|
|
) from exc
|
|
item = await faq_item_crud.get_item(db, item_id)
|
|
if not item or not item.study_id:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="FAQ 不存在")
|
|
return item.study_id
|
|
|
|
raise HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail="缺少项目 ID",
|
|
)
|
|
|
|
async def dependency(
|
|
request: Request,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
):
|
|
study_id = await resolve_study_id(request, db)
|
|
study = await study_crud.get(db, study_id)
|
|
if not study:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="项目不存在",
|
|
)
|
|
if study.is_locked:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="项目已锁定，无法进行任何修改操作",
|
|
)
|
|
return study
|
|
|
|
return dependency
|