增加前测模板

video-gen-api/app/api/v1/pre_test_template.py

@@ -0,0 +1,294 @@
+import json
+from typing import Any, Optional
+
+from fastapi import APIRouter, Depends, HTTPException, Query, status
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select
+from app.models.pre_test_template import PreTestTemplate
+
+from app.dependencies import get_current_user, get_db
+from app.models.user import User
+from app.schemas.pre_test_template import (
+    PreTestTemplateCreate,
+    PreTestTemplateListResponse,
+    PreTestTemplateOut,
+    PreTestTemplateUpdate,
+)
+from app.services.pre_test_template_service import (
+    create_pre_test_template,
+    delete_pre_test_template,
+    get_default_template,
+    get_pre_test_template,
+    get_pre_test_template_list,
+    update_pre_test_template,
+)
+
+router = APIRouter(prefix="/pre-test-template", tags=["前测模板"])
+
+
+def _template_to_dict(template):
+    return {
+        "id": template.id,
+        "name": template.name,
+        "note": template.note,
+        "platform": template.platform,
+        "external_action": template.external_action,
+        "cpa_bid": template.cpa_bid,
+        "audience_gender": template.audience_gender,
+        "audience_age": json.loads(template.audience_age) if template.audience_age else None,
+        "audience_region": json.loads(template.audience_region) if template.audience_region else None,
+        "audience_network": json.loads(template.audience_network) if template.audience_network else None,
+        "cus_name": template.cus_name,
+        "pricing_type": template.pricing_type,
+        "cost_cap": template.cost_cap,
+        "target_cost": template.target_cost,
+        "nobid": template.nobid,
+        "cpc_bid": template.cpc_bid,
+        "budget": template.budget,
+        "is_default": template.is_default,
+        "user_id": template.user_id,
+        "created_at": template.created_at,
+        "updated_at": template.updated_at,
+    }
+
+
+@router.post(
+    "/create",
+    summary="创建前测模板",
+    description="创建一个新的前测模板",
+)
+async def create_template(
+    req: PreTestTemplateCreate,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+) -> Any:
+    try:
+        #新增一个判断，如果模板名称重复，提示用户修改
+        result = await db.execute(
+            select(PreTestTemplate).where(
+                PreTestTemplate.name == req.name,
+                PreTestTemplate.user_id == current_user.id,
+                PreTestTemplate.deleted_at.is_(None),
+            )
+        )
+        existing_template = result.scalar()
+        if existing_template:
+            raise ValueError("模板名称已存在")
+        
+        template = await create_pre_test_template(
+            user_id=current_user.id,
+            db=db,
+            name=req.name,
+            note=req.note,
+            platform=req.platform,
+            external_action=req.external_action,
+            cpa_bid=req.cpa_bid,
+            audience_gender=req.audience_gender,
+            audience_age=req.audience_age,
+            audience_region=req.audience_region,
+            audience_network=req.audience_network,
+            cus_name=req.cus_name,
+            pricing_type=req.pricing_type,
+            cost_cap=req.cost_cap,
+            target_cost=req.target_cost,
+            nobid=req.nobid,
+            cpc_bid=req.cpc_bid,
+            budget=req.budget,
+            is_default=req.is_default,
+        )
+
+        return {
+            "code": 0,
+            "message": "创建成功",
+        }
+    except ValueError as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=str(e),
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"创建失败: {str(e)}",
+        )
+
+
+@router.get(
+    "/list",
+    summary="获取前测模板列表",
+    description="获取当前用户的前测模板列表，支持按平台筛选和分页",
+    response_model=PreTestTemplateListResponse,
+)
+async def list_templates(
+    platform: Optional[str] = Query(None, description="投放平台筛选（AD/QIANCHUAN/LOCAL）"),
+    page: int = Query(1, description="页码，默认1"),
+    page_size: int = Query(10, description="每页数量，默认10，最大100"),
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+) -> Any:
+    try:
+        result = await get_pre_test_template_list(
+            user_id=current_user.id,
+            db=db,
+            platform=platform,
+            page=page,
+            page_size=page_size,
+        )
+
+        return {
+            "code": 0,
+            "message": "查询成功",
+            "data": [_template_to_dict(t) for t in result["data"]],
+            "pagination": {
+                "page": result["page"],
+                "page_size": result["page_size"],
+                "total": result["total"],
+                "total_pages": (result["total"] + result["page_size"] - 1) // result["page_size"],
+            },
+        }
+    except ValueError as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=str(e),
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"查询失败: {str(e)}",
+        )
+
+
+@router.get(
+    "/default",
+    summary="获取默认前测模板",
+    description="获取当前用户的默认前测模板",
+)
+async def get_default(
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+) -> Any:
+    try:
+        template = await get_default_template(current_user.id, db)
+
+        if not template:
+            return {
+                "code": 0,
+                "message": "未设置默认模板",
+                "data": None,
+            }
+
+        return {
+            "code": 0,
+            "message": "查询成功",
+            "data": _template_to_dict(template),
+        }
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"查询失败: {str(e)}",
+        )
+
+
+@router.get(
+    "/select/{template_id}",
+    summary="获取前测模板详情",
+    description="根据模板id获取前测模板详情",
+)
+async def get_template(
+    template_id: str,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+) -> Any:
+    try:
+        template = await get_pre_test_template(template_id, current_user.id, db)
+
+        if not template:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="模板不存在",
+            )
+
+        return {
+            "code": 0,
+            "message": "查询成功",
+            "data": _template_to_dict(template),
+        }
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"查询失败: {str(e)}",
+        )
+
+
+@router.post(
+    "/update/{template_id}",
+    summary="更新前测模板",
+    description="更新指定的前测模板",
+)
+async def update_template(
+    template_id: str,
+    req: PreTestTemplateUpdate,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+) -> Any:
+    try:
+        update_data = req.dict(exclude_none=True)
+        template = await update_pre_test_template(template_id, current_user.id, db, **update_data)
+
+        if not template:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="模板不存在",
+            )
+
+        return {
+            "code": 0,
+            "message": "更新成功",
+            "data": _template_to_dict(template),
+        }
+    except HTTPException:
+        raise
+    except ValueError as e:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=str(e),
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"更新失败: {str(e)}",
+        )
+
+
+@router.get(
+    "/delete/{template_id}",
+    summary="删除前测模板",
+    description="软删除指定的前测模板",
+)
+async def delete_template(
+    template_id: str,
+    current_user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+) -> Any:
+    try:
+        success = await delete_pre_test_template(template_id, current_user.id, db)
+
+        if not success:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="模板不存在",
+            )
+
+        return {
+            "code": 0,
+            "message": "删除成功",
+        }
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"删除失败: {str(e)}",
+        )