mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-01 00:05:17 +00:00
zhangtao
7d2367e34e
本次提交包含多项优化： 1. 移除大量冗余的文件头注释与过时的from __future__导入 2. 将CRUD的list方法统一重命名为get_list保持接口一致 3. 修复前后端状态字段类型不匹配问题，将string类型status改为number 4. 修正前端文案错别字，将"代办事项"修正为标准写法 5. 更新sqlalchemy版本并调整依赖配置 6. 新增缓存工具类替代fastapi-cache2，重构缓存调用逻辑 7. 新增开源授权函生成相关工具与数据库字段支持 8. 为多个业务模块添加防重复提交loading状态 9. 修复邮件模型的外键关联缺失问题 10. 优化pdf生成工具的导入时机与文档注释
216 lines
8.1 KiB
Python
216 lines
8.1 KiB
Python
from app.core.base_schema import AuthSchema, BatchSetAvailable
|
|
from app.core.exceptions import CustomException
|
|
from app.core.logger import logger
|
|
from app.utils.excel_util import ExcelUtil
|
|
|
|
from .crud import NoticeCRUD
|
|
from .model import NoticeModel, NoticeReadModel
|
|
from .schema import (
|
|
NoticeCreateSchema,
|
|
NoticeOutSchema,
|
|
NoticeQueryParam,
|
|
NoticeUpdateSchema,
|
|
PanelDataOut,
|
|
PanelMessageItem,
|
|
)
|
|
|
|
|
|
class NoticeService:
|
|
"""
|
|
公告管理服务
|
|
|
|
提供公告 CRUD、状态切换、已启用公告分页查询、消息面板、Excel 导出等业务能力。
|
|
"""
|
|
|
|
def __init__(self, auth: AuthSchema) -> None:
|
|
self.auth = auth
|
|
|
|
async def detail(self, id: int) -> NoticeOutSchema:
|
|
return await NoticeCRUD(self.auth).get_or_404(id=id, out_schema=NoticeOutSchema)
|
|
|
|
async def get_list(
|
|
self,
|
|
search: NoticeQueryParam | None = None,
|
|
order_by: list[dict] | None = None,
|
|
) -> list[NoticeOutSchema]:
|
|
notice_obj_list = await NoticeCRUD(self.auth).get_list(search=vars(search) if search else None, order_by=order_by)
|
|
return [NoticeOutSchema.model_validate(notice_obj) for notice_obj in notice_obj_list]
|
|
|
|
async def page(
|
|
self,
|
|
page_no: int,
|
|
page_size: int,
|
|
search: NoticeQueryParam | None = None,
|
|
order_by: list[dict] | None = None,
|
|
) -> dict:
|
|
offset = (page_no - 1) * page_size
|
|
return await NoticeCRUD(self.auth).page(
|
|
offset=offset,
|
|
limit=page_size,
|
|
order_by=order_by or [{"id": "asc"}],
|
|
search=vars(search) if search else None,
|
|
out_schema=NoticeOutSchema,
|
|
)
|
|
|
|
async def available_page(self) -> dict:
|
|
return await NoticeCRUD(self.auth).page(
|
|
offset=0,
|
|
limit=10,
|
|
order_by=[{"id": "asc"}],
|
|
search={"status": 0},
|
|
out_schema=NoticeOutSchema,
|
|
)
|
|
|
|
async def create(self, data: NoticeCreateSchema) -> NoticeOutSchema:
|
|
notice = await NoticeCRUD(self.auth).get(notice_title=data.notice_title)
|
|
if notice:
|
|
raise CustomException(msg="创建失败，该数据已存在")
|
|
notice_obj = await NoticeCRUD(self.auth).create(data=data)
|
|
return NoticeOutSchema.model_validate(notice_obj)
|
|
|
|
async def update(self, id: int, data: NoticeUpdateSchema) -> NoticeOutSchema:
|
|
_ = await NoticeCRUD(self.auth).get_or_404(id=id, msg="更新失败，该数据不存在")
|
|
exist_notice = await NoticeCRUD(self.auth).get(notice_title=data.notice_title)
|
|
if exist_notice and exist_notice.id != id:
|
|
raise CustomException(msg="更新失败，标题已存在")
|
|
notice_obj = await NoticeCRUD(self.auth).update(id=id, data=data)
|
|
return NoticeOutSchema.model_validate(notice_obj)
|
|
|
|
async def delete(self, ids: list[int]) -> None:
|
|
if len(ids) < 1:
|
|
raise CustomException(msg="删除失败，删除对象不能为空")
|
|
notices = await NoticeCRUD(self.auth).get_list(search={"id": ("in", ids)})
|
|
notice_map = {n.id: n for n in notices}
|
|
for nid in ids:
|
|
if nid not in notice_map:
|
|
raise CustomException(msg="删除失败，该数据不存在")
|
|
await NoticeCRUD(self.auth).delete(ids=ids)
|
|
|
|
async def set_available(self, data: BatchSetAvailable) -> None:
|
|
await NoticeCRUD(self.auth).set(ids=data.ids, status=data.status)
|
|
|
|
@staticmethod
|
|
def export(notice_list: list[dict]) -> bytes:
|
|
mapping_dict = {
|
|
"id": "编号",
|
|
"notice_title": "公告标题",
|
|
"notice_type": "公告类型（1通知 2公告）",
|
|
"notice_content": "公告内容",
|
|
"status": "状态",
|
|
"description": "备注",
|
|
"created_time": "创建时间",
|
|
"updated_time": "更新时间",
|
|
"created_id": "创建者ID",
|
|
"updated_id": "更新者ID",
|
|
}
|
|
data = notice_list.copy()
|
|
for item in data:
|
|
item["status"] = "启用" if item.get("status") == 0 else "停用"
|
|
item["notice_type"] = "通知" if item.get("notice_type") == "1" else "公告"
|
|
return ExcelUtil.export_list2excel(list_data=data, mapping_dict=mapping_dict)
|
|
|
|
async def latest(self, limit: int = 5) -> list[NoticeOutSchema]:
|
|
from sqlalchemy import desc, select
|
|
|
|
stmt = select(NoticeModel).where(NoticeModel.status == 0).order_by(desc(NoticeModel.created_time)).limit(limit)
|
|
result = await self.auth.db.execute(stmt)
|
|
notices = result.scalars().all()
|
|
return [NoticeOutSchema.model_validate(n) for n in notices]
|
|
|
|
async def mark_read(self, notice_id: int) -> None:
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import select
|
|
|
|
notice = await NoticeCRUD(self.auth).get(id=notice_id)
|
|
if not notice:
|
|
raise CustomException(msg="该公告不存在")
|
|
|
|
exist_stmt = select(NoticeReadModel).where(
|
|
NoticeReadModel.user_id == self.auth.user.id,
|
|
NoticeReadModel.notice_id == notice_id,
|
|
)
|
|
result = await self.auth.db.execute(exist_stmt)
|
|
if result.scalar_one_or_none():
|
|
return
|
|
|
|
read_record = NoticeReadModel(
|
|
user_id=self.auth.user.id,
|
|
notice_id=notice_id,
|
|
read_time=datetime.now(),
|
|
)
|
|
self.auth.db.add(read_record)
|
|
await self.auth.db.flush()
|
|
|
|
async def mark_all_read(self) -> int:
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import insert, select
|
|
|
|
read_ids_stmt = select(NoticeReadModel.notice_id).where(NoticeReadModel.user_id == self.auth.user.id)
|
|
read_ids_result = await self.auth.db.execute(read_ids_stmt)
|
|
read_ids = {row[0] for row in read_ids_result.fetchall()}
|
|
|
|
notices_stmt = select(NoticeModel.id).where(NoticeModel.status == 0)
|
|
notices_result = await self.auth.db.execute(notices_stmt)
|
|
all_ids = {row[0] for row in notices_result.fetchall()}
|
|
|
|
unread_ids = all_ids - read_ids
|
|
if not unread_ids:
|
|
return 0
|
|
|
|
now = datetime.now()
|
|
if unread_ids:
|
|
await self.auth.db.execute(
|
|
insert(NoticeReadModel),
|
|
[{"user_id": self.auth.user.id, "notice_id": nid, "read_time": now} for nid in unread_ids],
|
|
)
|
|
await self.auth.db.flush()
|
|
return len(unread_ids)
|
|
|
|
async def get_unread_count(self) -> int:
|
|
from sqlalchemy import func, select
|
|
|
|
total_stmt = select(func.count()).select_from(NoticeModel).where(NoticeModel.status == 0)
|
|
total_result = await self.auth.db.execute(total_stmt)
|
|
total_count = total_result.scalar() or 0
|
|
|
|
read_stmt = select(func.count()).select_from(NoticeReadModel).where(NoticeReadModel.user_id == self.auth.user.id)
|
|
read_result = await self.auth.db.execute(read_stmt)
|
|
read_count = read_result.scalar() or 0
|
|
|
|
return max(0, total_count - read_count)
|
|
|
|
async def panel_data(self) -> PanelDataOut:
|
|
from sqlalchemy import desc, select
|
|
|
|
notices = await self.latest(limit=5)
|
|
|
|
messages = []
|
|
try:
|
|
from app.api.v1.module_system.log.model import OperationLogModel
|
|
|
|
stmt = select(OperationLogModel).order_by(desc(OperationLogModel.created_time)).limit(5)
|
|
result = await self.auth.db.execute(stmt)
|
|
logs = result.scalars().all()
|
|
for log_entry in logs:
|
|
messages.append(
|
|
PanelMessageItem(
|
|
id=log_entry.id,
|
|
title=log_entry.request_path or "系统操作",
|
|
content=f"{log_entry.request_method} {log_entry.request_path}",
|
|
time=log_entry.created_time.strftime("%Y-%m-%d %H:%M") if log_entry.created_time else "",
|
|
type="system",
|
|
)
|
|
)
|
|
except Exception as e:
|
|
logger.warning(f"获取面板消息数据失败（操作日志表可能不存在），已跳过: {e}")
|
|
|
|
pendings: list[dict] = []
|
|
|
|
return PanelDataOut(
|
|
notices=notices,
|
|
messages=messages,
|
|
pendings=pendings,
|
|
)
|