"""RecipeTask and RecipeSubtask models.""" from typing import TYPE_CHECKING, Optional from sqlalchemy import ( Boolean, DECIMAL, Enum, ForeignKey, Index, Integer, JSON, String, Text, UniqueConstraint, ) from sqlalchemy.orm import Mapped, mapped_column, relationship from src.backend.database import Base if TYPE_CHECKING: from src.backend.models.orm.recipe import RecipeVersion # What a task is, declared by whoever writes the recipe instead of guessed from its # contents. The last two are not built yet; they are in the enum from the start # because widening a MySQL enum later is an ALTER on a live table. # 'check' = Controllo OK/NOK, senza quote (#54). TASK_TYPES = ("note", "measure", "drawing", "dxf_compare", "camera_measure", "check") # Types the operator screen treats as "there are quotes to take here". # dxf_compare produce quote come camera_measure (#46) MEASURING_TASK_TYPES = ("measure", "camera_measure", "dxf_compare") # Tipi che entrano nel ciclo di produzione se in_loop non e' dichiarato (#53): # le misure di sempre piu' il controllo OK/NOK. DEFAULT_LOOP_TASK_TYPES = MEASURING_TASK_TYPES + ("check",) def _default_in_loop(context) -> bool: """Default alla creazione di in_loop: dipende dal tipo del task (#53).""" return context.get_current_parameters().get("task_type", "note") in DEFAULT_LOOP_TASK_TYPES class RecipeTask(Base): __tablename__ = "recipe_tasks" id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True) version_id: Mapped[int] = mapped_column( Integer, ForeignKey("recipe_versions.id"), nullable=False ) order_index: Mapped[int] = mapped_column(Integer, nullable=False, default=0) # Declared, not deduced. The old rule - "has subtasks, therefore a measurement" - # meant a measurement task whose quotes had not been entered yet behaved as a # note, so the system acted differently depending on how finished the recipe was. task_type: Mapped[str] = mapped_column( Enum(*TASK_TYPES, name="task_type_enum"), nullable=False, default="note", server_default="note", index=True, ) title: Mapped[str] = mapped_column(String(255), nullable=False) directive: Mapped[Optional[str]] = mapped_column(Text, nullable=True) description: Mapped[Optional[str]] = mapped_column(Text, nullable=True) file_path: Mapped[Optional[str]] = mapped_column(String(500), nullable=True) file_type: Mapped[Optional[str]] = mapped_column( Enum("image", "pdf", name="file_type_enum"), nullable=True ) annotations_json: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True) # The vision graph, serialised by vs-task. Opaque here on purpose: the # server routes it and never interprets it, which is what keeps VisionSuite # out of this container. vision_json: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True) # Cartella immutabile (relativa a upload_path) con recipe.json e i file di # meta della ricetta VisionSuite v2 (#46). vision_recipe_dir: Mapped[Optional[str]] = mapped_column(String(500), nullable=True) # #97: camera e modo di analisi del task, {camera_url, camera_source, modo}; # NULL = camera della stazione corrente, analisi impulsiva. acquisition_json: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True) # #53: il task fa parte del ciclo di produzione (default dal tipo: misure e # check si', gli altri no; un task 'note' si puo' mettere nel ciclo a mano). in_loop: Mapped[bool] = mapped_column( Boolean, nullable=False, default=_default_in_loop, server_default="0" ) # #53: task 'note' compilato alla fine produzione, prima della modale del capoturno. at_close: Mapped[bool] = mapped_column( Boolean, nullable=False, default=False, server_default="0" ) # #55: campi extra del task 'note' ([{key,label,type,default,attachment}]). fields: Mapped[Optional[list]] = mapped_column(JSON, nullable=True) # Relationships version: Mapped["RecipeVersion"] = relationship(back_populates="tasks") subtasks: Mapped[list["RecipeSubtask"]] = relationship( back_populates="task", cascade="all, delete-orphan", lazy="selectin" ) __table_args__ = ( {"mysql_engine": "InnoDB", "mysql_charset": "utf8mb4"}, ) @property def recipe_id(self) -> int | None: """Shortcut: recipe_id via the version relationship.""" if self.version: return self.version.recipe_id return None def __repr__(self) -> str: return f"" class RecipeSubtask(Base): __tablename__ = "recipe_subtasks" id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True) task_id: Mapped[int] = mapped_column( Integer, ForeignKey("recipe_tasks.id", ondelete="CASCADE"), nullable=False ) marker_number: Mapped[int] = mapped_column(Integer, nullable=False) description: Mapped[str] = mapped_column(String(500), nullable=False) measurement_type: Mapped[Optional[str]] = mapped_column(String(100), nullable=True) # Tolerance values nominal: Mapped[Optional[float]] = mapped_column(DECIMAL(12, 6), nullable=True) utl: Mapped[Optional[float]] = mapped_column(DECIMAL(12, 6), nullable=True) # Upper Tolerance Limit uwl: Mapped[Optional[float]] = mapped_column(DECIMAL(12, 6), nullable=True) # Upper Warning Limit lwl: Mapped[Optional[float]] = mapped_column(DECIMAL(12, 6), nullable=True) # Lower Warning Limit ltl: Mapped[Optional[float]] = mapped_column(DECIMAL(12, 6), nullable=True) # Lower Tolerance Limit unit: Mapped[str] = mapped_column(String(20), nullable=False, default="mm") image_path: Mapped[Optional[str]] = mapped_column(String(500), nullable=True) # Which output of the vision graph fills this quote: ".". # A link is an id, not a pointer - the same rule vs-task follows, and for # the same reason: a pointer does not survive being written to disk. vision_output: Mapped[Optional[str]] = mapped_column( String(120), nullable=True ) # Relationships task: Mapped["RecipeTask"] = relationship(back_populates="subtasks") __table_args__ = ( UniqueConstraint("task_id", "marker_number", name="uq_task_marker"), Index("ix_subtask_task_id", "task_id"), {"mysql_engine": "InnoDB", "mysql_charset": "utf8mb4"}, ) def __repr__(self) -> str: return f""