73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
"""超期预警 模块服务层 —— 只读查询 + 标记处理；写入由每日扫描 job 直接落库。"""
|
|
|
|
from typing import Any
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.base_schema import AuthSchema, PageResultSchema
|
|
from app.core.exceptions import CustomException
|
|
|
|
from .crud import BreedingAlertCRUD
|
|
from .model import AlertModel
|
|
from .schema import AlertOutSchema, AlertQueryParam
|
|
|
|
|
|
class AlertService:
|
|
"""预警中心 —— 查询 + 处理。生成侧在 job.py，直接 db.add（不经 CRUDBase，防自审计递归）。"""
|
|
|
|
def __init__(self, auth: AuthSchema, db: AsyncSession) -> None:
|
|
self.auth = auth
|
|
self.db = db
|
|
|
|
async def page(
|
|
self,
|
|
page_no: int,
|
|
page_size: int,
|
|
search: AlertQueryParam | None = None,
|
|
order_by: list[dict[str, str]] | None = None,
|
|
) -> PageResultSchema[AlertOutSchema]:
|
|
from app.utils.common_util import search_to_dict
|
|
|
|
return await BreedingAlertCRUD(self.auth, self.db).page(
|
|
offset=(page_no - 1) * page_size,
|
|
limit=page_size,
|
|
order_by=order_by or [{"id": "desc"}],
|
|
search=search_to_dict(search, {}),
|
|
out_schema=AlertOutSchema,
|
|
)
|
|
|
|
async def get_list(self, search: AlertQueryParam | None = None) -> list[AlertOutSchema]:
|
|
from app.utils.common_util import search_to_dict
|
|
|
|
objs = await BreedingAlertCRUD(self.auth, self.db).get_list(
|
|
search=search_to_dict(search, {}), order_by=[{"id": "desc"}]
|
|
)
|
|
return [AlertOutSchema.model_validate(o) for o in objs]
|
|
|
|
async def detail(self, id: int) -> AlertOutSchema:
|
|
obj = await BreedingAlertCRUD(self.auth, self.db).get(id=id)
|
|
if not obj:
|
|
raise CustomException(msg="该预警不存在")
|
|
return AlertOutSchema.model_validate(obj)
|
|
|
|
async def resolve(self, id: int) -> AlertOutSchema:
|
|
obj = await BreedingAlertCRUD(self.auth, self.db).get(id=id)
|
|
if not obj:
|
|
raise CustomException(msg="该预警不存在")
|
|
obj.status = "resolved"
|
|
obj.updated_id = self.auth.user.id if self.auth.user.id else None
|
|
self.db.add(obj)
|
|
await self.db.flush()
|
|
return AlertOutSchema.model_validate(obj)
|
|
|
|
async def options(self) -> dict[str, Any]:
|
|
"""前端筛选下拉：预警类型 / 实体类型 去重集合。"""
|
|
alert_types = (await self.db.execute(
|
|
select(AlertModel.alert_type).distinct()
|
|
)).scalars().all()
|
|
entity_types = (await self.db.execute(
|
|
select(AlertModel.entity_type).distinct()
|
|
)).scalars().all()
|
|
return {"alert_types": sorted(alert_types), "entity_types": sorted(entity_types)}
|