Upload 2 files

router_tasks.py

@@ -476,13 +476,13 @@ async def update_task(task_id: str, update_data: TaskUpdate, current_user: str =
             
             # 📢 如果有接单人，发送通知
             if task.get("assignee") and changes:
-                add_notification(
-                    task["assignee"],
-                    "task_updated",
-                    f"任务「{task['title']}」已更新：{'、'.join(changes)}",
-                    related_id=task_id,
-                    sender=current_user
-                )
+                add_notification(task.get("assignee"), {
+                    "type": "task_updated",
+                    "from_user": current_user,
+                    "target_item_id": task_id,
+                    "target_item_title": task.get("title", ""),
+                    "content": f"任务《{task.get('title', '')}》已更新：{'、'.join(changes)}"
+                })
             
             return {"status": "success"}
     

router_wallet.py

@@ -23,6 +23,7 @@ from database_sql import get_db
 from models_sql import Wallet, Transaction, Ownership, Refund
 from models import RechargeRequest, WithdrawRequest, PurchaseRequest, TipRequest
 import 数据库连接 as json_db
+from notifications import add_notification
 
 # 🔒 P0安全优化：API限流
 from slowapi import Limiter
@@ -338,6 +339,16 @@ async def purchase_item(request: Request, req: PurchaseRequest, db: Session = De
         # 📝 P2优化：购买审计日志
         logger.info(f"PURCHASE | buyer={req.account} | seller={seller_account} | item={req.item_id} | amount={price} | tx={tx_id}")
         
+        # 🔔 通知卖家作品被购买
+        if seller_account and seller_account != req.account:
+            add_notification(seller_account, {
+                "type": "purchase",
+                "from_user": req.account,
+                "target_item_id": req.item_id,
+                "target_item_title": item.get("title", ""),
+                "content": f"您的作品《{item.get('title', '')}》已被购买"
+            })
+        
         # ☁️ 购买成功后返回网盘密码
         return {
             "status": "success", 
@@ -401,6 +412,24 @@ async def tip_user(request: Request, req: TipRequest, db: Session = Depends(get_
         # 📝 P2优化：打赏审计日志
         logger.info(f"TIP | from={req.sender_account} | to={req.target_account} | amount={req.amount} | item={req.item_id or 'N/A'} | anon={req.is_anonymous}")
         
+        # 🔔 打赏通知（考虑匿名）
+        if not req.is_anonymous:
+            add_notification(req.target_account, {
+                "type": "tip",
+                "from_user": req.sender_account,
+                "target_item_id": req.item_id or "",
+                "target_item_title": "",
+                "content": f"您收到来自 {req.sender_account} 的 {req.amount} 积分打赏"
+            })
+        else:
+            add_notification(req.target_account, {
+                "type": "tip",
+                "from_user": "anonymous",
+                "target_item_id": req.item_id or "",
+                "target_item_title": "",
+                "content": f"您收到了一份 {req.amount} 积分的匿名打赏"
+            })
+        
         # 🚀 核心新增：记录打赏榜单和月度收益趋势 (写入 JSON 以供高频读取)
         users_db = json_db.load_data("users.json", default_data={})
         items_db = json_db.load_data("items.json", default_data=[])
@@ -808,6 +837,16 @@ async def refund_purchase(request: Request, account: str, item_id: str, db: Sess
         
         logger.info(f"REFUND | buyer={account} | seller={seller_account} | item={item_id} | amount={refund_amount} | ban_until={ban_until.isoformat()}")
         
+        # 🔔 通知卖家退款
+        if seller_account:
+            add_notification(seller_account, {
+                "type": "refund",
+                "from_user": account,
+                "target_item_id": item_id,
+                "target_item_title": item.get("title", ""),
+                "content": f"您的作品《{item.get('title', '')}》已被退款"
+            })
+        
         return {
             "status": "success",
             "message": f"退款成功，{refund_amount}积分已退还",