57957d11a0
- 将AI验审改为AI预审，高风险不再拦截而是随单流转给审批人复核 - 新增平台风险规则评估引擎，支持事由过短、票据异常、重复发票等多种评估器 - 用户上下文增加部门信息(department_name)，认证流程同步关联组织架构 - 规则scenario_json改为中文标签(差旅/费用科目)，统一场景分类 - 新增orchestrator审核流程测试用例 - 前端更新审计视图、差旅报销等相关页面
107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
from collections.abc import Generator
|
|
from dataclasses import dataclass
|
|
from typing import Annotated
|
|
|
|
from fastapi import Depends, Header, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.db.session import get_session_factory
|
|
|
|
|
|
def get_db() -> Generator[Session, None, None]:
|
|
db = get_session_factory()()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class CurrentUserContext:
|
|
username: str
|
|
name: str
|
|
role_codes: list[str]
|
|
is_admin: bool
|
|
department_name: str = ""
|
|
|
|
|
|
def get_current_user(
|
|
x_auth_username: Annotated[
|
|
str | None,
|
|
Header(description="当前登录用户名。知识库接口至少需要提供用户名或姓名。"),
|
|
] = None,
|
|
x_auth_name: Annotated[
|
|
str | None,
|
|
Header(description="当前登录人展示姓名。未传时默认回退到用户名。"),
|
|
] = None,
|
|
x_auth_role_codes: Annotated[
|
|
str | None,
|
|
Header(description="角色编码列表，多个角色使用英文逗号分隔，例如 `manager,finance`。"),
|
|
] = None,
|
|
x_auth_is_admin: Annotated[
|
|
str | None,
|
|
Header(description="是否管理员，支持 `true/false/1/0`。"),
|
|
] = None,
|
|
x_auth_department: Annotated[
|
|
str | None,
|
|
Header(description="当前登录人的所属部门。"),
|
|
] = None,
|
|
) -> CurrentUserContext:
|
|
role_codes = [item.strip() for item in (x_auth_role_codes or "").split(",") if item.strip()]
|
|
is_admin = str(x_auth_is_admin or "").strip().lower() in {"1", "true", "yes", "on"}
|
|
|
|
username = (x_auth_username or "").strip()
|
|
name = (x_auth_name or username).strip()
|
|
|
|
if not username and not name:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="请先登录后再访问知识库。",
|
|
)
|
|
|
|
return CurrentUserContext(
|
|
username=username or name,
|
|
name=name or username,
|
|
role_codes=role_codes,
|
|
is_admin=is_admin,
|
|
department_name=(x_auth_department or "").strip(),
|
|
)
|
|
|
|
|
|
def require_admin_user(
|
|
current_user: Annotated[CurrentUserContext, Depends(get_current_user)],
|
|
) -> CurrentUserContext:
|
|
if current_user.is_admin or "manager" in current_user.role_codes:
|
|
return current_user
|
|
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="只有管理员可以上传、删除或修改知识库文件。",
|
|
)
|
|
|
|
|
|
def require_rule_editor_user(
|
|
current_user: Annotated[CurrentUserContext, Depends(get_current_user)],
|
|
) -> CurrentUserContext:
|
|
role_codes = {item.strip() for item in current_user.role_codes}
|
|
if current_user.is_admin or "manager" in role_codes or "finance" in role_codes:
|
|
return current_user
|
|
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="只有财务人员或高级管理人员可以编辑规则草稿。",
|
|
)
|
|
|
|
|
|
def require_rule_reviewer_user(
|
|
current_user: Annotated[CurrentUserContext, Depends(get_current_user)],
|
|
) -> CurrentUserContext:
|
|
role_codes = {item.strip() for item in current_user.role_codes}
|
|
if current_user.is_admin or "manager" in role_codes:
|
|
return current_user
|
|
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="只有高级管理人员可以审核、发布或恢复正式规则。",
|
|
)
|