PMFlow/backend/app/models/human_capacity_override.py

29 lines
1.5 KiB
Python

"""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)