149 lines
4.1 KiB
Python
149 lines
4.1 KiB
Python
import json
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.dependencies import get_db, get_admin_user
|
|
from app.models.user import User
|
|
from app.models.recharge_package import RechargePackage
|
|
from app.schemas.recharge_package import (
|
|
RechargePackageCreate,
|
|
RechargePackageUpdate,
|
|
RechargePackageOut,
|
|
)
|
|
from app.services.operation_log import log_operation
|
|
from app.utils.id_gen import generate_id
|
|
|
|
router = APIRouter(prefix="/admin/recharge-packages", tags=["admin-recharge-packages"])
|
|
|
|
|
|
def _to_out(pkg: RechargePackage) -> dict:
|
|
return {
|
|
"id": pkg.id,
|
|
"name": pkg.name,
|
|
"credits": round(pkg.credits, 2),
|
|
"price": round(pkg.price, 2),
|
|
"bonus_credits": round(pkg.bonus_credits, 2),
|
|
"total_credits": round(pkg.credits + pkg.bonus_credits, 2),
|
|
"description": pkg.description,
|
|
"package_type": pkg.package_type,
|
|
"is_gift": pkg.is_gift,
|
|
"is_active": pkg.is_active,
|
|
"sort_order": pkg.sort_order,
|
|
}
|
|
|
|
|
|
@router.get("", response_model=list[RechargePackageOut])
|
|
async def admin_list_packages(
|
|
admin: User = Depends(get_admin_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
result = await db.execute(
|
|
select(RechargePackage).order_by(RechargePackage.sort_order)
|
|
)
|
|
return [_to_out(p) for p in result.scalars().all()]
|
|
|
|
|
|
@router.post("", response_model=RechargePackageOut)
|
|
async def create_package(
|
|
data: RechargePackageCreate,
|
|
admin: User = Depends(get_admin_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
pkg = RechargePackage(id=generate_id(), **data.model_dump())
|
|
db.add(pkg)
|
|
await db.flush()
|
|
await log_operation(
|
|
db,
|
|
admin.id,
|
|
admin.username,
|
|
f"创建充值套餐 {pkg.name}",
|
|
"POST",
|
|
"/admin/recharge-packages",
|
|
detail=json.dumps(
|
|
{
|
|
"id": pkg.id,
|
|
"name": pkg.name,
|
|
"price": round(pkg.price, 2),
|
|
"credits": round(pkg.credits, 2),
|
|
"bonus_credits": round(pkg.bonus_credits, 2),
|
|
},
|
|
ensure_ascii=False,
|
|
),
|
|
)
|
|
await db.commit()
|
|
return _to_out(pkg)
|
|
|
|
|
|
@router.put("/{pkg_id}", response_model=RechargePackageOut)
|
|
async def update_package(
|
|
pkg_id: str,
|
|
data: RechargePackageUpdate,
|
|
admin: User = Depends(get_admin_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
result = await db.execute(
|
|
select(RechargePackage).where(RechargePackage.id == pkg_id).limit(1)
|
|
)
|
|
pkg = result.scalar_one_or_none()
|
|
if not pkg:
|
|
raise HTTPException(status_code=404, detail="套餐不存在")
|
|
before = _to_out(pkg)
|
|
for k, v in data.model_dump(exclude_unset=True).items():
|
|
setattr(pkg, k, v)
|
|
await db.flush()
|
|
after = _to_out(pkg)
|
|
await log_operation(
|
|
db,
|
|
admin.id,
|
|
admin.username,
|
|
f"更新充值套餐 {pkg.name}",
|
|
"PUT",
|
|
f"/admin/recharge-packages/{pkg_id}",
|
|
detail=json.dumps(
|
|
{
|
|
"id": pkg_id,
|
|
"before": before,
|
|
"after": after,
|
|
},
|
|
ensure_ascii=False,
|
|
),
|
|
)
|
|
await db.commit()
|
|
return _to_out(pkg)
|
|
|
|
|
|
@router.delete("/{pkg_id}")
|
|
async def delete_package(
|
|
pkg_id: str,
|
|
admin: User = Depends(get_admin_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
result = await db.execute(
|
|
select(RechargePackage).where(RechargePackage.id == pkg_id).limit(1)
|
|
)
|
|
pkg = result.scalar_one_or_none()
|
|
if not pkg:
|
|
raise HTTPException(status_code=404, detail="套餐不存在")
|
|
pkg_name = pkg.name
|
|
await db.delete(pkg)
|
|
await db.flush()
|
|
await log_operation(
|
|
db,
|
|
admin.id,
|
|
admin.username,
|
|
f"删除充值套餐 {pkg_name}",
|
|
"DELETE",
|
|
f"/admin/recharge-packages/{pkg_id}",
|
|
detail=json.dumps(
|
|
{
|
|
"id": pkg_id,
|
|
"name": pkg_name,
|
|
},
|
|
ensure_ascii=False,
|
|
),
|
|
)
|
|
await db.commit()
|
|
return {"ok": True}
|