csg-server/app/api/v1/module_energy/report_generation/controller.py

from typing import Annotated

from fastapi import APIRouter, Body, Depends, Path
from fastapi.responses import JSONResponse

from app.common.response import ResponseSchema, SuccessResponse
from app.core.base_params import PaginationQueryParam
from app.core.base_schema import AuthSchema, PageResultSchema
from app.core.dependencies import AuthPermission
from app.core.router_class import OperationLogRoute

from .schema import (
    EnergyReportGenerationCreateSchema,
    EnergyReportGenerationLogCreateSchema,
    EnergyReportGenerationLogOutSchema,
    EnergyReportGenerationLogQueryParam,
    EnergyReportGenerationOutSchema,
    EnergyReportGenerationPageOutSchema,
    EnergyReportGenerationQueryParam,
    EnergyReportGenerationUpdateSchema,
)
from .service import EnergyReportGenerationLogService, EnergyReportGenerationService

EnergyReportGenerationRouter = APIRouter(route_class=OperationLogRoute, prefix="/report-generation", tags=["能源报告生成"])


@EnergyReportGenerationRouter.get(
    "/list",
    summary="查询能源报告生成任务列表",
    response_model=ResponseSchema[EnergyReportGenerationPageOutSchema],
)
async def get_energy_report_generation_list_controller(
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[EnergyReportGenerationQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:query"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationService(auth).page(
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by or [{"id": "desc"}],
    )
    return SuccessResponse(data=result, msg="查询能源报告生成任务列表成功")


@EnergyReportGenerationRouter.get(
    "/detail/{id}",
    summary="查询能源报告生成任务详情",
    response_model=ResponseSchema[EnergyReportGenerationOutSchema],
)
async def get_energy_report_generation_detail_controller(
    id: Annotated[int, Path(description="能源报告生成任务ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:detail"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationService(auth).detail(id=id)
    return SuccessResponse(data=result, msg="查询能源报告生成任务详情成功")


@EnergyReportGenerationRouter.post(
    "/create",
    summary="创建能源报告生成任务",
    response_model=ResponseSchema[EnergyReportGenerationOutSchema],
)
async def create_energy_report_generation_controller(
    data: EnergyReportGenerationCreateSchema,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:create"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationService(auth).create(data=data)
    return SuccessResponse(data=result, msg="创建能源报告生成任务成功")


@EnergyReportGenerationRouter.put(
    "/update/{id}",
    summary="修改能源报告生成任务",
    response_model=ResponseSchema[EnergyReportGenerationOutSchema],
)
async def update_energy_report_generation_controller(
    data: EnergyReportGenerationUpdateSchema,
    id: Annotated[int, Path(description="能源报告生成任务ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:update"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationService(auth).update(id=id, data=data)
    return SuccessResponse(data=result, msg="修改能源报告生成任务成功")


@EnergyReportGenerationRouter.delete(
    "/delete",
    summary="删除能源报告生成任务",
    response_model=ResponseSchema[None],
)
async def delete_energy_report_generation_controller(
    ids: Annotated[list[int], Body(description="ID列表")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:delete"]))],
) -> JSONResponse:
    await EnergyReportGenerationService(auth).delete(ids=ids)
    return SuccessResponse(msg="删除能源报告生成任务成功")


@EnergyReportGenerationRouter.get(
    "/log/list",
    summary="查询能源报告生成操作记录列表",
    response_model=ResponseSchema[PageResultSchema[EnergyReportGenerationLogOutSchema]],
)
async def get_energy_report_generation_log_list_controller(
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[EnergyReportGenerationLogQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:query"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationLogService(auth).page(
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by or [{"id": "desc"}],
    )
    return SuccessResponse(data=result, msg="查询能源报告生成操作记录列表成功")


@EnergyReportGenerationRouter.get(
    "/log/detail/{id}",
    summary="查询能源报告生成操作记录详情",
    response_model=ResponseSchema[EnergyReportGenerationLogOutSchema],
)
async def get_energy_report_generation_log_detail_controller(
    id: Annotated[int, Path(description="能源报告生成操作记录ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:query"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationLogService(auth).detail(id=id)
    return SuccessResponse(data=result, msg="查询能源报告生成操作记录详情成功")


@EnergyReportGenerationRouter.post(
    "/log/create",
    summary="创建能源报告生成操作记录",
    response_model=ResponseSchema[EnergyReportGenerationLogOutSchema],
)
async def create_energy_report_generation_log_controller(
    data: EnergyReportGenerationLogCreateSchema,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:update"]))],
) -> JSONResponse:
    result = await EnergyReportGenerationLogService(auth).create(data=data)
    return SuccessResponse(data=result, msg="创建能源报告生成操作记录成功")


@EnergyReportGenerationRouter.delete(
    "/log/delete",
    summary="删除能源报告生成操作记录",
    response_model=ResponseSchema[None],
)
async def delete_energy_report_generation_log_controller(
    ids: Annotated[list[int], Body(description="ID列表")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_energy:report_generation:delete"]))],
) -> JSONResponse:
    await EnergyReportGenerationLogService(auth).delete(ids=ids)
    return SuccessResponse(msg="删除能源报告生成操作记录成功")