bambuddy/backend/app/models/spoolman_slot_assignment.py

from datetime import datetime

from sqlalchemy import CheckConstraint, DateTime, ForeignKey, Integer, UniqueConstraint, func
from sqlalchemy.orm import Mapped, mapped_column, relationship

from backend.app.core.database import Base


class SpoolmanSlotAssignment(Base):
    """Assignment of a Spoolman spool to a specific AMS slot on a printer.

    Tracks which Spoolman spool ID occupies a given (printer, ams, tray) slot.
    This is the source of truth for Spoolman slot assignments — Spoolman's own
    ``spool.location`` field is NOT managed by Bambuddy and is left for the user.
    """

    __tablename__ = "spoolman_slot_assignments"

    id: Mapped[int] = mapped_column(primary_key=True)
    printer_id: Mapped[int] = mapped_column(ForeignKey("printers.id", ondelete="CASCADE"))
    ams_id: Mapped[int] = mapped_column(Integer)
    tray_id: Mapped[int] = mapped_column(Integer)
    spoolman_spool_id: Mapped[int] = mapped_column(Integer)
    assigned_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())

    printer: Mapped["Printer"] = relationship()

    __table_args__ = (
        UniqueConstraint("printer_id", "ams_id", "tray_id", name="uq_slot_assignment"),
        # 0-7: standard AMS units. 128-191: AMS-HT (each unit uses ams_id 128+,
        # single tray). 255: external / VT tray. Matches the value range the
        # internal `spool_assignment` table accepts. See #1274 — H2C with
        # AMS-HT on the left nozzle reports ams_id=128.
        CheckConstraint(
            "(ams_id >= 0 AND ams_id <= 7) OR (ams_id >= 128 AND ams_id <= 191) OR ams_id = 255",
            name="ck_ams_id_range",
        ),
        CheckConstraint("tray_id >= 0 AND tray_id <= 3", name="ck_tray_id_range"),
    )


from backend.app.models.printer import Printer  # noqa: E402, F401