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

# 数据库模型
class TimePeriodDB(Base):
    """配送时段模型"""
    __tablename__ = "time_periods"

    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(50), nullable=False)  # 时段名称，例如"上午"、"下午"、"晚上"
    from_time = Column(Time, nullable=False)  # 开始时间
    to_time = Column(Time, nullable=False)  # 结束时间
    create_time = Column(DateTime(timezone=True), server_default=func.now())
    update_time = Column(DateTime(timezone=True), onupdate=func.now())

# Pydantic 模型
class TimePeriodCreate(BaseModel):
    name: str = Field(..., max_length=50, description="时段名称")
    from_time: time = Field(..., description="开始时间")
    to_time: time = Field(..., description="结束时间")

class TimePeriodUpdate(BaseModel):
    name: Optional[str] = Field(None, max_length=50, description="时段名称")
    from_time: Optional[time] = Field(None, description="开始时间")
    to_time: Optional[time] = Field(None, description="结束时间")

class TimePeriodInfo(BaseModel):
    id: int
    name: str
    from_time: time
    to_time: time
    create_time: datetime
    update_time: Optional[datetime] = None
    
    class Config:
        from_attributes = True