mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-22 21:06:32 +00:00
zhangtao
4df6241e8b
refactor(backend): 重构CRUD基类查询方法去除重复结果 feat(application): 新增应用系统模块基础结构 fix(backend): 修复多处日志记录ID变量名错误 style(backend): 清理无用代码和注释 fix(frontend): 修复用户导入功能参数错误 docs(backend): 更新环境配置示例注释 perf(backend): 优化导出功能数据处理逻辑 test(frontend): 添加任务日志功能开发中提示 chore: 更新角色菜单权限数据
446 lines
19 KiB
Python
446 lines
19 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import io
|
|
from typing import Any, Dict, List
|
|
from fastapi import UploadFile
|
|
import pandas as pd
|
|
|
|
from app.core.exceptions import CustomException
|
|
from app.core.hash_bcrpy import PwdUtil
|
|
from app.core.base_schema import BatchSetAvailable, UploadResponseSchema
|
|
from app.core.logger import logger
|
|
from app.utils.excel_util import ExcelUtil
|
|
from app.utils.upload_util import UploadUtil
|
|
from ..position.crud import PositionCRUD
|
|
from ..role.crud import RoleCRUD
|
|
from ..menu.crud import MenuCRUD
|
|
from ..dept.crud import DeptCRUD
|
|
from ..auth.schema import AuthSchema
|
|
from ..menu.schema import MenuOutSchema
|
|
from .param import UserQueryParams
|
|
from .crud import UserCRUD
|
|
from .schema import (
|
|
CurrentUserUpdateSchema,
|
|
ResetPasswordSchema,
|
|
UserOutSchema,
|
|
UserCreateSchema,
|
|
UserUpdateSchema,
|
|
UserChangePasswordSchema,
|
|
UserRegisterSchema,
|
|
UserForgetPasswordSchema
|
|
)
|
|
|
|
|
|
|
|
class UserService:
|
|
"""用户模块服务层"""
|
|
|
|
@classmethod
|
|
async def get_detail_by_id_service(cls, auth: AuthSchema, id: int) -> Dict:
|
|
"""获取用户详情"""
|
|
user = await UserCRUD(auth).get_by_id_crud(id=id)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
|
|
# 如果用户绑定了部门,则获取部门名称
|
|
if user.dept_id:
|
|
dept = await DeptCRUD(auth).get_by_id_crud(id=user.dept_id)
|
|
user.dept_name = dept.name if dept else None
|
|
else:
|
|
user.dept_name = None
|
|
|
|
return UserOutSchema.model_validate(user).model_dump()
|
|
|
|
@classmethod
|
|
async def get_user_list_service(cls, auth: AuthSchema, search: UserQueryParams, order_by: List[Dict]= None) -> List[Dict]:
|
|
if order_by:
|
|
order_by = eval(order_by)
|
|
user_list = await UserCRUD(auth).get_list_crud(search=search.__dict__, order_by=order_by)
|
|
user_dict_list = []
|
|
for user in user_list:
|
|
if user.dept_id:
|
|
dept = await DeptCRUD(auth).get_by_id_crud(id=user.dept_id)
|
|
user.dept_name = dept.name if dept else None
|
|
else:
|
|
user.dept_name = None
|
|
user_dict = UserOutSchema.model_validate(user).model_dump()
|
|
user_dict_list.append(user_dict)
|
|
|
|
return user_dict_list
|
|
|
|
@classmethod
|
|
async def create_user_service(cls, data: UserCreateSchema, auth: AuthSchema) -> Dict:
|
|
# 检查用户名是否存在
|
|
user = await UserCRUD(auth).get_by_username_crud(username=data.username)
|
|
if user:
|
|
raise CustomException(msg='已存在相同用户名称的账号')
|
|
|
|
# 检查部门是否存在
|
|
if data.dept_id:
|
|
dept = await DeptCRUD(auth).get_by_id_crud(id=data.dept_id)
|
|
if not dept:
|
|
raise CustomException(msg='部门不存在')
|
|
|
|
# 创建用户
|
|
data.password = PwdUtil.set_password_hash(password=data.password)
|
|
user_dict = data.model_dump(exclude_unset=True, exclude={"role_ids", "position_ids"})
|
|
new_user = await UserCRUD(auth).create(data=user_dict)
|
|
|
|
# 设置角色和岗位
|
|
if data.role_ids and len(data.role_ids) > 0:
|
|
await UserCRUD(auth).set_user_roles_crud(user_ids=[new_user.id], role_ids=data.role_ids)
|
|
if data.position_ids and len(data.position_ids) > 0:
|
|
await UserCRUD(auth).set_user_positions_crud(user_ids=[new_user.id], position_ids=data.position_ids)
|
|
|
|
new_user_dict = UserOutSchema.model_validate(new_user).model_dump()
|
|
return new_user_dict
|
|
|
|
@classmethod
|
|
async def update_user_service(cls, id: int, data: UserUpdateSchema, auth: AuthSchema) -> Dict:
|
|
# 检查用户是否存在
|
|
user = await UserCRUD(auth).get_by_id_crud(id=id)
|
|
if not user:
|
|
raise CustomException(msg='用户不存在')
|
|
|
|
# 检查用户名是否重复
|
|
exist_user = await UserCRUD(auth).get_by_username_crud(username=data.username)
|
|
if exist_user and exist_user.id != id:
|
|
raise CustomException(msg='已存在相同的用户名')
|
|
|
|
# 检查部门是否存在且可用
|
|
if data.dept_id:
|
|
dept = await DeptCRUD(auth).get_by_id_crud(id=data.dept_id)
|
|
if not dept:
|
|
raise CustomException(msg='部门不存在')
|
|
if not dept.status:
|
|
raise CustomException(msg='部门已被禁用')
|
|
|
|
# 更新密码
|
|
if data.password:
|
|
data.password = PwdUtil.set_password_hash(password=data.password)
|
|
|
|
# 更新用户
|
|
user_dict = data.model_dump(exclude_unset=True, exclude={"role_ids", "position_ids"})
|
|
new_user = await UserCRUD(auth).update(id=id, data=user_dict)
|
|
|
|
# 更新角色和岗位
|
|
if data.role_ids and len(data.role_ids) > 0:
|
|
# 检查角色是否都存在且可用
|
|
roles = await RoleCRUD(auth).get_list_crud(search={"id": ("in", data.role_ids)})
|
|
if len(roles) != len(data.role_ids):
|
|
raise CustomException(msg='部分角色不存在')
|
|
if not all(role.status for role in roles):
|
|
raise CustomException(msg='部分角色已被禁用')
|
|
await UserCRUD(auth).set_user_roles_crud(user_ids=[id], role_ids=data.role_ids)
|
|
|
|
if data.position_ids and len(data.position_ids) > 0:
|
|
# 检查岗位是否都存在且可用
|
|
positions = await PositionCRUD(auth).get_list_crud(search={"id": ("in", data.position_ids)})
|
|
if len(positions) != len(data.position_ids):
|
|
raise CustomException(msg='部分岗位不存在')
|
|
if not all(position.status for position in positions):
|
|
raise CustomException(msg='部分岗位已被禁用')
|
|
await UserCRUD(auth).set_user_positions_crud(user_ids=[id], position_ids=data.position_ids)
|
|
|
|
user_dict = UserOutSchema.model_validate(new_user).model_dump()
|
|
return user_dict
|
|
|
|
@classmethod
|
|
async def delete_user_service(cls, auth: AuthSchema, ids: list[int]) -> None:
|
|
"""删除用户"""
|
|
if len(ids) < 1:
|
|
raise CustomException(msg='删除失败，删除对象不能为空')
|
|
for id in ids:
|
|
user = await UserCRUD(auth).get_by_id_crud(id=id)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
if user.is_superuser:
|
|
raise CustomException(msg="超级管理员不能删除")
|
|
if user.status:
|
|
raise CustomException(msg="用户已启用,不能删除")
|
|
if auth.user.id == id:
|
|
raise CustomException(msg="不能删除当前登陆用户")
|
|
# 删除用户角色关联数据
|
|
await UserCRUD(auth).set_user_roles_crud(user_ids=ids, role_ids=[])
|
|
|
|
# 删除用户岗位关联数据
|
|
await UserCRUD(auth).set_user_positions_crud(user_ids=ids, position_ids=[])
|
|
|
|
# 删除用户
|
|
await UserCRUD(auth).delete(ids=ids)
|
|
|
|
@classmethod
|
|
async def get_current_user_info_service(cls, auth: AuthSchema) -> Dict:
|
|
"""获取当前用户信息"""
|
|
# 获取用户基本信息
|
|
user = await UserCRUD(auth).get_by_id_crud(id=auth.user.id)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
# 获取部门名称
|
|
if user.dept_id:
|
|
dept = await DeptCRUD(auth).get_by_id_crud(id=auth.user.dept_id)
|
|
user.dept_name = dept.name if dept else None
|
|
user_dict = UserOutSchema.model_validate(user).model_dump()
|
|
|
|
# 获取菜单权限
|
|
if auth.user.is_superuser:
|
|
menu_all = await MenuCRUD(auth).get_list_crud(search={'type': ('in', [1, 2, 4]), 'status': True})
|
|
menus = [MenuOutSchema.model_validate(menu).model_dump() for menu in menu_all]
|
|
else:
|
|
menus = [
|
|
MenuOutSchema.model_validate(menu).model_dump()
|
|
for role in auth.user.roles
|
|
for menu in role.menus
|
|
if menu.status and menu.type in [1, 2, 4]
|
|
]
|
|
user_dict["menus"] = menus
|
|
return user_dict
|
|
|
|
@classmethod
|
|
async def update_current_user_info_service(cls, auth: AuthSchema, data: CurrentUserUpdateSchema) -> Dict:
|
|
"""更新当前用户信息"""
|
|
user = await UserCRUD(auth).get_by_id_crud(id=auth.user.id)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
new_user = await UserCRUD(auth).update(id=auth.user.id, data=data)
|
|
return UserOutSchema.model_validate(new_user).model_dump()
|
|
|
|
@classmethod
|
|
async def set_user_available_service(cls, auth: AuthSchema, data: BatchSetAvailable) -> None:
|
|
"""设置用户状态"""
|
|
for id in data.ids:
|
|
user = await UserCRUD(auth).get_by_id_crud(id=id)
|
|
if not user:
|
|
raise CustomException(msg=f"用户ID {id} 不存在")
|
|
if user.is_superuser:
|
|
raise CustomException(msg="超级管理员状态不能修改")
|
|
await UserCRUD(auth).set_available_crud(ids=data.ids, status=data.status)
|
|
|
|
@classmethod
|
|
async def upload_avatar_service(cls, base_url: str, file: UploadFile) -> Dict:
|
|
"""上传头像"""
|
|
if not file:
|
|
raise CustomException(msg="请选择要上传的文件")
|
|
filename, filepath, file_url = await UploadUtil.upload_file(file=file, base_url=base_url)
|
|
|
|
return UploadResponseSchema(
|
|
file_path=f'{filepath}',
|
|
file_name=filename,
|
|
origin_name=file.filename,
|
|
file_url=f'{file_url}',
|
|
).model_dump()
|
|
|
|
@classmethod
|
|
async def change_user_password_service(cls, auth: AuthSchema, data: UserChangePasswordSchema) -> Dict:
|
|
"""修改用户密码"""
|
|
if not data.old_password or not data.new_password:
|
|
raise CustomException(msg='密码不能为空')
|
|
|
|
# 验证原密码
|
|
user = await UserCRUD(auth).get_by_id_crud(id=auth.user.id)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
if not PwdUtil.verify_password(plain_password=data.old_password, password_hash=user.password):
|
|
raise CustomException(msg='原密码输入错误')
|
|
|
|
# 更新密码
|
|
new_password_hash = PwdUtil.set_password_hash(password=data.new_password)
|
|
new_user = await UserCRUD(auth).change_password_crud(id=user.id, password_hash=new_password_hash)
|
|
return UserOutSchema.model_validate(new_user).model_dump()
|
|
|
|
@classmethod
|
|
async def reset_user_password_service(cls, auth: AuthSchema, data: ResetPasswordSchema) -> Dict:
|
|
"""修改用户密码"""
|
|
if not data.password:
|
|
raise CustomException(msg='密码不能为空')
|
|
|
|
# 验证用户
|
|
user = await UserCRUD(auth).get_by_id_crud(id=data.id)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
|
|
# 更新密码
|
|
new_password_hash = PwdUtil.set_password_hash(password=data.password)
|
|
new_user = await UserCRUD(auth).change_password_crud(id=data.id, password_hash=new_password_hash)
|
|
return UserOutSchema.model_validate(new_user).model_dump()
|
|
|
|
@classmethod
|
|
async def register_user_service(cls, auth: AuthSchema, data: UserRegisterSchema) -> Dict:
|
|
"""用户注册"""
|
|
# 检查用户名是否存在
|
|
username_ok = await UserCRUD(auth).get_by_username_crud(username=data.username)
|
|
if username_ok:
|
|
raise CustomException(msg='账号已存在')
|
|
|
|
data.password = PwdUtil.set_password_hash(password=data.password)
|
|
data.name = data.username
|
|
data.creator_id = 1
|
|
dict_data = data.model_dump(exclude_unset=True)
|
|
# dict_data['creator_id'] = data.creator_id
|
|
# dict_data['dept_id'] = data.dept_id
|
|
# dict_data['description'] = data.description
|
|
result = await UserCRUD(auth).create(data=dict_data)
|
|
await UserCRUD(auth).set_user_roles_crud(user_ids=[result.id], role_ids=data.role_ids)
|
|
# await UserCRUD(auth).set_user_positions_crud(user_ids=[result.id], position_ids=data.position_ids)
|
|
return UserOutSchema.model_validate(result).model_dump()
|
|
|
|
@classmethod
|
|
async def forget_password_service(cls, auth: AuthSchema, data: UserForgetPasswordSchema) -> Dict:
|
|
"""用户忘记密码"""
|
|
user = await UserCRUD(auth).get_by_username_crud(username=data.username)
|
|
if not user:
|
|
raise CustomException(msg="用户不存在")
|
|
if not user.status:
|
|
raise CustomException(msg="用户已停用")
|
|
new_password_hash = PwdUtil.set_password_hash(password=data.new_password)
|
|
new_user = await UserCRUD(auth).forget_password_crud(id=user.id, password_hash=new_password_hash)
|
|
return UserOutSchema.model_validate(new_user).model_dump()
|
|
|
|
@classmethod
|
|
async def batch_import_user_service(cls, auth: AuthSchema, file: UploadFile, update_support: bool = False) -> str:
|
|
"""批量导入用户"""
|
|
|
|
header_dict = {
|
|
'部门编号': 'dept_id',
|
|
'用户名': 'username',
|
|
'名称': 'name',
|
|
'邮箱': 'email',
|
|
'手机号': 'mobile',
|
|
'性别': 'gender',
|
|
'状态': 'status'
|
|
}
|
|
|
|
try:
|
|
# 读取Excel文件
|
|
contents = await file.read()
|
|
df = pd.read_excel(io.BytesIO(contents))
|
|
await file.close()
|
|
|
|
if df.empty:
|
|
raise CustomException(msg="导入文件为空")
|
|
|
|
# 检查表头是否完整
|
|
missing_headers = [header for header in header_dict.keys() if header not in df.columns]
|
|
if missing_headers:
|
|
raise CustomException(msg=f"导入文件缺少必要的列: {', '.join(missing_headers)}")
|
|
|
|
# 重命名列名
|
|
df.rename(columns=header_dict, inplace=True)
|
|
|
|
# 验证必填字段
|
|
required_fields = ['username', 'name', 'dept_id']
|
|
for field in required_fields:
|
|
if df[field].isnull().any():
|
|
missing_rows = df[df[field].isnull()].index.tolist()
|
|
raise CustomException(msg=f"{[k for k,v in header_dict.items() if v == field][0]}不能为空，第{[i+1 for i in missing_rows]}行")
|
|
|
|
error_msgs = []
|
|
success_count = 0
|
|
|
|
# 处理每一行数据
|
|
for index, row in df.iterrows():
|
|
try:
|
|
# 数据转换前的类型检查
|
|
try:
|
|
dept_id = int(row['dept_id'])
|
|
except ValueError:
|
|
error_msgs.append(f"第{index+1}行: 部门编号必须是数字")
|
|
continue
|
|
|
|
# 检查部门是否存在
|
|
dept = await DeptCRUD(auth).get_by_id_crud(id=dept_id)
|
|
if not dept:
|
|
error_msgs.append(f"第{index+1}行: 部门ID {dept_id} 不存在")
|
|
continue
|
|
|
|
# 数据转换
|
|
gender = 1 if row['gender'] == '男' else (2 if row['gender'] == '女' else 1)
|
|
status = True if row['status'] == '正常' else False
|
|
|
|
# 构建用户数据
|
|
user_data = {
|
|
"username": str(row['username']).strip(),
|
|
"name": str(row['name']).strip(),
|
|
"email": str(row['email']).strip() if not pd.isna(row['email']) else None,
|
|
"mobile": str(row['mobile']).strip() if not pd.isna(row['mobile']) else None,
|
|
"gender": gender,
|
|
"status": status,
|
|
"dept_id": dept_id,
|
|
"password": PwdUtil.set_password_hash(password="123456") # 设置默认密码
|
|
}
|
|
|
|
# 处理用户导入
|
|
exists_user = await UserCRUD(auth).get_by_username_crud(username=user_data["username"])
|
|
if exists_user:
|
|
if update_support:
|
|
await UserCRUD(auth).update(id=exists_user.id, data=user_data)
|
|
success_count += 1
|
|
else:
|
|
error_msgs.append(f"第{index+1}行: 用户 {user_data['username']} 已存在")
|
|
else:
|
|
await UserCRUD(auth).create(data=user_data)
|
|
success_count += 1
|
|
|
|
except Exception as e:
|
|
error_msgs.append(f"第{index+1}行: {str(e)}")
|
|
continue
|
|
|
|
# 返回详细的导入结果
|
|
result = f"成功导入 {success_count} 条数据"
|
|
if error_msgs:
|
|
result += "\n错误信息:\n" + "\n".join(error_msgs)
|
|
return result
|
|
|
|
except Exception as e:
|
|
logger.error(f"批量导入用户失败: {str(e)}")
|
|
raise CustomException(msg=f"导入失败: {str(e)}")
|
|
|
|
@classmethod
|
|
async def get_import_template_user_service(cls) -> bytes:
|
|
"""获取用户导入模板"""
|
|
header_list = ['部门编号', '用户名', '名称', '邮箱', '手机号', '性别', '状态']
|
|
selector_header_list = ['性别', '状态']
|
|
option_list = [{'性别': ['男', '女', '未知']}, {'状态': ['正常', '停用']}]
|
|
return ExcelUtil.get_excel_template(
|
|
header_list=header_list,
|
|
selector_header_list=selector_header_list,
|
|
option_list=option_list
|
|
)
|
|
|
|
@classmethod
|
|
async def export_user_list_service(cls, user_list: List[Dict[str, Any]]) -> bytes:
|
|
"""导出用户列表"""
|
|
if not user_list:
|
|
raise CustomException(msg="没有数据可导出")
|
|
|
|
# 定义字段映射
|
|
mapping_dict = {
|
|
'id': '用户编号',
|
|
'avatar': '头像',
|
|
'username': '用户名称',
|
|
'name': '用户昵称',
|
|
'dept_name': '部门',
|
|
'email': '邮箱',
|
|
'mobile': '手机号',
|
|
'gender': '性别',
|
|
'status': '状态',
|
|
'is_superuser': '是否超级管理员',
|
|
'last_login': '最后登录时间',
|
|
'description': '备注',
|
|
'created_at': '创建时间',
|
|
'updated_at': '更新时间',
|
|
'creator': '创建者',
|
|
}
|
|
|
|
# 复制数据并转换
|
|
# creator = {'id': 1, 'name': '管理员', 'username': 'admin'}
|
|
data = user_list.copy()
|
|
for item in data:
|
|
item['status'] = '启用' if item.get('status') else '停用'
|
|
gender = item.get('gender')
|
|
item['gender'] = '男' if gender == '1' else ('女' if gender == '2' else '未知')
|
|
item['is_superuser'] = '是' if item.get('is_superuser') else '否'
|
|
item['creator'] = item.get('creator', {}).get('name', '未知') if isinstance(item.get('creator'), dict) else '未知'
|
|
|
|
return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)
|