FastapiAdmin/backend/app/api/v1/module_platform/package/controller.py

from typing import Annotated

from fastapi import APIRouter, Body, Depends, Path, Query, Security, status
from fastapi.responses import JSONResponse
from fastapi_cache import FastAPICache
from fastapi_cache.decorator import cache
from sqlalchemy.ext.asyncio import AsyncSession

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

from .schema import PackageCreateSchema, PackageMenuSetSchema, PackageOutSchema, PackageQueryParam, PackageUpdateSchema
from .service import PackageService

PackageRouter = APIRouter(route_class=OperationLogRoute, prefix="/package", tags=["套餐管理"])

_PKG_NS = "package"


@PackageRouter.get("/options", summary="获取套餐下拉选项", response_model=ResponseSchema[list[dict[str, int | str]]])
async def get_package_options_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    options = await PackageService(auth, db).get_options()
    return SuccessResponse(data=options, msg="获取套餐选项成功")


@PackageRouter.get("/detail/{id}", summary="获取套餐详情", response_model=ResponseSchema[PackageOutSchema])
@cache(expire=300, namespace=_PKG_NS)
async def get_obj_detail_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    id: Annotated[int, Path(description="套餐ID", ge=1)],
) -> JSONResponse:
    result_dict = await PackageService(auth, db).detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取套餐详情成功")


@PackageRouter.get("/list", summary="获取套餐列表", response_model=ResponseSchema[PageResultSchema[PackageOutSchema]])
async def get_obj_list_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    page: Annotated[PaginationQueryParam, Query(description="分页参数")],
    search: Annotated[PackageQueryParam, Query(description="查询参数")],
) -> JSONResponse:
    result_dict = await PackageService(auth, db).page(
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by,
    )
    return SuccessResponse(data=result_dict, msg="查询成功")


@PackageRouter.post("/create", status_code=status.HTTP_201_CREATED, summary="创建套餐", response_model=ResponseSchema[PackageOutSchema])
async def create_obj_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:create"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    data: Annotated[PackageCreateSchema, Body(description="套餐信息")],
) -> JSONResponse:
    result_dict = await PackageService(auth, db).create(data=data)
    await FastAPICache.clear(namespace=_PKG_NS)
    return SuccessResponse(data=result_dict, msg="创建成功")


@PackageRouter.put("/update/{id}", summary="更新套餐", response_model=ResponseSchema[PackageOutSchema])
async def update_obj_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:update"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    id: Annotated[int, Path(description="套餐ID", ge=1)],
    data: Annotated[PackageUpdateSchema, Body(description="套餐信息")],
) -> JSONResponse:
    result_dict = await PackageService(auth, db).update(id=id, data=data)
    await FastAPICache.clear(namespace=_PKG_NS)
    return SuccessResponse(data=result_dict, msg="更新成功")


@PackageRouter.delete("/delete", summary="删除套餐", response_model=ResponseSchema)
async def delete_obj_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:delete"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    ids: Annotated[list[int], Body(description="ID列表")],
) -> JSONResponse:
    await PackageService(auth, db).delete(ids=ids)
    await FastAPICache.clear(namespace=_PKG_NS)
    return SuccessResponse(msg="删除成功")


@PackageRouter.patch("/status/batch", summary="批量修改状态", response_model=ResponseSchema)
async def set_available_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:update"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    data: Annotated[BatchSetAvailable, Body(description="状态设置")],
) -> JSONResponse:
    for id in data.ids:
        await PackageService(auth, db).update(id=id, data=PackageUpdateSchema(status=data.status))
    await FastAPICache.clear(namespace=_PKG_NS)
    return SuccessResponse(msg="状态设置成功")


@PackageRouter.get("/menus/{package_id}", summary="获取套餐菜单", response_model=ResponseSchema[list[int]])
async def get_menus_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    package_id: Annotated[int, Path(description="套餐ID", ge=1)],
) -> JSONResponse:
    result = await PackageService(auth, db).get_menus(package_id=package_id)
    return SuccessResponse(data=result, msg="获取成功")


@PackageRouter.post("/menus/{package_id}/set", summary="设置套餐菜单", response_model=ResponseSchema)
async def set_menus_controller(
    auth: Annotated[AuthSchema, Security(AuthPermission(["module_package:package:update"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
    package_id: Annotated[int, Path(description="套餐ID", ge=1)],
    data: Annotated[PackageMenuSetSchema, Body(description="菜单列表")],
) -> JSONResponse:
    await PackageService(auth, db).set_menus(package_id=package_id, data=data)
    return SuccessResponse(msg="设置成功")