From 24354b937387e1032ce2d2716bf1f31dbe41bc38 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Wed, 22 Jan 2025 22:57:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=9C=A8=E7=BA=BF=E4=B9=B0?=
 =?UTF-8?q?=E5=8D=95=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/endpoints/merchant_pay_order.py | 138 ++++++++++++++++++++++++
 app/main.py                             |   3 +-
 app/models/merchant_pay_order.py        |  56 ++++++++++
 3 files changed, 196 insertions(+), 1 deletion(-)
 create mode 100644 app/api/endpoints/merchant_pay_order.py
 create mode 100644 app/models/merchant_pay_order.py

diff --git a/app/api/endpoints/merchant_pay_order.py b/app/api/endpoints/merchant_pay_order.py
new file mode 100644
index 0000000..d60657c
--- /dev/null
+++ b/app/api/endpoints/merchant_pay_order.py
@@ -0,0 +1,138 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+from typing import Optional
+from app.models.merchant_pay_order import (
+    MerchantPayOrderDB,
+    MerchantPayOrderCreate,
+    MerchantPayOrderInfo,
+    MerchantPayOrderStatus,
+    generate_pay_order_id
+)
+from app.models.merchant import MerchantDB
+from app.models.database import get_db
+from app.api.deps import get_current_user
+from app.models.user import UserDB
+from app.core.response import success_response, error_response, ResponseModel
+from datetime import datetime, timezone
+
+router = APIRouter()
+
+@router.post("", response_model=ResponseModel)
+async def create_pay_order(
+    order: MerchantPayOrderCreate,
+    db: Session = Depends(get_db),
+    current_user: UserDB = Depends(get_current_user)
+):
+    """创建支付订单"""
+    # 检查商家是否存在
+    merchant = db.query(MerchantDB).filter(
+        MerchantDB.id == order.merchant_id
+    ).first()
+    if not merchant:
+        return error_response(code=404, message="商家不存在")
+    
+    # 生成订单号
+    order_id = generate_pay_order_id()
+    
+    # 创建订单
+    db_order = MerchantPayOrderDB(
+        order_id=order_id,
+        merchant_id=order.merchant_id,
+        user_id=current_user.userid,
+        amount=order.amount,
+        gift_points=order.gift_points
+    )
+    
+    try:
+        db.add(db_order)
+        db.commit()
+        db.refresh(db_order)
+        
+        # 获取完整订单信息
+        order = db.query(MerchantPayOrderDB).join(
+            MerchantDB,
+            MerchantPayOrderDB.merchant_id == MerchantDB.id
+        ).filter(
+            MerchantPayOrderDB.id == db_order.id
+        ).first()
+        
+        # 获取商家名称
+        merchant_name = db.query(MerchantDB.name).filter(
+            MerchantDB.id == order.merchant_id
+        ).scalar()
+        
+        return success_response(data=MerchantPayOrderInfo(
+            id=order.id,
+            order_id=order.order_id,
+            merchant_id=order.merchant_id,
+            user_id=order.user_id,
+            amount=float(order.amount),
+            gift_points=float(order.gift_points),
+            status=order.status,
+            create_time=order.create_time,
+            update_time=order.update_time,
+            pay_time=order.pay_time,
+            merchant_name=merchant_name
+        ))
+    except Exception as e:
+        db.rollback()
+        return error_response(code=500, message=f"创建订单失败: {str(e)}")
+
+@router.get("", response_model=ResponseModel)
+async def list_pay_orders(
+    status: Optional[MerchantPayOrderStatus] = None,
+    skip: int = 0,
+    limit: int = 20,
+    db: Session = Depends(get_db),
+    current_user: UserDB = Depends(get_current_user)
+):
+    """获取支付订单列表"""
+    # 构建基础查询
+    query = db.query(MerchantPayOrderDB).join(
+        MerchantDB,
+        MerchantPayOrderDB.merchant_id == MerchantDB.id
+    ).filter(
+        MerchantPayOrderDB.user_id == current_user.userid
+    )
+    
+    # 添加状态筛选
+    if status:
+        query = query.filter(MerchantPayOrderDB.status == status)
+    
+    # 获取总数
+    total = query.count()
+    
+    # 获取分页数据
+    orders = query.order_by(
+        MerchantPayOrderDB.create_time.desc()
+    ).offset(skip).limit(limit).all()
+    
+    # 获取商家名称映射
+    merchant_ids = [order.merchant_id for order in orders]
+    merchant_names = dict(
+        db.query(MerchantDB.id, MerchantDB.name).filter(
+            MerchantDB.id.in_(merchant_ids)
+        ).all()
+    )
+    
+    # 处理返回数据
+    order_list = [
+        MerchantPayOrderInfo(
+            id=order.id,
+            order_id=order.order_id,
+            merchant_id=order.merchant_id,
+            user_id=order.user_id,
+            amount=float(order.amount),
+            gift_points=float(order.gift_points),
+            status=order.status,
+            create_time=order.create_time,
+            update_time=order.update_time,
+            pay_time=order.pay_time,
+            merchant_name=merchant_names.get(order.merchant_id)
+        ) for order in orders
+    ]
+    
+    return success_response(data={
+        "total": total,
+        "items": order_list
+    }) 
\ No newline at end of file
diff --git a/app/main.py b/app/main.py
index fadd751..6299c75 100644
--- a/app/main.py
+++ b/app/main.py
@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from app.api.endpoints import wechat,user, address, community, station, order, coupon, community_building, upload, merchant, merchant_product, merchant_order, point, config, merchant_category, log, account
+from app.api.endpoints import wechat,user, address, community, station, order, coupon, community_building, upload, merchant, merchant_product, merchant_order, point, config, merchant_category, log, account,merchant_pay_order
 from app.models.database import Base, engine
 from fastapi.exceptions import RequestValidationError
 from fastapi.responses import JSONResponse
