更新

backend/app/api/publish.py

@@ -12,6 +12,7 @@ router = APIRouter()
 publish_service = PublishService()
 
 class PublishRequest(BaseModel):
+    """Video publish request model"""
     video_path: str
     platform: str
     title: str
@@ -20,13 +21,25 @@ class PublishRequest(BaseModel):
     publish_time: Optional[datetime] = None
 
 class PublishResponse(BaseModel):
+    """Video publish response model"""
     success: bool
     message: str
     platform: str
     url: Optional[str] = None
 
+# Supported platforms for validation
+SUPPORTED_PLATFORMS = {"bilibili", "douyin", "xiaohongshu"}
+
 @router.post("/", response_model=PublishResponse)
 async def publish_video(request: PublishRequest, background_tasks: BackgroundTasks):
+    """发布视频到指定平台"""
+    # Validate platform
+    if request.platform not in SUPPORTED_PLATFORMS:
+        raise HTTPException(
+            status_code=400, 
+            detail=f"不支持的平台: {request.platform}。支持的平台: {', '.join(SUPPORTED_PLATFORMS)}"
+        )
+    
     try:
         result = await publish_service.publish(
             video_path=request.video_path,
@@ -56,6 +69,10 @@ async def list_accounts():
 
 @router.post("/login/{platform}")
 async def login_platform(platform: str):
+    """触发平台QR码登录"""
+    if platform not in SUPPORTED_PLATFORMS:
+        raise HTTPException(status_code=400, detail=f"不支持的平台: {platform}")
+    
     result = await publish_service.login(platform)
     if result.get("success"):
         return result
@@ -65,12 +82,18 @@ async def login_platform(platform: str):
 @router.post("/logout/{platform}")
 async def logout_platform(platform: str):
     """注销平台登录"""
+    if platform not in SUPPORTED_PLATFORMS:
+        raise HTTPException(status_code=400, detail=f"不支持的平台: {platform}")
+    
     result = publish_service.logout(platform)
     return result
 
 @router.get("/login/status/{platform}")
 async def get_login_status(platform: str):
     """检查登录状态 (优先检查活跃的扫码会话)"""
+    if platform not in SUPPORTED_PLATFORMS:
+        raise HTTPException(status_code=400, detail=f"不支持的平台: {platform}")
+    
     return publish_service.get_login_session_status(platform)
 
 @router.post("/cookies/save/{platform}")
@@ -82,7 +105,13 @@ async def save_platform_cookie(platform: str, cookie_data: dict):
         platform: 平台ID
         cookie_data: {"cookie_string": "document.cookie的内容"}
     """
+    if platform not in SUPPORTED_PLATFORMS:
+        raise HTTPException(status_code=400, detail=f"不支持的平台: {platform}")
+    
     cookie_string = cookie_data.get("cookie_string", "")
+    if not cookie_string:
+        raise HTTPException(status_code=400, detail="cookie_string 不能为空")
+    
     result = await publish_service.save_cookie_string(platform, cookie_string)
     
     if result.get("success"):