init

app/controllers/base.py

@@ -0,0 +1,27 @@
+from uuid import uuid4
+
+from fastapi import Request
+
+from app.config import config
+from app.models.exception import HttpException
+
+
+def get_task_id(request: Request):
+    task_id = request.headers.get('x-task-id')
+    if not task_id:
+        task_id = uuid4()
+    return str(task_id)
+
+
+def get_api_key(request: Request):
+    api_key = request.headers.get('x-api-key')
+    return api_key
+
+
+def verify_token(request: Request):
+    token = get_api_key(request)
+    if token != config.app.get("api_key", ""):
+        request_id = get_task_id(request)
+        request_url = request.url
+        user_agent = request.headers.get('user-agent')
+        raise HttpException(task_id=request_id, status_code=401, message=f"invalid token: {request_url}, {user_agent}")

app/controllers/ping.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter
+from fastapi import Request
+
+router = APIRouter()
+
+
+@router.get("/ping", tags=["Health Check"], description="检查服务可用性", response_description="pong")
+def ping(request: Request) -> str:
+    return "pong"

app/controllers/v1/base.py

@@ -0,0 +1,11 @@
+from fastapi import APIRouter, Depends
+
+
+def new_router(dependencies=None):
+    router = APIRouter()
+    router.tags = ['V1']
+    router.prefix = '/api/v1'
+    # 将认证依赖项应用于所有路由
+    if dependencies:
+        router.dependencies = dependencies
+    return router

app/controllers/v1/video.py

@@ -0,0 +1,44 @@
+from os import path
+
+from fastapi import Request, Depends, Path
+from loguru import logger
+
+from app.controllers import base
+from app.controllers.v1.base import new_router
+from app.models.exception import HttpException
+from app.models.schema import TaskVideoRequest, TaskQueryResponse, TaskResponse, TaskQueryRequest
+from app.services import task as tm
+from app.utils import utils
+
+# 认证依赖项
+# router = new_router(dependencies=[Depends(base.verify_token)])
+router = new_router()
+
+
+@router.post("/videos", response_model=TaskResponse, summary="使用主题来生成短视频")
+async def create_video(request: Request, body: TaskVideoRequest):
+    task_id = utils.get_uuid()
+    request_id = base.get_task_id(request)
+    try:
+        task = {
+            "task_id": task_id,
+            "request_id": request_id,
+        }
+        body_dict = body.dict()
+        task.update(body_dict)
+        result = tm.start(task_id=task_id, params=body)
+        task["result"] = result
+        logger.success(f"video created: {utils.to_json(task)}")
+        return utils.get_response(200, task)
+    except ValueError as e:
+        raise HttpException(task_id=task_id, status_code=400, message=f"{request_id}: {str(e)}")
+
+
+@router.get("/tasks/{task_id}", response_model=TaskQueryResponse, summary="查询任务状态")
+async def get_task(request: Request, task_id: str = Path(..., description="任务ID"),
+                   query: TaskQueryRequest = Depends()):
+    request_id = base.get_task_id(request)
+    data = query.dict()
+    data["task_id"] = task_id
+    raise HttpException(task_id=task_id, status_code=404,
+                        message=f"{request_id}: task not found", data=data)