from datetime import datetime
from typing import Optional, List
from sqlalchemy import Column, String, Integer, Float, DateTime, ForeignKey
from sqlalchemy.sql import func
from pydantic import BaseModel, Field
from .database import Base

# 数据库模型
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)
    coupon_id = Column(Integer, ForeignKey("user_coupons.id"), nullable=True)
    final_amount = Column(Float, nullable=False)
    create_time = Column(DateTime(timezone=True), server_default=func.now())

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)

class OrderCreate(BaseModel):
    addressid: int
    package_count: int = Field(..., gt=0)
    original_amount: float = Field(..., ge=0)
    coupon_id: Optional[int] = None
    packages: List[OrderPackage]

class OrderInfo(BaseModel):
    orderid: str
    userid: int
    addressid: int
    package_count: int
    original_amount: float
    coupon_discount_amount: float
    coupon_id: Optional[int]
    final_amount: float
    create_time: datetime
    
    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

def generate_order_id() -> str:
    """生成订单号：日期+时间戳"""
    now = datetime.now()
    date_str = now.strftime('%Y%m%d')
    timestamp = int(now.timestamp() * 1000)
    return f"{date_str}{timestamp}"