"""Human capacity override audit model."""

from __future__ import annotations

from typing import Any, Optional

from sqlalchemy import BigInteger, Date, Float, ForeignKey, Integer, JSON, String, Text, TIMESTAMP
from sqlalchemy.orm import Mapped, mapped_column

from .base import Base


class HumanCapacityOverride(Base):
    __tablename__ = "human_capacity_overrides"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
    user_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=False, index=True)
    date: Mapped[Any] = mapped_column(Date, nullable=False, index=True)
    override_type: Mapped[str] = mapped_column(String(32), nullable=False)
    added_minutes: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    overtime_type: Mapped[Optional[str]] = mapped_column(String(32), nullable=True)
    max_planned_utilization: Mapped[Optional[float]] = mapped_column(Float, nullable=True)
    work_type_bias: Mapped[dict[str, Any]] = mapped_column(JSON, nullable=False, default=dict)
    high_risk_limit: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
    reason: Mapped[str] = mapped_column(Text, nullable=False)
    expires_at: Mapped[Any] = mapped_column(TIMESTAMP, nullable=False)
    created_by: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=False, index=True)