bambuddy/backend/app/schemas/slicer_pipeline.py

"""Pydantic schemas for the Slicer Pipeline API (#1425, PR A).

A pipeline bundles printer / process / filament(s) / bed-type picks under a
reusable name. PR A surfaces only the bundle; target_kind / target_printer_id /
target_model_class / fanout_strategy are persisted but the API treats them as
opaque defaults — they come alive in PR B (single-target dispatch) and PR C
(multi-copy + class targeting + fanout).
"""

from datetime import datetime
from typing import Literal

from pydantic import BaseModel, Field, field_validator

from backend.app.schemas.not_null import IgnoresNullForRequiredColumns
from backend.app.schemas.slicer import PresetRef, canonical_bed_type, checked_bed_type


class SlicerPipelineBase(BaseModel):
    """Fields editable on create + update."""

    name: str = Field(..., min_length=1, max_length=200)
    description: str | None = Field(default=None, max_length=1000)

    printer_preset: PresetRef
    process_preset: PresetRef
    filament_presets: list[PresetRef] = Field(
        ...,
        min_length=1,
        description="One PresetRef per AMS slot. Order matches the source plate's filament-slot order.",
    )
    bed_type: str | None = Field(default=None, max_length=64)

    @field_validator("bed_type")
    @classmethod
    def slicer_bed_type(cls, value: str | None) -> str | None:
        # A row saved with an old name reads back as the slicer's name (#3359).
        return canonical_bed_type(value)


class SlicerPipelineCreate(SlicerPipelineBase):
    """Payload for POST /slicer-pipelines."""

    @field_validator("bed_type")
    @classmethod
    def slicer_bed_type(cls, value: str | None) -> str | None:
        return checked_bed_type(value)


class SlicerPipelineUpdate(IgnoresNullForRequiredColumns):
    """Payload for PUT /slicer-pipelines/{id}. All fields optional; only those
    present are written. Preset and filament list are replaced wholesale when
    set (we don't support partial filament-slot edits)."""

    __orm_table__ = "slicer_pipelines"

    name: str | None = Field(default=None, min_length=1, max_length=200)
    description: str | None = Field(default=None, max_length=1000)
    printer_preset: PresetRef | None = None
    process_preset: PresetRef | None = None
    filament_presets: list[PresetRef] | None = Field(default=None, min_length=1)
    bed_type: str | None = Field(default=None, max_length=64)

    @field_validator("bed_type")
    @classmethod
    def slicer_bed_type(cls, value: str | None) -> str | None:
        return checked_bed_type(value)

    # PR B target binding. ``target_kind='specific_printer'`` requires
    # ``target_printer_id`` to be set OR cleared in the same payload (route
    # handler enforces). ``target_kind='printer_class'`` is wired by PR C
    # together with ``target_model_class`` (a Bambu model code like 'X1C')
    # and the fanout strategy that distributes copies across matching
    # printers.
    target_kind: Literal["specific_printer", "printer_class"] | None = None
    target_printer_id: int | None = None
    target_model_class: str | None = Field(default=None, max_length=20)
    fanout_strategy: Literal["max_parallel", "fill_one_first", "round_robin"] | None = None


class SlicerPipelineResponse(SlicerPipelineBase):
    """A single pipeline as returned by the API."""

    id: int
    created_by: int | None
    created_at: datetime
    updated_at: datetime

    # Echoed for PR B+ readiness; PR A always returns the persisted defaults.
    target_kind: Literal["specific_printer", "printer_class"] = "printer_class"
    target_printer_id: int | None = None
    target_model_class: str | None = None
    fanout_strategy: Literal["max_parallel", "fill_one_first", "round_robin"] = "max_parallel"

    model_config = {"from_attributes": True}


class SlicerPipelineListResponse(BaseModel):
    """Wraps the list so the response stays additive when run/job counts get
    surfaced in PR B+ (e.g. a ``meta`` field for last-run timestamps)."""

    pipelines: list[SlicerPipelineResponse] = []