Upload api_share.py

api_share.py

@@ -6,7 +6,7 @@ from datetime import datetime
 from fastapi import APIRouter, UploadFile, File, Form
 from fastapi.responses import FileResponse
 from huggingface_hub import hf_hub_download, upload_file
-from core import get_db_list, save_db, api, REPO_ID, TOKEN, ShareLikeReq, ShareCommentReq, ShareCommentActionReq, ShareEditReq, ShareDeleteReq
+from core import get_db_list, save_db, api, REPO_ID, TOKEN, ShareLikeReq, ShareCommentReq, ShareCommentActionReq, ShareDeleteReq
 
 router = APIRouter()
 
@@ -17,7 +17,6 @@ async def get_share_list():
 @router.get("/api/share/image")
 async def get_share_image(name: str):
     try: 
-        # 🌟 优化：将下载图片操作放入线程池，防止阻塞主事件循环
         path = await asyncio.to_thread(hf_hub_download, repo_id=REPO_ID, filename=f"community_images/{name}", repo_type="dataset", token=TOKEN)
         return FileResponse(path)
     except: return {"status": "error"}
@@ -26,7 +25,7 @@ async def get_share_image(name: str):
 async def publish_share(username: str = Form(...), title: str = Form(...), content: str = Form(...), image: UploadFile = File(None)):
     try:
         db = get_db_list("community_posts.json")
-        post_id = uuid.uuid4().hex # 🌟 优化：使用完整 32 位 UUID 防碰撞
+        post_id = uuid.uuid4().hex
         post_data = { "id": post_id, "author": username, "title": title[:20], "content": content, "time": int(time.time() * 1000), "likes": 0, "liked_by": [], "comments": [] }
 
         if image and image.filename:
@@ -34,25 +33,41 @@ async def publish_share(username: str = Form(...), title: str = Form(...), conte
             image_name = f"{post_id}{ext}"
             temp_path = f"temp_share_{uuid.uuid4().hex}{ext}"
             with open(temp_path, "wb") as f: f.write(await image.read())
-            # 🌟 优化：将上传图片操作放入线程池，防止假死
             await asyncio.to_thread(upload_file, path_or_fileobj=temp_path, path_in_repo=f"community_images/{image_name}", repo_id=REPO_ID, repo_type="dataset", token=TOKEN)
             if os.path.exists(temp_path): os.remove(temp_path)
             post_data["image"] = image_name
 
         db.insert(0, post_data)
-        save_db(db, "community_posts.json") # 瞬间落入内存字典
+        save_db(db, "community_posts.json")
         return {"status": "success", "post": post_data}
     except Exception as e: return {"status": "error", "message": str(e)}
 
+# 🌟 核心升级：编辑接口重构，支持表单图片修改
 @router.post("/api/share/edit")
-async def edit_share(req: ShareEditReq):
+async def edit_share(post_id: str = Form(...), username: str = Form(...), title: str = Form(...), content: str = Form(...), image: UploadFile = File(None)):
     try:
         db = get_db_list("community_posts.json")
         for p in db:
-            if p["id"] == req.post_id:
-                if p["author"] != req.username: return {"status": "error", "message": "无权限"}
-                p["title"] = req.title[:20]
-                p["content"] = req.content
+            if p["id"] == post_id:
+                if p["author"] != username: return {"status": "error", "message": "无权限"}
+                p["title"] = title[:20]
+                p["content"] = content
+                p["last_edit_date"] = datetime.now().strftime("%Y-%m-%d")
+
+                # 如果上传了新图片，删除老图并上传新图
+                if image and image.filename:
+                    ext = os.path.splitext(image.filename)[1].lower()
+                    image_name = f"{post_id}_{uuid.uuid4().hex[:8]}{ext}"
+                    temp_path = f"temp_share_edit_{uuid.uuid4().hex}{ext}"
+                    with open(temp_path, "wb") as f: f.write(await image.read())
+                    await asyncio.to_thread(upload_file, path_or_fileobj=temp_path, path_in_repo=f"community_images/{image_name}", repo_id=REPO_ID, repo_type="dataset", token=TOKEN)
+                    if os.path.exists(temp_path): os.remove(temp_path)
+                    
+                    if p.get("image"):
+                        try: asyncio.create_task(asyncio.to_thread(api.delete_file, path_in_repo=f"community_images/{p['image']}", repo_id=REPO_ID, token=TOKEN))
+                        except: pass
+                    p["image"] = image_name
+
                 save_db(db, "community_posts.json")
                 return {"status": "success"}
         return {"status": "error", "message": "帖子不存在"}
@@ -68,7 +83,6 @@ async def delete_share(req: ShareDeleteReq):
                 del_post = db.pop(i)
                 if del_post.get("image"):
                     try:
-                        # 🌟 优化：后台丢入线程池静默删除垃圾图片，不让用户等待
                         asyncio.create_task(asyncio.to_thread(api.delete_file, path_in_repo=f"community_images/{del_post['image']}", repo_id=REPO_ID, token=TOKEN))
                     except: pass
                 save_db(db, "community_posts.json")