TorbatYar/backend/services/healthcare/app/models/appointments.py

"""Appointment scheduling aggregates — Phase 13.1."""
from __future__ import annotations

import uuid
from datetime import datetime, time

from sqlalchemy import Boolean, DateTime, Index, Integer, String, Text, Time, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.types import JSON

from app.core.database import Base
from app.models.base import (
    ActorAuditMixin,
    OptimisticLockMixin,
    SoftDeleteMixin,
    TenantMixin,
    TimestampMixin,
    UUIDPrimaryKeyMixin,
)
from app.models.types import AppointmentStatus, DayOfWeek, GUID, LifecycleStatus


class Schedule(
    Base,
    UUIDPrimaryKeyMixin,
    TenantMixin,
    TimestampMixin,
    SoftDeleteMixin,
    ActorAuditMixin,
):
    __tablename__ = "schedules"

    clinic_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    doctor_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    department_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
    code: Mapped[str] = mapped_column(String(50), nullable=False)
    name: Mapped[str] = mapped_column(String(255), nullable=False)
    day_of_week: Mapped[DayOfWeek] = mapped_column(nullable=False)
    start_time: Mapped[time] = mapped_column(Time, nullable=False)
    end_time: Mapped[time] = mapped_column(Time, nullable=False)
    slot_duration_minutes: Mapped[int] = mapped_column(Integer, default=30, nullable=False)
    is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
    metadata_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)

    __table_args__ = (
        UniqueConstraint("tenant_id", "code", name="uq_schedules_tenant_code"),
        Index("ix_schedules_doctor", "tenant_id", "doctor_id"),
        Index("ix_schedules_clinic", "tenant_id", "clinic_id"),
    )


class AvailabilityWindow(
    Base,
    UUIDPrimaryKeyMixin,
    TenantMixin,
    TimestampMixin,
    SoftDeleteMixin,
    ActorAuditMixin,
):
    __tablename__ = "availability_windows"

    clinic_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    doctor_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    code: Mapped[str] = mapped_column(String(50), nullable=False)
    starts_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    ends_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    is_bookable: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
    metadata_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)

    __table_args__ = (
        UniqueConstraint("tenant_id", "code", name="uq_availability_windows_tenant_code"),
        Index("ix_availability_windows_doctor", "tenant_id", "doctor_id"),
        Index("ix_availability_windows_starts", "tenant_id", "starts_at"),
    )


class AppointmentType(
    Base,
    UUIDPrimaryKeyMixin,
    TenantMixin,
    TimestampMixin,
    SoftDeleteMixin,
    ActorAuditMixin,
):
    __tablename__ = "appointment_types"

    clinic_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    code: Mapped[str] = mapped_column(String(50), nullable=False)
    name: Mapped[str] = mapped_column(String(255), nullable=False)
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    default_duration_minutes: Mapped[int] = mapped_column(Integer, default=30, nullable=False)
    status: Mapped[LifecycleStatus] = mapped_column(
        default=LifecycleStatus.ACTIVE, nullable=False
    )

    __table_args__ = (
        UniqueConstraint("tenant_id", "code", name="uq_appointment_types_tenant_code"),
        Index("ix_appointment_types_clinic", "tenant_id", "clinic_id"),
    )


class Appointment(
    Base,
    UUIDPrimaryKeyMixin,
    TenantMixin,
    TimestampMixin,
    SoftDeleteMixin,
    ActorAuditMixin,
    OptimisticLockMixin,
):
    __tablename__ = "appointments"

    clinic_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    doctor_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    patient_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    department_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
    appointment_type_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
    code: Mapped[str] = mapped_column(String(50), nullable=False)
    status: Mapped[AppointmentStatus] = mapped_column(
        default=AppointmentStatus.DRAFT, nullable=False
    )
    scheduled_start: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    scheduled_end: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    notes: Mapped[str | None] = mapped_column(Text, nullable=True)
    status_reason: Mapped[str | None] = mapped_column(String(500), nullable=True)
    status_changed_at: Mapped[datetime | None] = mapped_column(
        DateTime(timezone=True), nullable=True
    )
    reminder_scheduled: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
    metadata_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)

    __table_args__ = (
        UniqueConstraint("tenant_id", "code", name="uq_appointments_tenant_code"),
        Index("ix_appointments_tenant_status", "tenant_id", "status"),
        Index("ix_appointments_doctor", "tenant_id", "doctor_id"),
        Index("ix_appointments_patient", "tenant_id", "patient_id"),
        Index("ix_appointments_scheduled_start", "tenant_id", "scheduled_start"),
    )


class WaitingListEntry(
    Base,
    UUIDPrimaryKeyMixin,
    TenantMixin,
    TimestampMixin,
    SoftDeleteMixin,
    ActorAuditMixin,
):
    __tablename__ = "waiting_list_entries"

    clinic_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    doctor_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
    patient_id: Mapped[uuid.UUID] = mapped_column(GUID(), nullable=False)
    appointment_type_id: Mapped[uuid.UUID | None] = mapped_column(GUID(), nullable=True)
    code: Mapped[str] = mapped_column(String(50), nullable=False)
    preferred_date: Mapped[str | None] = mapped_column(String(32), nullable=True)
    notes: Mapped[str | None] = mapped_column(Text, nullable=True)
    priority: Mapped[int] = mapped_column(Integer, default=100, nullable=False)

    __table_args__ = (
        UniqueConstraint("tenant_id", "code", name="uq_waiting_list_entries_tenant_code"),
        Index("ix_waiting_list_entries_clinic", "tenant_id", "clinic_id"),
        Index("ix_waiting_list_entries_patient", "tenant_id", "patient_id"),
    )