"""评价模板模型。

6 轴动态评价指标体系的模板配置，支持基金类型、存续期、企业阶段、产业赛道、投资策略、投资人类型的动态组合。
"""

import uuid
from datetime import datetime, timezone

from sqlalchemy import Boolean, DateTime, ForeignKey, Integer, String
from sqlalchemy.orm import Mapped, mapped_column

from app.core.database import Base
from app.core.types import JSONBType


class EvaluationTemplate(Base):
    """评价模板 — 6 轴配置单元，定义维度权重和专属指标。"""

    __tablename__ = "evaluation_templates"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    tenant_id: Mapped[str] = mapped_column(String(36), ForeignKey("tenants.id"), nullable=False, index=True)
    name: Mapped[str] = mapped_column(String(200), nullable=False, comment="模板名称")

    # 6 轴参数
    fund_type: Mapped[str] = mapped_column(String(50), nullable=False, comment="基金类型：angel/early_vc/growth_vc/pe/cvc/fof/distress/esg")
    fund_lifecycle: Mapped[str] = mapped_column(String(50), nullable=False, default="investment", comment="存续期阶段：investment/growth/exit_preparation/liquidation")
    company_stage: Mapped[str] = mapped_column(String(50), nullable=False, default="a", comment="企业阶段：seed/a/b/c/pre_ipo")
    industry: Mapped[str] = mapped_column(String(50), nullable=False, default="ai", comment="产业赛道：ai/saas/hardware/biotech/consumer/fintech/manufacturing")
    strategy: Mapped[str] = mapped_column(String(50), nullable=False, default="growth", comment="投资策略：growth/value/empowerment/turnaround")
    investor_type: Mapped[str] = mapped_column(String(50), nullable=False, default="investor", comment="投资人类型：gp/post_invest_lead/investor")

    # 权重配置 — {dimension_key: weight}，归一化后总和 = 1.0
    weights_json: Mapped[dict] = mapped_column(JSONBType, nullable=False, comment="14 维度权重（归一化后）")

    # 维度裁剪
    enabled_dimensions: Mapped[list] = mapped_column(JSONBType, nullable=False, comment="启用的维度 key 列表")
    disabled_dimensions: Mapped[list] = mapped_column(JSONBType, nullable=False, default=list, comment="禁用的维度 key 列表")

    # 专属指标 — [{key, label, description, data_source}]
    custom_metrics_json: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="赛道专属指标定义")

    # 修饰因子
    lp_focus_metrics: Mapped[list | None] = mapped_column(JSONBType, nullable=True, comment="LP 附加指标列表")
    regional_benchmark: Mapped[str | None] = mapped_column(String(50), nullable=True, comment="地域基准标识：china_mainland/us/sea/europe")

    # 元数据
    is_default: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, comment="是否为该组合的默认模板")
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, comment="是否启用")
    version: Mapped[int] = mapped_column(Integer, nullable=False, default=1, comment="版本号")

    created_by: Mapped[str | None] = mapped_column(String(36), nullable=True, comment="创建人")
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True),
        default=lambda: datetime.now(timezone.utc),
        onupdate=lambda: datetime.now(timezone.utc),
    )