app/models/procedure.py

import uuid
import datetime
from decimal import Decimal
from typing import Optional, List, TYPE_CHECKING
from sqlmodel import Field, SQLModel, Relationship
from .enums import GenderEnum, InternmentTypeEnum, AccessRouteEnum, ComplicationEnum
if TYPE_CHECKING:
    from .user import User
    from .procedure_type import ProcedureType
    from .team import Team

class Procedure(SQLModel, table=True):
    """
    Represents a row in the `procedures` table.
    This logs the actual procedures performed.
    """
    __tablename__ = "procedures"

    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
    
    # Foreign Keys
    doctor_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
    procedure_type_id: int = Field(foreign_key="procedure_types.id", nullable=False)
    team_id: uuid.UUID = Field(foreign_key="teams.id")
    
    # Patient Data
    patient_name: str = Field(nullable=False)
    patient_age: Optional[int] = Field(default=None)
    patient_gender: Optional[GenderEnum] = Field(default=None)
    exam_number: Optional[str] = Field(default=None, index=True)
    
    # Clinical Data
    internment_type: Optional[InternmentTypeEnum] = Field(default=None)
    access_route: Optional[AccessRouteEnum] = Field(default=None)
    complication: Optional[ComplicationEnum] = Field(default=None)
    description: Optional[str] = Field(default=None, max_length=255)
    
    # Financial/Audit Data
    value_at_time: Decimal = Field(default=0, max_digits=10, decimal_places=2)
    date: datetime.date = Field(nullable=False, index=True)
    created_at: datetime.datetime = Field(default_factory=lambda: datetime.datetime.now(datetime.timezone.utc))
    updated_at: Optional[datetime.datetime] = Field(default=None)

    # Relationships
    doctor: "User" = Relationship(back_populates="procedures")
    procedure_type: "ProcedureType" = Relationship(back_populates="procedures")
    team: "Team" = Relationship(back_populates="procedures")