Upload 3 files

models_sql.py

@@ -16,10 +16,10 @@ class Wallet(Base):
     __tablename__ = "wallets"
     
     account = Column(String, primary_key=True, index=True)
-    balance = Column(Integer, default=0)         # 用于消费的余额 (充值获得)
-    earn_balance = Column(Integer, default=0)    # 创作者销售收益余额 (别人购买获得)
-    tip_balance = Column(Integer, default=0)     # 【新增】创作者打赏收益余额 (粉丝赞助获得)
-    frozen_balance = Column(Integer, default=0)  # 提现审核冻结中的余额
+    balance = Column(Integer, default=0, nullable=False)         # 用于消费的余额 (充值获得)
+    earn_balance = Column(Integer, default=0, nullable=False)    # 创作者销售收益余额 (别人购买获得)
+    tip_balance = Column(Integer, default=0, nullable=False)     # 【新增】创作者打赏收益余额 (粉丝赞助获得)
+    frozen_balance = Column(Integer, default=0, nullable=False)  # 提现审核冻结中的余额
     
     # 乐观锁版本号，防止并发扣款被击穿
     version = Column(Integer, default=1)

requirements.txt

@@ -1,18 +1,18 @@
-# 🔒 P0安全修复：依赖版本锁定，防止供应链攻击和不兼容更新
-fastapi==0.104.1
-uvicorn==0.24.0
-pydantic==2.5.2
-huggingface_hub==0.19.4
-datasets==2.15.0
-python-multipart==0.0.6
-alibabacloud_dysmsapi20170525==2.0.24
-sqlalchemy==2.0.23
-psycopg2-binary==2.9.9
-httpx==0.25.2
-aiofiles==23.2.1
-# 🚀 P2优化：速率限制
-slowapi==0.1.9
-# 🔒 P0安全增强：bcrypt密码哈希
-bcrypt==4.1.2
-# 🔧 P0修复：升级支付宝SDK，打破旧版死锁，自动适配最新安全加密库
+# 🔒 P0安全修复：依赖版本锁定，防止供应链攻击和不兼容更新
+fastapi==0.104.1
+uvicorn==0.24.0
+pydantic==2.5.2
+huggingface_hub==0.19.4
+datasets==2.15.0
+python-multipart==0.0.6
+alibabacloud_dysmsapi20170525==2.0.24
+sqlalchemy==2.0.23
+psycopg2-binary==2.9.9
+httpx==0.25.2
+aiofiles==23.2.1
+# 🚀 P2优化：速率限制
+slowapi==0.1.9
+# 🔒 P0安全增强：bcrypt密码哈希
+bcrypt==4.1.2
+# 🔧 P0修复：升级支付宝SDK，打破旧版死锁，自动适配最新安全加密库
 python-alipay-sdk>=3.3.0

router_wallet.py

@@ -147,10 +147,10 @@ async def alipay_notify(request: Request, db: Session = Depends(get_db)):
             try:
                 wallet = db.query(Wallet).filter(Wallet.account == account).with_for_update().first()
                 if not wallet:
-                    wallet = Wallet(account=account)
+                    wallet = Wallet(account=account, balance=0, earn_balance=0, tip_balance=0, frozen_balance=0)
                     db.add(wallet)
                 
-                wallet.balance += amount
+                wallet.balance = (wallet.balance or 0) + amount
                 
                 last_tx = db.query(Transaction).filter(Transaction.account == account).order_by(Transaction.created_at.desc()).first()
                 prev_hash = last_tx.tx_hash if last_tx else "GENESIS_HASH"
@@ -191,10 +191,10 @@ async def check_order(order_id: str, account: str = None, db: Session = Depends(
                 
                 wallet = db.query(Wallet).filter(Wallet.account == account).with_for_update().first()
                 if not wallet:
-                    wallet = Wallet(account=account)
+                    wallet = Wallet(account=account, balance=0, earn_balance=0, tip_balance=0, frozen_balance=0)
                     db.add(wallet)
                 
-                wallet.balance += amount
+                wallet.balance = (wallet.balance or 0) + amount
                 
                 last_tx = db.query(Transaction).filter(Transaction.account == account).order_by(Transaction.created_at.desc()).first()
                 prev_hash = last_tx.tx_hash if last_tx else "GENESIS_HASH"
@@ -307,11 +307,11 @@ async def purchase_item(request: Request, req: PurchaseRequest, db: Session = De
             
         seller_wallet = db.query(Wallet).filter(Wallet.account == seller_account).with_for_update().first()
         if not seller_wallet:
-            seller_wallet = Wallet(account=seller_account)
+            seller_wallet = Wallet(account=seller_account, balance=0, earn_balance=0, tip_balance=0, frozen_balance=0)
             db.add(seller_wallet)
             
-        buyer_wallet.balance -= price
-        seller_wallet.earn_balance += price
+        buyer_wallet.balance = (buyer_wallet.balance or 0) - price
+        seller_wallet.earn_balance = (seller_wallet.earn_balance or 0) + price
         
         # 🔄 P7后悔模式：记录购买价格
         new_ownership = Ownership(account=req.account, item_id=req.item_id, price_paid=price)