"""试验 数据模型"""
from datetime import datetime

from sqlalchemy import Float, ForeignKey, Index, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.core.base_model import MappedBase, ModelMixin, UserMixin


class TrialModel(ModelMixin, UserMixin, MappedBase):
    """试验 主数据表。"""

    __tablename__ = "bre_trial"

    trial_name: Mapped[str] = mapped_column(String(128), nullable=False, comment="试验名称")

    plan_id: Mapped[int | None] = mapped_column(
        Integer, ForeignKey("bre_plan.id", ondelete="SET NULL"), nullable=True, index=True,
        comment="所属育种计划", default=None,
    )

    target_id: Mapped[int | None] = mapped_column(
        Integer, ForeignKey("bre_target.id", ondelete="SET NULL"), nullable=True, index=True,
        comment="服务育种目标", default=None,
    )

    trial_type: Mapped[str | None] = mapped_column(String(32), nullable=True, comment="试验类型", default=None)

    design_type: Mapped[str | None] = mapped_column(String(32), nullable=True, comment="设计类型", default=None)

    start_year: Mapped[int | None] = mapped_column(Integer, nullable=True, comment="起始年份", default=None)

    end_year: Mapped[int | None] = mapped_column(Integer, nullable=True, comment="结束年份", default=None)

    objective: Mapped[str | None] = mapped_column(Text, nullable=True, comment="试验目的", default=None)

    remark: Mapped[str | None] = mapped_column(Text, nullable=True, comment="备注", default=None)

    # 无 status 列：覆盖基类默认 ix_<表>_status_deleted 索引，
    # 仅保留 (created_time, is_deleted) 复合索引用于数据权限过滤。
    __table_args__ = (
        Index("ix_bre_trial_created_deleted", "created_time", "is_deleted"),
    )