Files
fastapi-best-architecture/backend/app/task/service/result_service.py
T
Wu Clan 15a9527866 Refactor the service layer db session call method (#850)
* Refactor the service layer db session call method

* Fix lint

* Update paging interfaces

* Fix some errors
2025-10-16 12:49:56 +08:00

56 lines
1.5 KiB
Python

from typing import Any
from sqlalchemy.ext.asyncio import AsyncSession
from backend.app.task.crud.crud_result import task_result_dao
from backend.app.task.model import TaskResult
from backend.app.task.schema.result import DeleteTaskResultParam
from backend.common.exception import errors
from backend.common.pagination import paging_data
class TaskResultService:
@staticmethod
async def get(*, db: AsyncSession, pk: int) -> TaskResult:
"""
获取任务结果详情
:param db: 数据库会话
:param pk: 任务 ID
:return:
"""
result = await task_result_dao.get(db, pk)
if not result:
raise errors.NotFoundError(msg='任务结果不存在')
return result
@staticmethod
async def get_list(*, db: AsyncSession, name: str | None, task_id: str | None) -> dict[str, Any]:
"""
获取任务结果列表
:param db: 数据库会话
:param name: 任务名称
:param task_id: 任务 ID
:return:
"""
result_select = await task_result_dao.get_select(name, task_id)
return await paging_data(db, result_select)
@staticmethod
async def delete(*, db: AsyncSession, obj: DeleteTaskResultParam) -> int:
"""
批量删除任务结果
:param db: 数据库会话
:param obj: 任务结果 ID 列表
:return:
"""
count = await task_result_dao.delete(db, obj.pks)
return count
task_result_service: TaskResultService = TaskResultService()