feat: track designer downloads and notifications

backend/app/api/payment.py

@@ -17,6 +17,7 @@ from app.models.download import DownloadRecord
 from app.ysm_sdk import create_payment, query_order, PaymentNotify
 from app.api.orders import get_current_user
 from app.models.user import User
+from app.services.download_tracker import record_download
 
 router = APIRouter(prefix="/payment", tags=["支付"])
 
@@ -138,19 +139,11 @@ async def payment_notify(
             order.status = OrderStatus.PAID
             order.payment_method = "ysm_wechat"  # 易收米微信支付
             order.paid_at = datetime.now()
-            
-            # 创建下载记录
-            download_record = DownloadRecord(
-                user_id=order.user_id,
-                work_id=order.work_id,
-                order_id=order.id
-            )
-            db.add(download_record)
-            
-            # 更新作品下载次数
+
             work = db.query(Work).filter(Work.id == order.work_id).first()
-            if work:
-                work.downloads += 1
+            buyer = db.query(User).filter(User.id == order.user_id).first()
+            if work and buyer:
+                record_download(db, order, work, buyer)
             
             db.commit()
             
@@ -203,19 +196,10 @@ async def query_order_status(
                 order.status = OrderStatus.PAID
                 order.payment_method = "ysm_wechat"
                 order.paid_at = datetime.now()
-                
-                # 创建下载记录
-                download_record = DownloadRecord(
-                    user_id=order.user_id,
-                    work_id=order.work_id,
-                    order_id=order.id
-                )
-                db.add(download_record)
-                
-                # 更新作品下载次数
+
                 work = db.query(Work).filter(Work.id == order.work_id).first()
                 if work:
-                    work.downloads += 1
+                    record_download(db, order, work, current_user)
                 
                 db.commit()