from datetime import date, datetime
from typing import Optional

from sqlalchemy import Column
from sqlmodel import Date, SQLModel, Field, INT, DateTime
from pydantic import field_validator

class LotteryDLT(SQLModel, table=True):
    """大乐透彩票数据表"""
    __tablename__ = "lottery_dlt"

    id: int | None = Field(default=None, primary_key=True, index=True)

    issue: str = Field(
        unique=True,
        nullable=False,
        max_length=10,
        sa_column_kwargs={"comment": "彩票期号,唯一"}
    )
    """彩票期号"""

    game_name: str = Field(
        max_length=50,
        nullable=False,
        sa_column_kwargs={"comment": "彩票种类名称"}
    )
    """彩票种类名称"""

    draw_date: date = Field(
        sa_column = Column(Date, nullable=False, comment="开奖日期")
    )
    """开奖日期"""

    draw_time: datetime = Field(
        sa_column=Column(
            DateTime(timezone=True),
            nullable=False,
            comment="开奖时间"
        )
    )
    """开奖时间"""

    # 前区排序后的5个号码
    front_sort_1: Optional[int] = Field(sa_column=Column(INT, comment="前区排序后的号码1"))
    """前区排序后的号码1"""
    front_sort_2: Optional[int] = Field(sa_column=Column(INT, comment="前区排序后的号码2"))
    """前区排序后的号码2"""
    front_sort_3: Optional[int] = Field(sa_column=Column(INT, comment="前区排序后的号码3"))
    """前区排序后的号码3"""
    front_sort_4: Optional[int] = Field(sa_column=Column(INT, comment="前区排序后的号码4"))
    """前区排序后的号码4"""
    front_sort_5: Optional[int] = Field(sa_column=Column(INT, comment="前区排序后的号码5"))
    """前区排序后的号码5"""

    # 后区排序后的2个号码
    back_sort_1: Optional[int] = Field(sa_column=Column(INT, comment="后区排序后的号码1"))
    """后区排序后的号码1"""
    back_sort_2: Optional[int] = Field(sa_column=Column(INT, comment="后区排序后的号码2"))
    """后区排序后的号码2"""

    # 前区5个号码
    front_1: int = Field(sa_column=Column(INT, comment="前区号码1"))
    """前区号码1"""
    front_2: int = Field(sa_column=Column(INT, comment="前区号码2"))
    """前区号码2"""
    front_3: int = Field(sa_column=Column(INT, comment="前区号码3"))
    """前区号码3"""
    front_4: int = Field(sa_column=Column(INT, comment="前区号码4"))
    """前区号码4"""
    front_5: int = Field(sa_column=Column(INT, comment="前区号码5"))
    """前区号码5"""

    # 后区2个号码
    back_1: int = Field(sa_column=Column(INT, comment="后区号码1"))
    """后区号码1"""
    back_2: int = Field(sa_column=Column(INT, comment="后区号码2"))
    """后区号码2"""

    # 元数据字段
    created_at: datetime = Field(
        default_factory=datetime.now,
        sa_column_kwargs={"comment": "创建时间"}
    )
    """创建时间"""

    updated_at: datetime = Field(
        default_factory=datetime.now,
        sa_column_kwargs={"onupdate": datetime.now, "comment": "更新时间"}
    )
    """更新时间"""

    # Pydantic 校验器：确保号码合法性
    @field_validator(
        'front_1', 'front_2', 'front_3', 'front_4', 'front_5',
        'front_sort_1', 'front_sort_2', 'front_sort_3', 'front_sort_4', 'front_sort_5'
    )
    @classmethod
    def validate_front(cls, v):
        if not 1 <= v <= 35:
            raise ValueError(f'前区号码必须在1-35之间，当前值: {v}')
        return v

    @field_validator('back_1', 'back_2', 'back_sort_1', 'back_sort_2')
    @classmethod
    def validate_back(cls, v):
        if not 1 <= v <= 12:
            raise ValueError(f'后区号码必须在1-12之间，当前值: {v}')
        return v