Delete main.py

main.py

@@ -1,262 +0,0 @@
-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="交易系统异常")