增加打赏功能

models.py

@@ -105,4 +105,10 @@ class WithdrawRequest(BaseModel):
 
 class PurchaseRequest(BaseModel):
     account: str
-    item_id: str
+    item_id: str
+
+class TipRequest(BaseModel):
+    sender_account: str
+    target_account: str
+    amount: int
+    is_anonymous: bool

router_wallet.py

@@ -225,4 +225,102 @@ async def withdraw_earnings(req: WithdrawRequest, db: Session = Depends(get_db))
         else:
             raise HTTPException(status_code=400, detail=f"打款风控拦截: {result.get('sub_msg')}")
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"提现接口异常: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"提现接口异常: {str(e)}")
+
+@router.post("/api/wallet/tip")
+async def tip_user(req: TipRequest, db: Session = Depends(get_db)):
+    """赞赏接口：处理扣费、收益及赞赏榜单记录"""
+    if req.sender_account == req.target_account:
+        raise HTTPException(status_code=400, detail="不能打赏自己")
+    if req.amount <= 0:
+        raise HTTPException(status_code=400, detail="打赏金额必须大于0")
+
+    users_db = json_db.load_data("users.json", default_data={})
+    if req.target_account not in users_db:
+        raise HTTPException(status_code=404, detail="目标用户不存在")
+
+    target_user = users_db[req.target_account]
+    tips_received = target_user.get("tips_received", {})
+
+    # 计算等级额度 (10分=1星, 50分=1月, 250分=1太阳。上限9太阳=2250分)
+    current_tip = tips_received.get(req.sender_account, {}).get("amount", 0)
+    if current_tip + req.amount > 2250:
+        raise HTTPException(status_code=400, detail=f"您对该用户的打赏已达上限 (9个太阳/2250积分)，最多还能打赏 {2250 - current_tip} 积分")
+
+    # 1. 扣除打赏者余额 (悲观锁防并发)
+    sender_wallet = db.query(Wallet).filter(Wallet.account == req.sender_account).with_for_update().first()
+    if not sender_wallet or sender_wallet.balance < req.amount:
+        raise HTTPException(status_code=400, detail="积分余额不足，请先充值")
+    
+    sender_wallet.balance -= req.amount
+    record_transaction(db, req.sender_account, "TIP_SEND", -req.amount, req.target_account)
+
+    # 2. 增加作者收益
+    target_wallet = db.query(Wallet).filter(Wallet.account == req.target_account).with_for_update().first()
+    if not target_wallet:
+        target_wallet = Wallet(account=req.target_account)
+        db.add(target_wallet)
+    target_wallet.earn_balance += req.amount
+    record_transaction(db, req.target_account, "TIP_RECEIVE", req.amount, req.sender_account)
+
+    # 3. 永久记录到该用户的榜单 JSON 中
+    tips_received[req.sender_account] = {
+        "amount": current_tip + req.amount,
+        "is_anonymous": req.is_anonymous,
+        "sender_name": users_db.get(req.sender_account, {}).get("name", req.sender_account)
+    }
+    target_user["tips_received"] = tips_received
+    
+    db.commit()
+    json_db.save_data("users.json", users_db)
+
+    return {"status": "success", "balance": sender_wallet.balance}
+
+@router.post("/api/wallet/tip")
+async def tip_user(req: TipRequest, db: Session = Depends(get_db)):
+    """赞赏接口：处理扣费、收益及赞赏榜单记录"""
+    if req.sender_account == req.target_account:
+        raise HTTPException(status_code=400, detail="不能打赏自己")
+    if req.amount <= 0:
+        raise HTTPException(status_code=400, detail="打赏金额必须大于0")
+
+    users_db = json_db.load_data("users.json", default_data={})
+    if req.target_account not in users_db:
+        raise HTTPException(status_code=404, detail="目标用户不存在")
+
+    target_user = users_db[req.target_account]
+    tips_received = target_user.get("tips_received", {})
+
+    # 计算等级额度 (10分=1星, 50分=1月, 250分=1太阳。上限9太阳=2250分)
+    current_tip = tips_received.get(req.sender_account, {}).get("amount", 0)
+    if current_tip + req.amount > 2250:
+        raise HTTPException(status_code=400, detail=f"您对该用户的打赏已达上限 (9个太阳/2250积分)，最多还能打赏 {2250 - current_tip} 积分")
+
+    # 1. 扣除打赏者余额 (悲观锁防并发)
+    sender_wallet = db.query(Wallet).filter(Wallet.account == req.sender_account).with_for_update().first()
+    if not sender_wallet or sender_wallet.balance < req.amount:
+        raise HTTPException(status_code=400, detail="积分余额不足，请先充值")
+    
+    sender_wallet.balance -= req.amount
+    record_transaction(db, req.sender_account, "TIP_SEND", -req.amount, req.target_account)
+
+    # 2. 增加作者收益
+    target_wallet = db.query(Wallet).filter(Wallet.account == req.target_account).with_for_update().first()
+    if not target_wallet:
+        target_wallet = Wallet(account=req.target_account)
+        db.add(target_wallet)
+    target_wallet.earn_balance += req.amount
+    record_transaction(db, req.target_account, "TIP_RECEIVE", req.amount, req.sender_account)
+
+    # 3. 永久记录到该用户的榜单 JSON 中
+    tips_received[req.sender_account] = {
+        "amount": current_tip + req.amount,
+        "is_anonymous": req.is_anonymous,
+        "sender_name": users_db.get(req.sender_account, {}).get("name", req.sender_account)
+    }
+    target_user["tips_received"] = tips_received
+    
+    db.commit()
+    json_db.save_data("users.json", users_db)
+
+    return {"status": "success", "balance": sender_wallet.balance}