"""干预事件与结果模型。

投后管理 Alpha 归因 — 干预事件 → 指标变化 → 估值影响 → 回报贡献。
"""

import uuid
from datetime import datetime, timezone

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

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


class InterventionEvent(Base):
    """干预事件。"""

    __tablename__ = "intervention_events"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    company_id: Mapped[str] = mapped_column(String(36), ForeignKey("companies.id"), nullable=False, index=True)
    intervention_type: Mapped[str] = mapped_column(String(50), nullable=False, comment="recruitment/customer_intro/strategy/governance/crisis/funding")
    title: Mapped[str] = mapped_column(String(200), nullable=False)
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    executed_by: Mapped[str | None] = mapped_column(String(36), ForeignKey("users.id"), nullable=True)
    executed_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )


class InterventionResult(Base):
    """干预结果。"""

    __tablename__ = "intervention_results"

    id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    intervention_id: Mapped[str] = mapped_column(String(36), ForeignKey("intervention_events.id"), nullable=False, index=True)
    metric_changes: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="指标变化")
    valuation_impact: Mapped[float | None] = mapped_column(Float, nullable=True, comment="估值影响")
    return_contribution: Mapped[float | None] = mapped_column(Float, nullable=True, comment="回报贡献")
    alpha_score: Mapped[float | None] = mapped_column(Float, nullable=True, comment="Alpha 归因评分")
    evidence: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="证据链")
    measured_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
    )