"""OKR 模型。

投资人与创始人共同制定 OKR + AI 对齐度评分。
"""

import uuid
from datetime import datetime, timezone

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

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


class OKR(Base):
    """OKR。"""

    __tablename__ = "okrs"

    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)
    quarter: Mapped[str] = mapped_column(String(10), nullable=False, comment="如 2025-Q1")
    objective: Mapped[str] = mapped_column(Text, nullable=False, comment="目标")
    key_results: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="关键结果列表 — 含进度")
    alignment_score: Mapped[float | None] = mapped_column(nullable=True, comment="对齐度评分（0-100）")
    deviation_alerts: Mapped[dict | None] = mapped_column(JSONBType, nullable=True, comment="偏差预警")
    review_notes: Mapped[str | None] = mapped_column(Text, nullable=True, comment="复盘记录")
    status: Mapped[str] = mapped_column(String(20), nullable=False, default="active", comment="active/completed/archived")
    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),
    )