110 lines
5.0 KiB
Python
110 lines
5.0 KiB
Python
from sqlalchemy import Column, String, Integer, DateTime, ForeignKey, Enum, Boolean, Date, Time
|
||
from sqlalchemy.dialects.mysql import DECIMAL
|
||
from sqlalchemy.sql import func
|
||
from pydantic import BaseModel, Field
|
||
from typing import Optional
|
||
from datetime import datetime, date, time
|
||
from .database import Base
|
||
from .merchant_product import DeliveryType, DeliveryTimeType
|
||
import random
|
||
import time as time_module
|
||
import enum
|
||
from sqlalchemy.orm import relationship
|
||
|
||
class MerchantOrderStatus(str, enum.Enum):
|
||
CREATED = "CREATED" # 已创建（待支付）
|
||
PENDING = "PENDING" # 待接单
|
||
DELIVERING = "DELIVERING" # 待配送
|
||
PICKUP_READY = "PICKUP_READY" # 待自提
|
||
COMPLETED = "COMPLETED" # 已完成
|
||
CANCELLED = "CANCELLED" # 已取消
|
||
REFUNDING = "REFUNDING" # 退款中
|
||
REFUNDED = "REFUNDED" # 已退款
|
||
|
||
class MerchantOrderDB(Base):
|
||
__tablename__ = "merchant_orders"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
order_id = Column(String(15), unique=True, nullable=False)
|
||
qrcode_url = Column(String(200)) # 核销码二维码图片地址
|
||
user_id = Column(Integer, ForeignKey("users.userid"), nullable=False)
|
||
merchant_id = Column(Integer, ForeignKey("merchants.id"), nullable=False) # 商家ID
|
||
merchant_product_id = Column(Integer, ForeignKey("merchant_products.id"), nullable=False)
|
||
address_id = Column(Integer, ForeignKey("delivery_addresses.id"), nullable=False) # 收货地址ID
|
||
qty = Column(Integer, nullable=False, default=1) # 购买数量
|
||
unit_price = Column(DECIMAL(10,2), nullable=False) # 产品单价
|
||
order_amount = Column(DECIMAL(10,2), nullable=False)
|
||
pay_amount = Column(DECIMAL(10,2), nullable=False, default=0)
|
||
gift_points = Column(Integer, nullable=False, default=0) # 赠送的积分
|
||
status = Column(Enum(MerchantOrderStatus), nullable=False, default=MerchantOrderStatus.CREATED)
|
||
pay_status = Column(Boolean, nullable=False, default=False)
|
||
order_verify_code = Column(String(21), unique=True, nullable=False)
|
||
verify_time = Column(DateTime(timezone=True), nullable=True)
|
||
verify_user_id = Column(Integer, ForeignKey("users.userid"), nullable=True)
|
||
|
||
# 商品快照数据
|
||
product_delivery_type = Column(Enum(DeliveryType), nullable=True) # 配送类型快照
|
||
product_pickup_place = Column(String(200), nullable=True) # 自提点快照
|
||
product_pickup_time_from = Column(Time, nullable=True) # 自提开始时间快照
|
||
product_pickup_time_to = Column(Time, nullable=True) # 自提结束时间快照
|
||
product_delivery_time_type = Column(Enum(DeliveryTimeType), nullable=True) # 配送时间类型快照
|
||
product_delivery_date = Column(Date, nullable=True) # 配送日期快照
|
||
product_delivery_deadline_time = Column(DateTime(timezone=True), nullable=True) # 配送截止时间快照
|
||
|
||
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
||
update_time = Column(DateTime(timezone=True), onupdate=func.now())
|
||
transaction_id = Column(String(64)) # 微信支付交易号
|
||
pay_time = Column(DateTime(timezone=True), nullable=True)
|
||
refund_transaction_id = Column(String(64)) # 微信退款交易号
|
||
refund_time = Column(DateTime(timezone=True), nullable=True)
|
||
|
||
merchant_product = relationship("MerchantProductDB", backref="merchant_orders")
|
||
merchant = relationship("MerchantDB", backref="merchant_orders")
|
||
address = relationship("AddressDB", backref="merchant_orders")
|
||
|
||
|
||
class MerchantOrderCreate(BaseModel):
|
||
merchant_product_id: int
|
||
qty: int = Field(..., gt=0) # 购买数量，必须大于0
|
||
address_id: int # 收货地址ID
|
||
|
||
class MerchantOrderVerify(BaseModel):
|
||
verify_code: str
|
||
|
||
class MerchantOrderInfo(BaseModel):
|
||
id: int
|
||
order_id: str
|
||
user_id: int
|
||
merchant_id: int # 商家ID
|
||
merchant_product_id: int
|
||
address_id: int # 收货地址ID
|
||
qty: int # 购买数量
|
||
unit_price: float # 产品单价
|
||
order_amount: float
|
||
pay_amount: float
|
||
gift_points: int
|
||
status: MerchantOrderStatus
|
||
order_verify_code: str
|
||
verify_time: Optional[datetime]
|
||
verify_user_id: Optional[int]
|
||
|
||
# 商品快照数据
|
||
product_delivery_type: Optional[DeliveryType] = None # 配送类型快照
|
||
product_pickup_place: Optional[str] = None # 自提点快照
|
||
product_pickup_time_from: Optional[time] = None # 自提开始时间快照
|
||
product_pickup_time_to: Optional[time] = None # 自提结束时间快照
|
||
product_delivery_time_type: Optional[DeliveryTimeType] = None # 配送时间类型快照
|
||
product_delivery_date: Optional[date] = None # 配送日期快照
|
||
product_delivery_deadline_time: Optional[datetime] = None # 配送截止时间快照
|
||
|
||
create_time: datetime
|
||
update_time: Optional[datetime]
|
||
product_name: Optional[str] = None
|
||
product_image: Optional[str] = None
|
||
product_price: Optional[float] = None
|
||
merchant_name: Optional[str] = None
|
||
merchant_address: Optional[str] = None
|
||
|
||
class Config:
|
||
from_attributes = True
|