1461a8d113
按 2026-11-02 需求补齐计划交付后端能力： - 角色授权：权限目录化，接口统一准入校验，未登记默认拒绝； 管理员判断改为按数据库角色重新判定，防止权限码持有者绕过 - 系统监控：单进程采集资源、请求与异常指标，分钟级聚合， 阈值配置化；敏感配置增加 repr 保护，Redis 密码 URL 转义 - 试用期计提：按回款日判定并写入快照，年度结清与调整单联动； 人员模型补试用期、离职日期与工资版本历史字段 - 计提模板：公式骨架受控，页面维护名称与参数绑定 - T+ 采集：自动化配置表与进度查询 内置角色禁停用保护对齐模块注释；核算规则版本提为常量并 补充批量核算失败堆栈日志。 影响范围：后端全部模块、数据库迁移 V22-V28 验证：uv run ruff check 通过；922 个单元测试全部通过； 集成测试因本机 MySQL 数据卷被其他容器锁定暂未运行
138 lines
5.2 KiB
Python
138 lines
5.2 KiB
Python
"""用户数据访问（角色相关的读取与维护见 `role_repository`）。
|
||
|
||
本层只 flush 不 commit —— 事务边界由 services 层控制。
|
||
"""
|
||
|
||
from datetime import datetime
|
||
|
||
from sqlalchemy import Select, func, select
|
||
from sqlalchemy.orm import Session, selectinload
|
||
|
||
from app.models import Role, User, UserRole
|
||
from app.repositories.query_utils import fuzzy
|
||
from app.schemas.user import UserQuery
|
||
|
||
|
||
def _base_query() -> Select[tuple[User]]:
|
||
"""用户查询基座。
|
||
|
||
预加载角色关联，避免列表渲染时逐行查角色（N+1）。
|
||
"""
|
||
return select(User).options(selectinload(User.user_roles).selectinload(UserRole.role))
|
||
|
||
|
||
def get_by_username(session: Session, username: str) -> User | None:
|
||
"""按用户名查询用户。不存在返回 None。"""
|
||
return session.execute(select(User).where(User.username == username)).scalar_one_or_none()
|
||
|
||
|
||
def get_by_id(session: Session, user_id: int) -> User | None:
|
||
"""按主键查询用户。不存在返回 None。"""
|
||
return session.get(User, user_id)
|
||
|
||
|
||
def get_role_codes(user: User) -> list[str]:
|
||
"""取用户的角色编码列表。
|
||
|
||
依赖 `User.user_roles` 的 `lazy="selectin"` 预加载（见模型定义），
|
||
故此处遍历不会触发 N+1 查询。
|
||
"""
|
||
return [link.role.role_code for link in user.user_roles if link.role is not None and link.role.status == 1]
|
||
|
||
|
||
def touch_last_login(session: Session, user: User, moment: datetime) -> None:
|
||
"""更新最后登录时间。只 flush，提交由 service 决定。"""
|
||
user.last_login_time = moment
|
||
session.flush()
|
||
|
||
|
||
# ==================== 列表查询 ====================
|
||
|
||
|
||
def list_users(session: Session, query: UserQuery) -> list[User]:
|
||
"""分页查询用户列表。
|
||
|
||
**按 id 倒序而非 create_time**：create_time 是秒级精度，批量导入的用户
|
||
可能落在同一秒，按它排序的分页结果不确定 —— 翻页时会出现重复或遗漏。
|
||
id 是自增主键，天然唯一且与创建顺序一致。
|
||
"""
|
||
stmt = _base_query()
|
||
stmt = _apply_filters(stmt, query)
|
||
if query.role_code:
|
||
# 按角色筛选需 join 关联表；角色是字典数据，数量极少，join 代价可忽略
|
||
stmt = stmt.join(UserRole).join(Role).where(Role.role_code == query.role_code)
|
||
stmt = stmt.order_by(User.id.desc()).offset(query.offset).limit(query.page_size)
|
||
return list(session.execute(stmt).scalars().unique())
|
||
|
||
|
||
def count_users(session: Session, query: UserQuery) -> int:
|
||
"""统计符合条件的用户总数，供分页使用。"""
|
||
stmt = select(func.count(User.id))
|
||
stmt = _apply_filters(stmt, query)
|
||
if query.role_code:
|
||
stmt = stmt.join(UserRole).join(Role).where(Role.role_code == query.role_code)
|
||
return int(session.execute(stmt).scalar_one())
|
||
|
||
|
||
def _apply_filters(stmt: Select, query: UserQuery) -> Select:
|
||
"""应用通用筛选条件。列表与计数共用，避免两处口径不一致。
|
||
|
||
模糊匹配统一走 `query_utils.fuzzy`（转义 LIKE 通配符）—— 用户名普遍
|
||
含下划线，直接拼接会让 `_` 当单字符通配执行、误配无关账号。
|
||
"""
|
||
if query.username:
|
||
stmt = fuzzy(stmt, User.username, query.username)
|
||
if query.real_name:
|
||
stmt = fuzzy(stmt, User.real_name, query.real_name)
|
||
if query.status is not None:
|
||
stmt = stmt.where(User.status == query.status)
|
||
return stmt
|
||
|
||
|
||
# ==================== 写入 ====================
|
||
|
||
|
||
def create(session: Session, user: User) -> User:
|
||
"""新增用户。只 flush 以取得自增 id。"""
|
||
session.add(user)
|
||
session.flush()
|
||
return user
|
||
|
||
|
||
def delete(session: Session, user: User) -> None:
|
||
"""物理删除用户。
|
||
|
||
本项目对用户采用「删除即禁用」的**业务**约定（见模型注释），
|
||
但保留物理删除能力供测试清理与误建账号的彻底移除使用。
|
||
`user_roles` 与 `operation_logs` 的外键分别配置了 CASCADE 与 SET NULL，
|
||
故不会留下悬挂引用。
|
||
"""
|
||
session.delete(user)
|
||
session.flush()
|
||
|
||
|
||
# ==================== 角色关联 ====================
|
||
|
||
|
||
def replace_roles(session: Session, user: User, roles: list[Role]) -> None:
|
||
"""整体替换用户的角色。
|
||
|
||
采用「先清后建」而非逐条比对：角色数量极少（本期仅两个），
|
||
差异计算的复杂度换不来收益，且整体替换的语义更直白。
|
||
|
||
**clear 之后必须立刻 flush 一次**：SQLAlchemy 在同一事务的 flush 中
|
||
**先处理 INSERT 再处理 DELETE**，若中间不落库，重插与旧集合有交集的
|
||
(user_id, role_id)（编辑用户时保留既有角色就会发生）会撞
|
||
`uk_user_roles_user_id_role_id` 唯一键，报 1062。
|
||
|
||
**必须经 `user.user_roles` 集合追加，不能用 `session.add(UserRole(...))`**：
|
||
后者只把对象放进 session，不会同步到关系的已加载集合，于是同一事务内
|
||
再读 `user.user_roles` 仍是空的 —— 表现为「保存成功，但接口返回的角色
|
||
列表是 []」。经集合操作时 `user_id` 由关系自动填充，集合也立即反映新状态。
|
||
"""
|
||
user.user_roles.clear()
|
||
session.flush()
|
||
for role in roles:
|
||
user.user_roles.append(UserRole(role_id=role.id))
|
||
session.flush()
|