Upload api_playlist.py

api_playlist.py

@@ -1,5 +1,6 @@
 import os
 import uuid
+import time
 from fastapi import APIRouter, UploadFile, File, Form
 from pydantic import BaseModel
 from huggingface_hub import upload_file
@@ -7,16 +8,11 @@ from core import get_db, save_db, REPO_ID, TOKEN
 
 router = APIRouter()
 
-class PlaylistAddReq(BaseModel):
-    username: str
-    type: str
-    playlist_name: str
-    filename: str
-
-class PlaylistDeleteReq(BaseModel):
-    username: str
-    type: str
-    name: str
+class PlaylistAddReq(BaseModel): username: str; type: str; playlist_name: str; filename: str
+class PlaylistDeleteReq(BaseModel): username: str; type: str; name: str
+# 🌟 新增请求模型
+class PlaylistLikeReq(BaseModel): username: str; type: str; author: str; name: str
+class PlaylistCommentReq(BaseModel): username: str; type: str; author: str; name: str; content: str
 
 @router.get("/api/cloud_pl/get")
 def get_cloud_playlists(): 
@@ -43,7 +39,8 @@ async def create_cloud_playlist(username: str = Form(...), type: str = Form(...)
             upload_file(path_or_fileobj=temp_path, path_in_repo=f"community_images/{cover_name}", repo_id=REPO_ID, repo_type="dataset", token=TOKEN)
             os.remove(temp_path)
 
-        target_dict[name] = {"description": description[:200], "cover": cover_name, "songs": []}
+        # 🌟 初始化点赞与评论空数据
+        target_dict[name] = {"description": description[:200], "cover": cover_name, "songs": [], "likes": 0, "liked_by": [], "comments": []}
         save_db(db, "cloud_playlists.json")
         return {"status": "success"}
     except Exception as e: return {"status": "error", "message": str(e)}
@@ -58,8 +55,9 @@ async def edit_cloud_playlist(username: str = Form(...), type: str = Form(...),
         if old_name != name and name in target_dict: return {"status": "error", "message": "新名称已存在"}
 
         pl_data = target_dict.pop(old_name)
-        # 兼容旧版本纯数组数据
         if isinstance(pl_data, list): pl_data = {"description": "", "cover": "", "songs": pl_data}
+        # 兼容补充互动字段
+        if "likes" not in pl_data: pl_data.update({"likes": 0, "liked_by": [], "comments": []})
         
         pl_data["description"] = description[:200]
         if image:
@@ -96,13 +94,52 @@ def add_cloud_playlist(req: PlaylistAddReq):
         target_dict = db.get("admin", {}) if req.type == "admin" else db.get("users", {}).get(req.username, {})
         pl = target_dict.get(req.playlist_name)
         if pl is None: return {"status": "error", "message": "歌单不存在"}
-        
-        # 兼容旧版本列表
         if isinstance(pl, list):
             if req.filename not in pl: pl.append(req.filename)
         else:
             if req.filename not in pl["songs"]: pl["songs"].append(req.filename)
+        save_db(db, "cloud_playlists.json")
+        return {"status": "success"}
+    except Exception as e: return {"status": "error", "message": str(e)}
+
+# 🌟 新增：歌单点赞
+@router.post("/api/cloud_pl/like")
+def like_cloud_playlist(req: PlaylistLikeReq):
+    try:
+        db = get_db("cloud_playlists.json")
+        target_dict = db.get("admin", {}) if req.type == "admin" else db.get("users", {}).get(req.author, {})
+        pl = target_dict.get(req.name)
+        if not pl or isinstance(pl, list): return {"status": "error", "message": "格式异常或不存在"}
+        if "liked_by" not in pl: pl["liked_by"] = []; pl["likes"] = 0
+        
+        if req.username in pl["liked_by"]:
+            pl["liked_by"].remove(req.username)
+            pl["likes"] = max(0, pl.get("likes", 1) - 1)
+        else:
+            pl["liked_by"].append(req.username)
+            pl["likes"] = pl.get("likes", 0) + 1
             
         save_db(db, "cloud_playlists.json")
         return {"status": "success"}
+    except Exception as e: return {"status": "error", "message": str(e)}
+
+# 🌟 新增：歌单评论
+@router.post("/api/cloud_pl/comment")
+def comment_cloud_playlist(req: PlaylistCommentReq):
+    try:
+        if len(req.content) > 99: return {"status": "error", "message": "评论过长"}
+        db = get_db("cloud_playlists.json")
+        target_dict = db.get("admin", {}) if req.type == "admin" else db.get("users", {}).get(req.author, {})
+        pl = target_dict.get(req.name)
+        if not pl or isinstance(pl, list): return {"status": "error", "message": "格式异常或不存在"}
+        if "comments" not in pl: pl["comments"] = []
+        
+        pl["comments"].append({
+            "id": uuid.uuid4().hex[:8],
+            "user": req.username,
+            "text": req.content,
+            "time": int(time.time() * 1000)
+        })
+        save_db(db, "cloud_playlists.json")
+        return {"status": "success"}
     except Exception as e: return {"status": "error", "message": str(e)}