mirror of
https://github.com/maziggy/bambuddy.git
synced 2026-08-11 00:30:12 -04:00
70 lines
3.5 KiB
Python
70 lines
3.5 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, ForeignKey, Integer, String, func
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from backend.app.core.database import Base
|
|
|
|
# Canonical VP mode values. The legacy values `immediate` (→ archive) and
|
|
# `print_queue` (→ queue) shipped before the UI labels were aligned with the
|
|
# wire format. `normalize_vp_mode()` translates input from either form and
|
|
# the DB migration in `core/database.py` rewrites existing rows once at boot.
|
|
VP_MODE_ARCHIVE = "archive"
|
|
VP_MODE_REVIEW = "review"
|
|
VP_MODE_QUEUE = "queue"
|
|
VP_MODE_PROXY = "proxy"
|
|
VP_MODE_VALUES = (VP_MODE_ARCHIVE, VP_MODE_REVIEW, VP_MODE_QUEUE, VP_MODE_PROXY)
|
|
|
|
# Legacy → canonical map. Kept narrow on purpose so unrelated typos surface
|
|
# instead of getting silently re-pointed at a default.
|
|
_VP_MODE_ALIASES = {
|
|
"immediate": VP_MODE_ARCHIVE,
|
|
"print_queue": VP_MODE_QUEUE,
|
|
}
|
|
|
|
|
|
def normalize_vp_mode(value: str | None) -> str | None:
|
|
"""Map legacy wire values (`immediate`, `print_queue`) to canonical names.
|
|
|
|
Returns `None` unchanged so callers can decide whether to apply a default.
|
|
Returns unknown values unchanged so validators still see them and reject.
|
|
"""
|
|
if value is None:
|
|
return None
|
|
return _VP_MODE_ALIASES.get(value, value)
|
|
|
|
|
|
class VirtualPrinter(Base):
|
|
"""Virtual printer configuration for multi-instance support."""
|
|
|
|
__tablename__ = "virtual_printers"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
name: Mapped[str] = mapped_column(String(100), default="Bambuddy")
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
mode: Mapped[str] = mapped_column(String(20), default=VP_MODE_ARCHIVE) # archive|review|queue|proxy
|
|
auto_dispatch: Mapped[bool] = mapped_column(Boolean, server_default="true") # queue mode: auto-start or manual
|
|
queue_force_color_match: Mapped[bool] = mapped_column(
|
|
Boolean, server_default="false"
|
|
) # queue mode: pin per-slot type+color from the 3MF onto the queue
|
|
# item so the scheduler refuses to dispatch onto a printer with the wrong
|
|
# filament loaded (#1188).
|
|
gcode_injection: Mapped[bool] = mapped_column(
|
|
Boolean, server_default="false"
|
|
) # queue mode: opt this VP's Send/Print jobs into per-model G-code snippet
|
|
# injection (#1516). Default off so existing gcode_snippets users don't
|
|
# silently start injecting; no-op when no snippets exist for the model.
|
|
model: Mapped[str | None] = mapped_column(String(50), nullable=True) # SSDP model code (server mode)
|
|
access_code: Mapped[str | None] = mapped_column(String(8), nullable=True) # 8 chars (server mode)
|
|
target_printer_id: Mapped[int | None] = mapped_column(
|
|
Integer, ForeignKey("printers.id", ondelete="SET NULL"), nullable=True
|
|
) # proxy mode
|
|
bind_ip: Mapped[str | None] = mapped_column(String(45), nullable=True) # dedicated IP (proxy mode)
|
|
remote_interface_ip: Mapped[str | None] = mapped_column(String(45), nullable=True) # SSDP advertise IP
|
|
tailscale_disabled: Mapped[bool] = mapped_column(
|
|
Boolean, server_default="true"
|
|
) # opt-in: user must explicitly enable; auto-detect only runs then
|
|
serial_suffix: Mapped[str] = mapped_column(String(9), default="391800001") # unique per printer
|
|
position: Mapped[int] = mapped_column(Integer, default=0)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())
|