mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-22 05:02:57 +00:00
zhangtao
3f7d5aa4b9
本次提交包含多项改进： 1. 国际化补充：新增通知空状态文案中英文支持 2. 启动优化：重构banner输出、移除环境参数依赖，简化启动日志 3. 配置清理：移除冗余的REDIS_ENABLE/SQL_DB_ENABLE配置项，同步更新env示例与测试配置 4. 接口分页改造：将全局分页参数从Query改为Depends自动解析，统一排序逻辑 5. 菜单权限优化：调整菜单可见范围默认值、修复平台菜单路由路径错误 6. 异常处理增强：完善数据库异常捕获逻辑，新增连接失败专项处理 7. API令牌重构：重命名API令牌路由与权限标识，拆分前端API文件 8. 定时任务整合：将系统任务注册移入调度器初始化逻辑 9. 数据库连接优化：新增连接检查，简化建表/删表逻辑 10. 控制台美化：重构启动控制台面板，优化信息展示格式 11. 租户/角色服务优化：移除不必要的超级管理员装饰器，统一分页排序逻辑 12. 用户菜单适配：修复超级用户菜单过滤逻辑，移除scope强制校验
534 lines
23 KiB
Python
534 lines
23 KiB
Python
from typing import Any
|
||
|
||
from fastapi import UploadFile
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.api.v1.module_platform.menu.crud import MenuCRUD
|
||
from app.api.v1.module_platform.menu.schema import MenuOutSchema, MenuTreeOutSchema
|
||
from app.api.v1.module_platform.package.service import PackageService
|
||
from app.api.v1.module_platform.tenant.model import TenantModel
|
||
from app.api.v1.module_platform.tenant.service import TenantService
|
||
from app.api.v1.module_system.dept.crud import DeptCRUD
|
||
from app.api.v1.module_system.position.crud import PositionCRUD
|
||
from app.api.v1.module_system.role.crud import RoleCRUD
|
||
from app.core.base_schema import AuthSchema, BatchSetAvailable, CommonSchema, PageResultSchema
|
||
from app.core.exceptions import CustomException
|
||
from app.core.logger import logger
|
||
from app.utils.common_util import search_to_dict, traversal_to_tree
|
||
from app.utils.excel_util import ExcelUtil
|
||
from app.utils.password_util import PwdUtil
|
||
|
||
from .crud import UserCRUD
|
||
from .schema import (
|
||
CurrentUserUpdateSchema,
|
||
ResetPasswordSchema,
|
||
UserChangePasswordSchema,
|
||
UserCreateSchema,
|
||
UserForgetPasswordSchema,
|
||
UserOutSchema,
|
||
UserQueryParam,
|
||
UserUpdateSchema,
|
||
)
|
||
|
||
|
||
class UserService:
|
||
"""用户管理服务"""
|
||
|
||
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
|
||
self.auth = auth
|
||
self.db = db
|
||
|
||
async def detail(self, id: int) -> UserOutSchema:
|
||
user = await UserCRUD(self.auth, self.db).get_or_404(id=id)
|
||
result = UserOutSchema.model_validate(user)
|
||
if user.dept_id:
|
||
dept = await DeptCRUD(self.auth, self.db).get(id=user.dept_id)
|
||
result.dept_name = dept.name if dept else None
|
||
return result
|
||
|
||
async def get_list(
|
||
self,
|
||
search: UserQueryParam | None = None,
|
||
order_by: list[dict[str, str]] | None = None,
|
||
) -> list[UserOutSchema]:
|
||
user_list = await UserCRUD(self.auth, self.db).get_list(search=search_to_dict(search), order_by=order_by)
|
||
return [UserOutSchema.model_validate(user) for user in user_list]
|
||
|
||
async def page(
|
||
self,
|
||
page_no: int,
|
||
page_size: int,
|
||
search: UserQueryParam | None = None,
|
||
order_by: list[dict[str, str]] | None = None,
|
||
) -> PageResultSchema[UserOutSchema]:
|
||
offset = (page_no - 1) * page_size
|
||
return await UserCRUD(self.auth, self.db).page(
|
||
offset=offset,
|
||
limit=page_size,
|
||
order_by=order_by or [{"id": "asc"}],
|
||
search=search_to_dict(search),
|
||
out_schema=UserOutSchema,
|
||
)
|
||
|
||
async def create(self, data: UserCreateSchema) -> UserOutSchema:
|
||
if not data.username:
|
||
raise CustomException(msg="用户名不能为空")
|
||
if data.is_superuser:
|
||
raise CustomException(msg="不允许创建超级管理员")
|
||
user = await UserCRUD(self.auth, self.db).get(username=data.username)
|
||
if user:
|
||
raise CustomException(msg="已存在相同用户名称的账号")
|
||
|
||
if data.dept_id:
|
||
dept = await DeptCRUD(self.auth, self.db).get(id=data.dept_id)
|
||
if not dept:
|
||
raise CustomException(msg="该数据不存在")
|
||
|
||
await TenantService(self.auth, self.db).check_quota(self.auth.user.tenant_id, "user")
|
||
|
||
if data.password:
|
||
data.password = PwdUtil.hash_password(password=data.password)
|
||
new_user = await UserCRUD(self.auth, self.db).create(data=data)
|
||
if data.role_ids and len(data.role_ids) > 0:
|
||
await UserCRUD(self.auth, self.db).set_user_roles(user_ids=[new_user.id], role_ids=data.role_ids)
|
||
if data.position_ids and len(data.position_ids) > 0:
|
||
await UserCRUD(self.auth, self.db).set_user_positions(user_ids=[new_user.id], position_ids=data.position_ids)
|
||
return UserOutSchema.model_validate(new_user)
|
||
|
||
async def update(self, id: int, data: UserUpdateSchema) -> UserOutSchema:
|
||
if not data.username:
|
||
raise CustomException(msg="账号不能为空")
|
||
|
||
user = await UserCRUD(self.auth, self.db).get_or_404(id=id)
|
||
if user.is_superuser:
|
||
raise CustomException(msg="超级管理员不允许修改")
|
||
|
||
await self._validate_unique_username(data.username, exclude_id=id)
|
||
await self._validate_mobile_email_unique(data, exclude_id=id)
|
||
await self._validate_dept_active(data.dept_id)
|
||
|
||
new_user = await UserCRUD(self.auth, self.db).update(id=id, data=data)
|
||
|
||
await self._assign_user_roles(id, data.role_ids)
|
||
await self._assign_user_positions(id, data.position_ids)
|
||
|
||
return UserOutSchema.model_validate(new_user)
|
||
|
||
async def _validate_unique_username(self, username: str, exclude_id: int) -> None:
|
||
"""校验用户名唯一性"""
|
||
exist_user = await UserCRUD(self.auth, self.db).get(username=username)
|
||
if exist_user and exist_user.id != exclude_id:
|
||
raise CustomException(msg="更新失败，账号已存在")
|
||
|
||
async def _validate_mobile_email_unique(self, data: UserUpdateSchema, exclude_id: int) -> None:
|
||
"""校验手机号和邮箱唯一性"""
|
||
if data.mobile:
|
||
exist_mobile_user = await UserCRUD(self.auth, self.db).get(mobile=data.mobile)
|
||
if exist_mobile_user and exist_mobile_user.id != exclude_id:
|
||
raise CustomException(msg="该数据已存在")
|
||
if data.email:
|
||
exist_email_user = await UserCRUD(self.auth, self.db).get(email=data.email)
|
||
if exist_email_user and exist_email_user.id != exclude_id:
|
||
raise CustomException(msg="该数据已存在")
|
||
|
||
async def _validate_dept_active(self, dept_id: int | None) -> None:
|
||
"""校验部门存在且已启用"""
|
||
if not dept_id:
|
||
return
|
||
dept = await DeptCRUD(self.auth, self.db).get(id=dept_id)
|
||
if not dept:
|
||
raise CustomException(msg="该数据不存在")
|
||
if dept.status == 1:
|
||
raise CustomException(msg="部门已被禁用")
|
||
|
||
async def _assign_user_roles(self, user_id: int, role_ids: list[int] | None) -> None:
|
||
"""校验并分配用户角色"""
|
||
if not role_ids or len(role_ids) < 1:
|
||
return
|
||
roles = await RoleCRUD(self.auth, self.db).get_list(search={"id": ("in", role_ids)})
|
||
if len(roles) != len(role_ids):
|
||
raise CustomException(msg="更新失败，部分角色不存在")
|
||
if not all(role.status == 0 for role in roles):
|
||
raise CustomException(msg="更新失败，部分角色已被禁用")
|
||
await UserCRUD(self.auth, self.db).set_user_roles(user_ids=[user_id], role_ids=role_ids)
|
||
|
||
async def _assign_user_positions(self, user_id: int, position_ids: list[int] | None) -> None:
|
||
"""校验并分配用户岗位"""
|
||
if not position_ids or len(position_ids) < 1:
|
||
return
|
||
positions = await PositionCRUD(self.auth, self.db).get_list(search={"id": ("in", position_ids)})
|
||
if len(positions) != len(position_ids):
|
||
raise CustomException(msg="更新失败，部分岗位不存在")
|
||
if not all(position.status == 0 for position in positions):
|
||
raise CustomException(msg="更新失败，部分岗位已被禁用")
|
||
await UserCRUD(self.auth, self.db).set_user_positions(user_ids=[user_id], position_ids=position_ids)
|
||
|
||
async def delete(self, ids: list[int]) -> None:
|
||
if not ids:
|
||
raise CustomException(msg="删除失败，删除对象不能为空")
|
||
users = await UserCRUD(self.auth, self.db).get_list(search={"id": ("in", ids)})
|
||
user_map = {u.id: u for u in users}
|
||
errors: list[str] = []
|
||
for uid in ids:
|
||
user = user_map.get(uid)
|
||
if not user:
|
||
errors.append(f"用户[{uid}]不存在")
|
||
continue
|
||
if user.is_superuser:
|
||
errors.append(f"用户[{uid}]是超级管理员，不能删除")
|
||
continue
|
||
if user.status == 0:
|
||
errors.append(f"用户[{uid}]已启用，不能删除")
|
||
continue
|
||
if self.auth.user.id == uid:
|
||
errors.append("不能删除当前登陆用户")
|
||
continue
|
||
if errors:
|
||
raise CustomException(msg="; ".join(errors))
|
||
if not user_map:
|
||
raise CustomException(msg="删除对象不存在")
|
||
|
||
await UserCRUD(self.auth, self.db).set_user_roles(user_ids=ids, role_ids=[])
|
||
await UserCRUD(self.auth, self.db).set_user_positions(user_ids=ids, position_ids=[])
|
||
await UserCRUD(self.auth, self.db).delete(ids=ids)
|
||
|
||
async def current_info(self) -> UserOutSchema:
|
||
if not self.auth.user.id:
|
||
raise CustomException(msg="该数据不存在")
|
||
user = await UserCRUD(self.auth, self.db).get(id=self.auth.user.id)
|
||
user_dict = UserOutSchema.model_validate(user)
|
||
if user and user.dept:
|
||
user_dict.dept_name = user.dept.name
|
||
if user and user.tenant_by:
|
||
user_dict.tenant_by = CommonSchema(id=user.tenant_by.id, name=user.tenant_by.name, status=user.tenant_by.status)
|
||
user_dict.is_impersonate = self.auth.is_impersonate
|
||
|
||
_pc_only = {"client": "pc"}
|
||
if self.auth.user.is_superuser:
|
||
menu_all = await MenuCRUD(self.auth, self.db).tree_list(
|
||
search={"type": ("in", [1, 2, 3, 4]), "status": 0, **_pc_only},
|
||
order_by=[{"order": "asc"}],
|
||
)
|
||
menus_raw = [MenuOutSchema.model_validate(menu) for menu in menu_all]
|
||
else:
|
||
menu_ids = set(self.auth.menu_ids)
|
||
|
||
if menu_ids and self.auth.user.tenant_id:
|
||
allowed_ids = await PackageService(self.auth, self.db).get_tenant_available_menu_ids(self.auth.user.tenant_id)
|
||
allowed_set = set(allowed_ids)
|
||
menu_ids = menu_ids & allowed_set
|
||
|
||
menus_raw = (
|
||
[
|
||
MenuOutSchema.model_validate(menu)
|
||
for menu in await MenuCRUD(self.auth, self.db).tree_list(
|
||
search={"id": ("in", list(menu_ids)), **_pc_only},
|
||
order_by=[{"order": "asc"}],
|
||
)
|
||
]
|
||
if menu_ids
|
||
else []
|
||
)
|
||
|
||
for menu in menus_raw:
|
||
menu.scope = None
|
||
menu_tree = [MenuTreeOutSchema(**item) for item in traversal_to_tree([menu.model_dump(mode="json") for menu in menus_raw])]
|
||
user_dict.menus = menu_tree
|
||
return user_dict
|
||
|
||
async def update_current_info(self, data: CurrentUserUpdateSchema) -> UserOutSchema:
|
||
if not self.auth.user.id:
|
||
raise CustomException(msg="该数据不存在")
|
||
user = await UserCRUD(self.auth, self.db).get(id=self.auth.user.id)
|
||
if not user:
|
||
raise CustomException(msg="该数据不存在")
|
||
if user.is_superuser:
|
||
raise CustomException(msg="超级管理员不能修改个人信息")
|
||
if data.mobile:
|
||
exist_mobile_user = await UserCRUD(self.auth, self.db).get(mobile=data.mobile)
|
||
if exist_mobile_user and exist_mobile_user.id != self.auth.user.id:
|
||
raise CustomException(msg="该数据已存在")
|
||
if data.email:
|
||
exist_email_user = await UserCRUD(self.auth, self.db).get(email=data.email)
|
||
if exist_email_user and exist_email_user.id != self.auth.user.id:
|
||
raise CustomException(msg="该数据已存在")
|
||
user_update_data = UserUpdateSchema(**data.model_dump())
|
||
new_user = await UserCRUD(self.auth, self.db).update(id=self.auth.user.id, data=user_update_data)
|
||
return UserOutSchema.model_validate(new_user)
|
||
|
||
async def set_available(self, data: BatchSetAvailable) -> None:
|
||
users = await UserCRUD(self.auth, self.db).get_list(search={"id": ("in", data.ids)})
|
||
for user in users:
|
||
if user.is_superuser:
|
||
raise CustomException(msg="超级管理员状态不能修改")
|
||
await UserCRUD(self.auth, self.db).set(ids=data.ids, status=data.status)
|
||
# 停用的用户立即让旧 token 失效
|
||
if data.status == 1:
|
||
for user in users:
|
||
await self._invalidate_user_sessions(user_id=user.id)
|
||
|
||
async def change_password(self, data: UserChangePasswordSchema) -> UserOutSchema:
|
||
if not self.auth.user.id:
|
||
raise CustomException(msg="该数据不存在")
|
||
if not data.old_password or not data.new_password:
|
||
raise CustomException(msg="密码不能为空")
|
||
|
||
user = await UserCRUD(self.auth, self.db).get(id=self.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.hash_password(password=data.new_password)
|
||
new_user = await UserCRUD(self.auth, self.db).change_password(id=user.id, password_hash=new_password_hash)
|
||
# 改密后立即让旧 token 失效：递增 token_version + 清掉该用户的所有 Redis session
|
||
await self._invalidate_user_sessions(user_id=user.id)
|
||
return UserOutSchema.model_validate(new_user)
|
||
|
||
async def reset_password(self, data: ResetPasswordSchema) -> UserOutSchema:
|
||
if not data.password:
|
||
raise CustomException(msg="密码不能为空")
|
||
|
||
user = await UserCRUD(self.auth, self.db).get(id=data.id)
|
||
if not user:
|
||
raise CustomException(msg="该数据不存在")
|
||
|
||
if user.is_superuser:
|
||
raise CustomException(msg="超级管理员密码不能重置")
|
||
|
||
new_password_hash = PwdUtil.hash_password(password=data.password)
|
||
new_user = await UserCRUD(self.auth, self.db).change_password(id=data.id, password_hash=new_password_hash)
|
||
# 重置密码后立即让旧 token 失效
|
||
await self._invalidate_user_sessions(user_id=user.id)
|
||
return UserOutSchema.model_validate(new_user)
|
||
|
||
async def _invalidate_user_sessions(self, user_id: int) -> None:
|
||
"""使指定用户的所有活跃 session 立即失效。
|
||
|
||
递增 ``UserModel.token_version``：JWT 中携带的旧 token_version 与 DB 不匹配 ⇒ 401。
|
||
|
||
Redis 中存储的 key 格式为 ``user_session:<session_id>``（session_id = UUID），
|
||
与 ``user_id`` 无直接映射关系，因此无法按 user_id 精确清理孤立 session 数据。
|
||
但 ``token_version`` 已确保旧 JWT 无法通过校验，安全无虞。
|
||
"""
|
||
await UserCRUD(self.auth, self.db).bump_token_version(user_id=user_id)
|
||
|
||
async def forget_password(self, data: UserForgetPasswordSchema) -> UserOutSchema:
|
||
from sqlalchemy import select
|
||
|
||
# 根据租户名称查租户
|
||
tenant_stmt = (
|
||
select(TenantModel)
|
||
.where(
|
||
TenantModel.name == data.tenant_name,
|
||
TenantModel.status == 0,
|
||
TenantModel.is_deleted.is_(False),
|
||
)
|
||
.limit(1)
|
||
)
|
||
result = await self.db.execute(tenant_stmt)
|
||
tenant = result.scalar_one_or_none()
|
||
if not tenant:
|
||
raise CustomException(msg="租户不存在")
|
||
|
||
# 在租户范围内查找用户
|
||
user = await UserCRUD(self.auth, self.db).get(username=data.username, tenant_id=tenant.id)
|
||
if not user:
|
||
raise CustomException(msg="该数据不存在")
|
||
if user.status == 1:
|
||
raise CustomException(msg="用户已停用")
|
||
|
||
if user.is_superuser:
|
||
raise CustomException(msg="超级管理员密码不能重置")
|
||
|
||
if data.mobile and user.mobile != data.mobile:
|
||
raise CustomException(msg="手机号不匹配")
|
||
|
||
new_password_hash = PwdUtil.hash_password(password=data.new_password)
|
||
new_user = await UserCRUD(self.auth, self.db).forget_password(id=user.id, password_hash=new_password_hash)
|
||
# 忘记密码后使旧 token 失效
|
||
await self._invalidate_user_sessions(user_id=user.id)
|
||
return UserOutSchema.model_validate(new_user)
|
||
|
||
async def batch_import(self, file: UploadFile, update_support: bool = False) -> str:
|
||
header_dict = {
|
||
"部门编号": "dept_id",
|
||
"账号": "username",
|
||
"昵称": "name",
|
||
"邮箱": "email",
|
||
"手机号": "mobile",
|
||
"性别": "gender",
|
||
"状态": "status",
|
||
}
|
||
|
||
try:
|
||
contents = await file.read()
|
||
rows = ExcelUtil.read_excel_to_dicts(contents)
|
||
await file.close()
|
||
|
||
if not rows:
|
||
raise CustomException(msg="导入文件为空")
|
||
|
||
missing_headers = [h for h in header_dict if h not in rows[0]]
|
||
if missing_headers:
|
||
raise CustomException(msg=f"导入文件缺少必要的列: {', '.join(missing_headers)}")
|
||
|
||
# 将中文字段名映射为英文字段
|
||
mapped_rows = []
|
||
for row in rows:
|
||
mapped_rows.append({en: row.get(ch) for ch, en in header_dict.items()})
|
||
|
||
required_fields = ["username", "name", "dept_id"]
|
||
errors = []
|
||
for field in required_fields:
|
||
missing_count = sum(1 for r in mapped_rows if r.get(field) is None)
|
||
if missing_count:
|
||
errors.append(f"字段'{field}'有{missing_count}行缺少数据")
|
||
|
||
if errors:
|
||
raise CustomException(msg="\n".join(errors))
|
||
|
||
success_count = 0
|
||
error_msgs = []
|
||
|
||
for i, row in enumerate(mapped_rows, start=2):
|
||
count_delta, err = await self._process_import_row(i, row, update_support)
|
||
if err:
|
||
error_msgs.append(err)
|
||
else:
|
||
success_count += count_delta
|
||
|
||
result = f"成功导入 {success_count} 条数据"
|
||
if error_msgs:
|
||
result += "\n错误信息:\n" + "\n".join(error_msgs)
|
||
return result
|
||
|
||
except Exception as e:
|
||
logger.error(f"批量导入用户失败: {e!s}")
|
||
raise CustomException(msg=f"导入失败: {e!s}") from e
|
||
|
||
async def _process_import_row(
|
||
self,
|
||
row_num: int,
|
||
row: dict,
|
||
update_support: bool,
|
||
) -> tuple[int, str | None]:
|
||
"""处理单行导入数据
|
||
|
||
验证字段合法性，执行创建或更新操作。
|
||
|
||
参数:
|
||
- row_num (int): Excel 行号（用于错误提示）
|
||
- row (dict): 经过字段映射后的用户数据行
|
||
- update_support (bool): 是否支持更新已存在用户
|
||
|
||
返回:
|
||
- tuple[int, str | None]: (成功计数增量, 错误信息或 None)
|
||
"""
|
||
try:
|
||
username = (str(row["username"]) if row["username"] is not None else "").strip()
|
||
name = (str(row["name"]) if row["name"] is not None else "").strip()
|
||
if not username:
|
||
return 0, f"第{row_num}行: 账号不能为空"
|
||
if not name:
|
||
return 0, f"第{row_num}行: 昵称不能为空"
|
||
|
||
dept_id = int(row["dept_id"])
|
||
dept = await DeptCRUD(self.auth, self.db).get(id=dept_id)
|
||
if not dept:
|
||
return 0, f"第{row_num}行: 部门ID {dept_id} 不存在"
|
||
if not self.auth.user.is_superuser and dept.tenant_id != self.auth.user.tenant_id:
|
||
return 0, f"第{row_num}行: 部门ID {dept_id} 不属于当前租户"
|
||
|
||
user_data = {
|
||
"username": username,
|
||
"name": name,
|
||
"email": str(row["email"]).strip() if row.get("email") is not None else None,
|
||
"mobile": str(row["mobile"]).strip() if row.get("mobile") is not None else None,
|
||
"gender": str(row["gender"]).strip() if row.get("gender") is not None else "1",
|
||
"status": 0 if str(row["status"]).strip() == "正常" else 1,
|
||
"dept_id": dept_id,
|
||
"password": PwdUtil.hash_password(password="123456"),
|
||
}
|
||
|
||
exists_user = await UserCRUD(self.auth, self.db).get(username=user_data["username"])
|
||
if exists_user:
|
||
if exists_user.is_superuser:
|
||
return 0, f"第{row_num}行: 超级管理员不允许修改"
|
||
if update_support:
|
||
user_update_data = UserUpdateSchema(**user_data)
|
||
await UserCRUD(self.auth, self.db).update(id=exists_user.id, data=user_update_data)
|
||
return 1, None
|
||
else:
|
||
return 0, f"第{row_num}行: 用户 {user_data['username']} 已存在"
|
||
else:
|
||
user_create_schema = UserCreateSchema(**user_data)
|
||
new_user = await UserCRUD(self.auth, self.db).create(data=user_create_schema)
|
||
if user_create_schema.role_ids and len(user_create_schema.role_ids) > 0:
|
||
await UserCRUD(self.auth, self.db).set_user_roles(
|
||
user_ids=[new_user.id], role_ids=user_create_schema.role_ids
|
||
)
|
||
if user_create_schema.position_ids and len(user_create_schema.position_ids) > 0:
|
||
await UserCRUD(self.auth, self.db).set_user_positions(
|
||
user_ids=[new_user.id], position_ids=user_create_schema.position_ids
|
||
)
|
||
return 1, None
|
||
|
||
except Exception as e:
|
||
return 0, f"第{row_num}行: 异常{e!s}"
|
||
|
||
@staticmethod
|
||
def get_import_template() -> 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,
|
||
)
|
||
|
||
@staticmethod
|
||
def export_list(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_time": "创建时间",
|
||
"updated_time": "更新时间",
|
||
"updated_id": "更新者ID",
|
||
}
|
||
|
||
data = user_list.copy()
|
||
for item in data:
|
||
item["status"] = "启用" if item.get("status") == 0 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("created_by", {}).get("name", "未知") if isinstance(item.get("created_by"), dict) else "未知"
|
||
|
||
return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)
|