Upload 4 files

app.py

@@ -6,6 +6,7 @@ from sqlalchemy.orm import Session
 from pydantic import BaseModel
 from huggingface_hub import hf_hub_download, HfApi
 import hashlib
+import uuid
 import urllib.parse
 import urllib.request
 import urllib.error
@@ -325,14 +326,19 @@ def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
     if file_type == "avatar":
         max_size = 2 * 1024 * 1024   
     elif file_type == "cover":
-        max_size = 5 * 1024 * 1024   
+        max_size = 5 * 1024 * 1024
+    elif file_type == "post_video":
+        max_size = 50 * 1024 * 1024
         
     if len(content) > max_size:
         raise HTTPException(status_code=400, detail=f"文件过大，{file_type} 类型请限制在 {max_size // (1024*1024)}MB 以内")
         
     ext = file.filename.split(".")[-1].lower()
-    if ext not in ["jpg", "jpeg", "png", "gif", "webp", "json", "mp4"]:
-        raise HTTPException(status_code=400, detail="不支持的文件格式")
+    allowed_exts = ["jpg", "jpeg", "png", "gif", "webp", "json", "mp4"]
+    if file_type == "post_video":
+        allowed_exts = ["mp4", "webm", "mov"]
+    if ext not in allowed_exts:
+        raise HTTPException(status_code=400, detail=f"不支持的文件格式，{file_type} 类型仅支持 {', '.join(allowed_exts)}")
     
     # 🔒 P0安全优化：图片内容审核
     if ext in ["jpg", "jpeg", "png", "gif", "webp"]:
@@ -343,8 +349,11 @@ def upload_file(file: UploadFile = File(...), file_type: str = Form(...)):
                 detail=f"图片内容审核未通过：{moderation_result.label or '内容不合规'}"
             )
         
-    file_hash = hashlib.md5(content).hexdigest()[:10]
-    safe_filename = f"{file_type}_{file_hash}.{ext}"
+    if file_type == "post_video":
+        safe_filename = f"post_video_{uuid.uuid4().hex[:8]}.{ext}"
+    else:
+        file_hash = hashlib.md5(content).hexdigest()[:10]
+        safe_filename = f"{file_type}_{file_hash}.{ext}"
     
     local_tmp_path = f"/tmp/{safe_filename}"
     with open(local_tmp_path, "wb") as f:

models.py

@@ -228,6 +228,8 @@ class PostCreate(BaseModel):
     images: Optional[List[str]] = []       # 图片列表（最多9张）
     author: str                             # 作者账号
     is_original: Optional[bool] = False     # 🎨 是否为原创作品
+    post_type: str = "image"                # "image" 或 "video"
+    video_url: Optional[str] = None         # 视频 URL（仅 video 类型）
 
 class PostUpdate(BaseModel):
     """ 更新帖子 """
@@ -235,4 +237,6 @@ class PostUpdate(BaseModel):
     content: Optional[str] = None
     cover_image: Optional[str] = None
     images: Optional[List[str]] = None
-    is_original: Optional[bool] = None      # 🎨 是否为原创作品
+    is_original: Optional[bool] = None      # 🎨 是否为原创作品
+    post_type: Optional[str] = None         # "image" 或 "video"
+    video_url: Optional[str] = None         # 视频 URL（仅 video 类型）

router_posts.py

@@ -152,6 +152,12 @@ async def create_post(post: PostCreate, current_user: str = Depends(require_auth
     
     # 限制图片数量
     images = (post.images or [])[:9]
+    post_type = post.post_type if hasattr(post, "post_type") else "image"
+    video_url = post.video_url if hasattr(post, "video_url") else None
+    
+    # 视频帖：images 置为空列表
+    if post_type == "video":
+        images = []
     
     new_post = {
         "id": f"post_{int(time.time())}_{uuid.uuid4().hex[:6]}",
@@ -162,6 +168,8 @@ async def create_post(post: PostCreate, current_user: str = Depends(require_auth
         "author": current_user,
         "created_at": int(time.time()),
         "is_original": post.is_original if post.is_original is not None else False,  # 🎨 原创作品标记
+        "post_type": post_type,
+        "video_url": video_url,
         # 互动数据
         "likes": 0,
         "favorites": 0,
@@ -210,6 +218,13 @@ async def update_post(post_id: str, update_data: PostUpdate, current_user: str =
                     post["images"] = update_data.images[:9]
                 if update_data.is_original is not None:
                     post["is_original"] = update_data.is_original  # 🎨 更新原创作品标记
+                if update_data.post_type is not None:
+                    post["post_type"] = update_data.post_type
+                    # 如果切换为视频帖，清空图片列表
+                    if update_data.post_type == "video":
+                        post["images"] = []
+                if update_data.video_url is not None:
+                    post["video_url"] = update_data.video_url
                 result_holder["post"] = post
                 return True
         result_holder["error"] = "not_found"