135 lines
4.9 KiB
Python
135 lines
4.9 KiB
Python
from datetime import datetime
|
||
from typing import Optional, List
|
||
from sqlalchemy import Column, String, Integer, Float, DateTime, ForeignKey, Enum, Boolean
|
||
from sqlalchemy.sql import func
|
||
from pydantic import BaseModel, Field
|
||
from .database import Base
|
||
import enum
|
||
|
||
class OrderStatus(str, enum.Enum):
|
||
CREATED = "CREATED" # 已创建
|
||
CANCELLED = "CANCELLED" # 已取消
|
||
RECEIVED = "RECEIVED" # 已接单
|
||
DELIVERING = "DELIVERING" # 配送中
|
||
UNPAID = "UNPAID" # 未支付
|
||
COMPLETED = "COMPLETED" # 已完成
|
||
|
||
class DeliveryMethod(str, enum.Enum):
|
||
DELIVERY_AT_DOORSTEP = "DELIVERY_AT_DOORSTEP" # 放在门口
|
||
DELIVERY_TO_ROOM = "DELIVERY_TO_ROOM" # 投递到家
|
||
|
||
# 数据库模型
|
||
class ShippingOrderDB(Base):
|
||
__tablename__ = "shipping_orders"
|
||
|
||
orderid = Column(String(32), primary_key=True)
|
||
userid = Column(Integer, ForeignKey("users.userid"), index=True)
|
||
addressid = Column(Integer, ForeignKey("delivery_addresses.id"), index=True)
|
||
package_count = Column(Integer, nullable=False)
|
||
original_amount = Column(Float, nullable=False)
|
||
coupon_discount_amount = Column(Float, default=0)
|
||
point_discount_amount = Column(Float, default=0)
|
||
coupon_id = Column(Integer, ForeignKey("user_coupons.id"), nullable=True)
|
||
final_amount = Column(Float, nullable=False)
|
||
status = Column(Enum(OrderStatus), nullable=False, default=OrderStatus.CREATED)
|
||
cancel_reason = Column(String(200), nullable=True) # 取消原因
|
||
complete_images = Column(String(1000), nullable=True) # 完成订单的图片URL，多个URL用逗号分隔
|
||
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
||
delivery_method = Column(
|
||
Enum(DeliveryMethod),
|
||
nullable=False,
|
||
default=DeliveryMethod.DELIVERY_AT_DOORSTEP
|
||
)
|
||
deliveryman_user_id = Column(Integer, ForeignKey("users.userid"), nullable=True)
|
||
cancel_user_id = Column(Integer, ForeignKey("users.userid"), nullable=True)
|
||
pay_status = Column(Boolean, default=False) # 支付状态
|
||
prepay_id = Column(String(64)) # 微信支付预支付ID
|
||
pay_time = Column(DateTime(timezone=True)) # 支付时间
|
||
transaction_id = Column(String(64)) # 微信支付交易号
|
||
|
||
class ShippingOrderPackageDB(Base):
|
||
__tablename__ = "shipping_order_packages"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
orderid = Column(String(32), ForeignKey("shipping_orders.orderid"), index=True)
|
||
station_id = Column(Integer, ForeignKey("stations.id"), index=True)
|
||
pickup_codes = Column(String(100), nullable=False)
|
||
create_time = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
||
# Pydantic 模型
|
||
class OrderPackage(BaseModel):
|
||
station_id: int
|
||
pickup_codes: str = Field(..., max_length=100)
|
||
|
||
# 先定义 OrderPriceCalculateRequest
|
||
class OrderPriceCalculateRequest(BaseModel):
|
||
packages: List[OrderPackage]
|
||
|
||
# 然后再定义 OrderCreate
|
||
class OrderCreate(BaseModel):
|
||
addressid: int
|
||
price_request: OrderPriceCalculateRequest
|
||
delivery_method: DeliveryMethod = Field(
|
||
default=DeliveryMethod.DELIVERY_AT_DOORSTEP,
|
||
description="配送方式：放在门口或投递到家"
|
||
)
|
||
|
||
class OrderInfo(BaseModel):
|
||
orderid: str
|
||
userid: int
|
||
addressid: int
|
||
package_count: int
|
||
original_amount: float
|
||
coupon_discount_amount: float
|
||
point_discount_amount: float = 0
|
||
coupon_id: Optional[int]
|
||
final_amount: float
|
||
status: OrderStatus
|
||
complete_images: Optional[List[str]] = None
|
||
create_time: datetime
|
||
delivery_method: DeliveryMethod
|
||
deliveryman_user_id: Optional[int] = None
|
||
cancel_reason: Optional[str] = None
|
||
cancel_user_id: Optional[int] = None
|
||
|
||
def __init__(self, **data):
|
||
super().__init__(**data)
|
||
# 将逗号分隔的图片URL字符串转换为列表
|
||
if self.complete_images and isinstance(self.complete_images, str):
|
||
self.complete_images = self.complete_images.split(",")
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
class OrderPackageInfo(BaseModel):
|
||
id: int
|
||
orderid: str
|
||
station_id: int
|
||
pickup_codes: str
|
||
create_time: datetime
|
||
|
||
class Config:
|
||
from_attributes = True
|
||
|
||
class OrderPriceInfo(BaseModel):
|
||
package_count: int
|
||
original_amount: float
|
||
coupon_discount_amount: float = 0
|
||
points_discount_amount: float = 0
|
||
points_used: Optional[int] = None
|
||
coupon_id: Optional[int] = None
|
||
final_amount: float
|
||
|
||
# 添加取消订单请求模型
|
||
class OrderCancel(BaseModel):
|
||
reason: str = Field(..., max_length=200, description="取消原因")
|
||
|
||
# 完成订单请求模型
|
||
class OrderComplete(BaseModel):
|
||
images: Optional[List[str]] = Field(None, max_items=5) # 最多5张图片，可选
|
||
|
||
class OrderPriceResult(BaseModel):
|
||
"""订单价格计算结果"""
|
||
price_info: OrderPriceInfo
|
||
used_coupon_id: Optional[int] = None # 使用的优惠券ID
|
||
used_points: Optional[int] = None # 使用的积分数 |