Upload 3 files

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9-slim
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+# Hugging Face Spaces 默认将应用暴露在 7860 端口
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,262 @@
+import os
+import json
+import uuid
+import requests
+import threading
+from datetime import datetime, timezone
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from huggingface_hub import HfApi, CommitOperationAdd
+
+app = FastAPI()
+
+# ==========================================
+# ⚙️ 全局配置与安全锁
+# ==========================================
+HF_TOKEN = os.environ.get("HF_TOKEN")
+# 🌟 必须替换为你自己的 Dataset 地址！
+DATASET_REPO_ID = "ZHIWEI666/ComfyUI-Ranking"
+
+api = HfApi(token=HF_TOKEN)
+db_lock = threading.Lock() # 全局内存锁：强制所有写请求排队，杜绝并发覆盖
+
+# ==========================================
+# 📦 数据请求模型 (Pydantic)
+# ==========================================
+class PublishReq(BaseModel): 
+    uid: str; author_name: str; type: str; name: str; short_desc: str; detailed_desc: str; target_url: str; price: int
+class RedeemReq(BaseModel): 
+    uid: str; code: str
+class UnlockReq(BaseModel): 
+    uid: str; content_id: str
+
+# ==========================================
+# 🛠️ 核心工具函数：极速读取与原子提交
+# ==========================================
+def fetch_json(filename: str):
+    """通过 Raw URL 极速拉取最新 JSON 数据（绕过本地缓存）"""
+    url = f"https://huggingface.co/datasets/{DATASET_REPO_ID}/raw/main/{filename}"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
+    res = requests.get(url, headers=headers, timeout=10)
+    
+    if res.status_code == 200:
+        try:
+            return res.json()
+        except json.JSONDecodeError:
+            pass
+    
+    # 如果文件不存在或损坏，返回初始结构
+    if filename in ["users.json", "codes.json"]:
+        return {}
+    return []
+
+def commit_multiple_jsons(commit_message: str, file_dict: dict):
+    """
+    高级原子提交：将多个修改后的 dict 直接在内存中转为 bytes，
+    并一次性 Commit 到 Hugging Face，保证数据绝对一致性。
+    """
+    operations = []
+    for filename, data in file_dict.items():
+        # 将 Python 字典转为格式化的 JSON 字节流
+        json_bytes = json.dumps(data, ensure_ascii=False, indent=2).encode('utf-8')
+        operations.append(
+            CommitOperationAdd(path_in_repo=filename, path_or_fileobj=json_bytes)
+        )
+    
+    # 执行打包提交
+    api.create_commit(
+        repo_id=DATASET_REPO_ID,
+        repo_type="dataset",
+        operations=operations,
+        commit_message=commit_message
+    )
+
+# ==========================================
+# 🚀 业务路由：拉取数据
+# ==========================================
+@app.get("/")
+def read_root():
+    return {"status": "Running", "message": "Ranking API is online!"}
+
+@app.get("/api/rankings")
+def get_rankings():
+    """直接返回全量作品列表供前端渲染"""
+    try:
+        contents = fetch_json("contents.json")
+        return {"status": "success", "data": contents}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# ==========================================
+# 📝 业务路由：发布新作品
+# ==========================================
+@app.post("/api/publish")
+def publish_content(req: PublishReq):
+    with db_lock:
+        try:
+            contents_db = fetch_json("contents.json")
+            users_db = fetch_json("users.json")
+            
+            # 初始化发布者资料
+            if req.uid not in users_db:
+                users_db[req.uid] = {"name": req.author_name, "balance": 0}
+            
+            new_item = {
+                "id": f"{req.type}_{uuid.uuid4().hex[:8]}",
+                "type": req.type,
+                "name": req.name,
+                "short_desc": req.short_desc,
+                "detailed_desc": req.detailed_desc,
+                "author": req.author_name,
+                "uid": req.uid,
+                "target_url": req.target_url,
+                "price": req.price,
+                "likes": 0, "favs": 0, "cmts": 0, "downloads": 0,
+                "comments": [],
+                "publish_time": datetime.now(timezone.utc).isoformat()
+            }
+            
+            # 插入到列表最前方
+            contents_db.insert(0, new_item)
+            
+            # 提交到云端
+            commit_multiple_jsons(
+                commit_message=f"Publish new {req.type}: {req.name}",
+                file_dict={"contents.json": contents_db, "users.json": users_db}
+            )
+            
+            return {"status": "success", "message": "发布成功！", "item": new_item}
+        except Exception as e:
+            print(f"Publish Error: {e}")
+            raise HTTPException(status_code=500, detail="云端写入失败")
+
+# ==========================================
+# 💰 业务路由：卡密核销充值
+# ==========================================
+@app.post("/api/wallet/redeem")
+def redeem_code(req: RedeemReq):
+    code_str = req.code.strip().upper()
+    with db_lock:
+        try:
+            codes_db = fetch_json("codes.json")
+            users_db = fetch_json("users.json")
+            tx_db = fetch_json("transactions.json")
+            
+            # 1. 校验卡密有效性
+            if code_str not in codes_db:
+                return {"status": "error", "message": "无效的卡密，请检查是否输入正确。"}
+            if codes_db[code_str].get("status") != "active":
+                return {"status": "error", "message": "该卡密已被使用或已失效。"}
+            
+            # 2. 初始化用户
+            if req.uid not in users_db:
+                users_db[req.uid] = {"name": f"User_{req.uid[:4]}", "balance": 0}
+                
+            value = codes_db[code_str]["value"]
+            current_time = datetime.now(timezone.utc).isoformat()
+            
+            # 3. 更新状态 (卡密作废，增加余额)
+            codes_db[code_str]["status"] = "used"
+            codes_db[code_str]["used_by_uid"] = req.uid
+            codes_db[code_str]["used_at"] = current_time
+            
+            users_db[req.uid]["balance"] += value
+            new_balance = users_db[req.uid]["balance"]
+            
+            # 4. 记录流水
+            tx_record = {
+                "tx_id": f"TX-RCG-{uuid.uuid4().hex[:6].upper()}",
+                "uid": req.uid, "type": "recharge", "amount": value,
+                "related_id": code_str, "timestamp": current_time
+            }
+            tx_db.append(tx_record)
+            
+            # 5. 原子提交 3 个文件
+            commit_multiple_jsons(
+                commit_message=f"User {req.uid} redeemed code {code_str}",
+                file_dict={
+                    "codes.json": codes_db,
+                    "users.json": users_db,
+                    "transactions.json": tx_db
+                }
+            )
+            
+            return {"status": "success", "message": f"成功充值 {value} C币！", "new_balance": new_balance}
+        except Exception as e:
+            print(f"Redeem Error: {e}")
+            raise HTTPException(status_code=500, detail="核销系统异常")
+
+# ==========================================
+# 🔓 业务路由：扣费与资产解锁
+# ==========================================
+@app.post("/api/unlock")
+def unlock_content(req: UnlockReq):
+    with db_lock:
+        try:
+            contents_db = fetch_json("contents.json")
+            users_db = fetch_json("users.json")
+            orders_db = fetch_json("orders.json")
+            tx_db = fetch_json("transactions.json")
+            
+            buyer_uid = req.uid
+            content = next((item for item in contents_db if item["id"] == req.content_id), None)
+            if not content:
+                return {"status": "error", "message": "未找到该作品"}
+                
+            price = content.get("price", 0)
+            seller_uid = content.get("uid", "admin")
+            
+            # 1. 免费直接下发
+            if price == 0:
+                return {"status": "success", "message": "获取成功", "target_url": content["target_url"], "type": content["type"]}
+
+            # 2. 幂等性校验 (已经买过或自己买自己的，直接放行，不扣费)
+            already_bought = any(o["buyer_uid"] == buyer_uid and o["content_id"] == req.content_id for o in orders_db)
+            if already_bought or buyer_uid == seller_uid:
+                return {"status": "success", "message": "已恢复购买记录", "target_url": content["target_url"], "type": content["type"], "new_balance": users_db.get(buyer_uid, {}).get("balance", 0)}
+
+            # 3. 校验买家余额
+            buyer_data = users_db.get(buyer_uid)
+            if not buyer_data or buyer_data.get("balance", 0) < price:
+                return {"status": "error", "message": "可用余额不足，请先充值"}
+
+            # 4. 执行扣费事务
+            # 4.1 资金转移
+            users_db[buyer_uid]["balance"] -= price
+            if seller_uid in users_db:
+                users_db[seller_uid]["balance"] += price
+                
+            # 4.2 增加作品下载量
+            content["downloads"] = content.get("downloads", 0) + 1
+            
+            # 4.3 生成订单
+            current_time = datetime.now(timezone.utc).isoformat()
+            order_id = f"ORD-{uuid.uuid4().hex[:8].upper()}"
+            orders_db.append({"order_id": order_id, "buyer_uid": buyer_uid, "seller_uid": seller_uid, "content_id": req.content_id, "price": price, "timestamp": current_time})
+            
+            # 4.4 生成买卖双方流水
+            tx_db.append({"tx_id": f"TX-BUY-{uuid.uuid4().hex[:6]}", "uid": buyer_uid, "type": "consume", "amount": -price, "related_id": order_id, "timestamp": current_time})
+            tx_db.append({"tx_id": f"TX-SEL-{uuid.uuid4().hex[:6]}", "uid": seller_uid, "type": "earn", "amount": price, "related_id": order_id, "timestamp": current_time})
+
+            # 5. 原子提交 4 个受影响的文件
+            commit_multiple_jsons(
+                commit_message=f"Unlock {req.content_id} by {buyer_uid}",
+                file_dict={
+                    "users.json": users_db,
+                    "orders.json": orders_db,
+                    "transactions.json": tx_db,
+                    "contents.json": contents_db # 下载量+1了，所以要更新总表
+                }
+            )
+
+            return {
+                "status": "success", 
+                "message": "支付并解锁成功！", 
+                "target_url": content["target_url"], 
+                "type": content["type"],
+                "new_balance": users_db[buyer_uid]["balance"]
+            }
+
+        except Exception as e:
+            print(f"Unlock Error: {e}")
+            raise HTTPException(status_code=500, detail="交易系统异常")

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+uvicorn
+huggingface_hub
+pydantic
+requests
+filelock