fengchuanhn@gmail.com
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
"""OEM 卡密：列表、备注、批量分配给代理。"""
|
|
|
|
from typing import Literal
|
|
|
|
from fastapi import APIRouter, Query
|
|
|
|
from app.dependencies import DbSession, OemUser
|
|
from app.schemas.admin_card_key import CardKeyOut
|
|
from app.schemas.common import ApiResponse, PaginatedData
|
|
from app.schemas.oem_card_key import (
|
|
OemCardKeyBatchAssign,
|
|
OemCardKeyBatchAssignResult,
|
|
OemCardKeyUpdate,
|
|
)
|
|
from app.services import oem_card_key as card_key_service
|
|
|
|
router = APIRouter(prefix="/oem/card-keys", tags=["OEM-卡密"])
|
|
|
|
|
|
@router.get("", response_model=ApiResponse[PaginatedData[CardKeyOut]])
|
|
async def list_card_keys(
|
|
oem: OemUser,
|
|
db: DbSession,
|
|
page: int = Query(1, ge=1),
|
|
page_size: int = Query(
|
|
card_key_service.DEFAULT_PAGE_SIZE,
|
|
ge=1,
|
|
le=card_key_service.MAX_PAGE_SIZE,
|
|
),
|
|
status: Literal["all", "unused", "used"] | None = Query(
|
|
default="all",
|
|
description="筛选：all 全部 / unused 未使用 / used 已激活",
|
|
),
|
|
username: str | None = Query(default=None, description="按激活用户名模糊筛选"),
|
|
agent_id: int | None = Query(default=None, description="按预分配代理筛选"),
|
|
) -> ApiResponse[PaginatedData[CardKeyOut]]:
|
|
filter_status = None if status == "all" else status
|
|
cards, total = await card_key_service.list_card_keys(
|
|
db,
|
|
oem_id=oem.id,
|
|
page=page,
|
|
page_size=page_size,
|
|
status=filter_status,
|
|
username=username,
|
|
agent_id=agent_id,
|
|
)
|
|
return ApiResponse(
|
|
ok=True,
|
|
message="",
|
|
data=PaginatedData.build(
|
|
[CardKeyOut.model_validate(c) for c in cards],
|
|
total=total,
|
|
page=page,
|
|
page_size=page_size,
|
|
),
|
|
)
|
|
|
|
|
|
@router.patch("/{card_id}", response_model=ApiResponse[CardKeyOut])
|
|
async def update_card_key(
|
|
card_id: int,
|
|
body: OemCardKeyUpdate,
|
|
oem: OemUser,
|
|
db: DbSession,
|
|
) -> ApiResponse[CardKeyOut]:
|
|
try:
|
|
card = await card_key_service.update_card_key(
|
|
db, card_id, body, oem_id=oem.id
|
|
)
|
|
except card_key_service.OemCardKeyError as exc:
|
|
return ApiResponse(ok=False, message=exc.message)
|
|
|
|
return ApiResponse(
|
|
ok=True,
|
|
message="备注已更新",
|
|
data=CardKeyOut.model_validate(card),
|
|
)
|
|
|
|
|
|
@router.post("/assign-agent", response_model=ApiResponse[OemCardKeyBatchAssignResult])
|
|
async def batch_assign_to_agent(
|
|
body: OemCardKeyBatchAssign,
|
|
oem: OemUser,
|
|
db: DbSession,
|
|
) -> ApiResponse[OemCardKeyBatchAssignResult]:
|
|
try:
|
|
count = await card_key_service.batch_assign_to_agent(
|
|
db, body, oem_id=oem.id
|
|
)
|
|
except card_key_service.OemCardKeyError as exc:
|
|
return ApiResponse(ok=False, message=exc.message)
|
|
|
|
return ApiResponse(
|
|
ok=True,
|
|
message=f"已分配 {count} 张卡密",
|
|
data=OemCardKeyBatchAssignResult(updated_count=count),
|
|
)
|