@@ -44,6 +44,7 @@ app.include_router(merchant.router, prefix="/api/merchant", tags=["商家"])
 app.include_router(merchant_category.router, prefix="/api/merchant-categories", tags=["商家分类"])
 app.include_router(merchant_product.router, prefix="/api/merchant/product", tags=["商家产品"])
 app.include_router(merchant_order.router, prefix="/api/merchant/order", tags=["商家订单"])
+app.include_router(merchant_pay_order.router,prefix="/api/merchant-pay",tags=["商家在线买单"]) 
 app.include_router(upload.router, prefix="/api/upload", tags=["文件上传"])
 app.include_router(config.router, prefix="/api/config", tags=["系统配置"])
 app.include_router(log.router, prefix="/api/logs", tags=["系统日志"])
diff --git a/app/models/merchant_pay_order.py b/app/models/merchant_pay_order.py
new file mode 100644
index 0000000..6b91ee7
--- /dev/null
+++ b/app/models/merchant_pay_order.py
@@ -0,0 +1,56 @@
+from sqlalchemy import Column, String, Integer, DateTime, ForeignKey, Enum, DECIMAL
+from sqlalchemy.sql import func
+from pydantic import BaseModel, Field
+from typing import Optional
+from datetime import datetime
+from .database import Base
+import time
+import enum
+
+class MerchantPayOrderStatus(str, enum.Enum):
+    UNPAID = "UNPAID"  # 未支付
+    PAID = "PAID"      # 已支付
+
+class MerchantPayOrderDB(Base):
+    __tablename__ = "merchant_pay_orders"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    order_id = Column(String(20), unique=True, nullable=False)
+    merchant_id = Column(Integer, ForeignKey("merchants.id"), nullable=False)
+    user_id = Column(Integer, ForeignKey("users.userid"), nullable=False)
+    amount = Column(DECIMAL(10,2), nullable=False)
+    gift_points = Column(DECIMAL(10,1), nullable=False, default=0)
+    status = Column(Enum(MerchantPayOrderStatus), nullable=False, default=MerchantPayOrderStatus.UNPAID)
+    create_time = Column(DateTime(timezone=True), server_default=func.now())
+    update_time = Column(DateTime(timezone=True), onupdate=func.now())
+    pay_time = Column(DateTime(timezone=True), nullable=True)
+
+class MerchantPayOrderCreate(BaseModel):
+    merchant_id: int
+    amount: float = Field(..., gt=0)
+    gift_points: float = Field(0, ge=0)
+
+class MerchantPayOrderInfo(BaseModel):
+    id: int
+    order_id: str
+    merchant_id: int
+    user_id: int
+    amount: float
+    gift_points: float
+    status: MerchantPayOrderStatus
+    create_time: datetime
+    update_time: Optional[datetime]
+    pay_time: Optional[datetime]
+    merchant_name: Optional[str] = None
+    merchant_logo: Optional[str] = None
+    
+    class Config:
+        from_attributes = True
+
+def generate_pay_order_id() -> str:
+    """生成支付订单号：P + 8位日期 + 7位时间戳"""
+    now = datetime.now()
+    date_str = now.strftime('%Y%m%d')
+    # 取时间戳后7位
+    timestamp = str(int(time.time() * 1000))[-7:]
+    return f"P{date_str}{timestamp}" 
\ No newline at end of file