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="删除能源报告生成操作记录成功")