add IlluminaFlowCellDevice model

Clinical-Genomics · May 16, 2024 · bf4ce71 · bf4ce71
1 parent 776d338
commit bf4ce71
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/cg/constants/demultiplexing.py b/cg/constants/demultiplexing.py
@@ -212,3 +212,14 @@ class IndexSettings(BaseModel):
     "NovaSeq6000Post1.5Kits": NOVASEQ_6000_POST_1_5_KITS_INDEX_SETTINGS,
     "NoReverseComplements": NO_REVERSE_COMPLEMENTS_INDEX_SETTINGS,
 }
+
+
+class IlluminaFlowCellType(StrEnum):
+    # "10B"
+    # "25B"
+    # "1.5B"
+    # "S4"
+    # "S1"
+    # "SP"
+    # "S2"
+    pass
diff --git a/cg/store/models.py b/cg/store/models.py
@@ -27,6 +27,7 @@
     SexOptions,
     StatusOptions,
 )
+from cg.constants.demultiplexing import IlluminaFlowCellType
 from cg.constants.devices import DeviceType
 from cg.constants.priority import SlurmQos
 from cg.constants.symbols import EMPTY_STRING
@@ -975,6 +976,18 @@ class RunDevice(Base):
     }
 
 
+class IlluminaFlowCellDevice(RunDevice):
+    """Model for storing Illumina flow cells."""
+
+    __tablename__ = "illumina_flow_cell"
+
+    id: Mapped[int] = mapped_column(ForeignKey("run_device.id"), primary_key=True)
+    flow_cell_type: Mapped[IlluminaFlowCellType | None]
+    flow_cell_name: Mapped[UniqueStr64 | None]
+
+    __mapper_args__ = {"polymorphic_identity": DeviceType.ILLUMINA}
+
+
 class RunMetrics(Base):
     """Model for storing run devices."""