Files
FastapiAdmin/backend/app/api/v1/module_system/user/crud.py
T
zhangtao 73f2823692 refactor: 大规模代码整理与功能优化
1. 重构后端API路由、CRUD与模块结构,整合日志管理,移除废弃demo代码
2. 优化前端组件类型定义、样式与路由配置,修复权限判断逻辑
3. 调整默认排序规则、滚动条样式与工具类函数,更新依赖与配置文件
4. 修复多处类型不匹配与默认值问题,完善表单与菜单验证逻辑
2026-06-17 01:56:31 +08:00

103 lines
2.9 KiB
Python

from datetime import datetime
from app.api.v1.module_system.position.crud import PositionCRUD
from app.api.v1.module_system.role.crud import RoleCRUD
from app.core.base_crud import CRUDBase
from app.core.base_schema import AuthSchema
from .model import UserModel
from .schema import (
UserCreateSchema,
UserUpdateSchema,
)
class UserCRUD(CRUDBase[UserModel, UserCreateSchema, UserUpdateSchema]):
"""用户模块数据层"""
def __init__(self, auth: AuthSchema) -> None:
"""
初始化用户数据层。
参数:
- auth (AuthSchema): 认证信息模型(含 DB 会话等上下文)。
返回:
- None
"""
super().__init__(model=UserModel, auth=auth)
async def update_last_login(self, id: int) -> None:
"""
更新用户最后登录时间
参数:
- id (int): 用户ID
"""
await self.set([id], last_login=datetime.now())
async def set_user_roles(self, user_ids: list[int], role_ids: list[int]) -> None:
"""
批量设置用户角色
参数:
- user_ids (list[int]): 用户ID列表
- role_ids (list[int]): 角色ID列表
返回:
- None
"""
user_objs = await self.list(search={"id": ("in", user_ids)})
if role_ids:
role_objs = await RoleCRUD(self.auth).list(search={"id": ("in", role_ids)})
else:
role_objs = []
for obj in user_objs:
relationship = obj.roles
relationship.clear()
relationship.extend(role_objs)
await self.auth.db.flush()
async def set_user_positions(self, user_ids: list[int], position_ids: list[int]) -> None:
"""
批量设置用户岗位
参数:
- user_ids (list[int]): 用户ID列表
- position_ids (list[int]): 岗位ID列表
返回:
- None
"""
user_objs = await self.list(search={"id": ("in", user_ids)})
if position_ids:
position_objs = await PositionCRUD(self.auth).list(
search={"id": ("in", position_ids)}
)
else:
position_objs = []
for obj in user_objs:
relationship = obj.positions
relationship.clear()
relationship.extend(position_objs)
await self.auth.db.flush()
async def change_password(self, id: int, password_hash: str) -> UserModel:
"""
修改用户密码
参数:
- id (int): 用户ID
- password_hash (str): 密码哈希值
返回:
- UserModel: 更新后的用户信息
"""
return await self.update(id=id, data=UserUpdateSchema(password=password_hash))
async def forget_password(self, id: int, password_hash: str) -> UserModel:
"""重置密码(与 change_password 逻辑相同)"""
return await self.change_password(id=id, password_hash=password_hash)