Pre-reagent/submission association update.

src/submissions/backend/db/models/controls.py

@@ -71,6 +71,8 @@ class Control(BaseClass):
     refseq_version = Column(String(16)) #: version of refseq used in fastq parsing
     kraken2_version = Column(String(16)) #: version of kraken2 used in fastq parsing
     kraken2_db_version = Column(String(32)) #: folder name of kraken2 db
+    sample = relationship("BacterialCultureSample", back_populates="control")
+    sample_id = Column(INTEGER, ForeignKey("_samples.id", ondelete="SET NULL", name="cont_BCS_id"))
 
     def __repr__(self) -> str:
         return f"<Control({self.name})>"
@@ -243,3 +245,8 @@ class Control(BaseClass):
             case _:
                 pass
         return query_return(query=query, limit=limit)
+
+    def save(self):
+        self.__database_session__.add(self)
+        self.__database_session__.commit()
+

src/submissions/backend/db/models/submissions.py

@@ -58,7 +58,7 @@ class BasicSubmission(BaseClass):
     run_cost = Column(FLOAT(2)) #: total cost of running the plate. Set from constant and mutable kit costs at time of creation.
     uploaded_by = Column(String(32)) #: user name of person who submitted the submission to the database.
     comment = Column(JSON) #: user notes
-    submission_category = Column(String(64)) #: ["Research", "Diagnostic", "Surveillance"], else defaults to submission_type_name
+    submission_category = Column(String(64)) #: ["Research", "Diagnostic", "Surveillance", "Validation"], else defaults to submission_type_name
 
     submission_sample_associations = relationship(
         "SubmissionSampleAssociation",
@@ -1483,6 +1483,7 @@ class BacterialCultureSample(BasicSample):
     # id = Column(INTEGER, ForeignKey('basicsample.id'), primary_key=True)
     organism = Column(String(64)) #: bacterial specimen
     concentration = Column(String(16)) #: sample concentration
+    control = relationship("Control", back_populates="sample", uselist=False)
     __mapper_args__ = {"polymorphic_identity": "Bacterial Culture Sample", "polymorphic_load": "inline"}
 
     def to_sub_dict(self, submission_rsl:str) -> dict:
@@ -1496,6 +1497,13 @@ class BacterialCultureSample(BasicSample):
         sample['name'] = f"{self.submitter_id} - ({self.organism})"
         return sample
 
+    def to_hitpick(self, submission_rsl: str | None = None) -> dict | None:
+        sample = super().to_hitpick(submission_rsl)
+        if self.control != None:
+            sample['colour'] = [0,128,0]
+            sample['tooltip'] += f"<br>- Control: {self.control.controltype.name} - {self.control.controltype.targets}"
+        return sample
+
 # Submission to Sample Associations
 
 class SubmissionSampleAssociation(BaseClass):

src/submissions/backend/validators/pydant.py

@@ -325,7 +325,7 @@ class PydSubmission(BaseModel, extra='allow'):
     @field_validator("submission_category")
     @classmethod
     def rescue_category(cls, value, values):
-        if value['value'] not in ["Research", "Diagnostic", "Surveillance"]:
+        if value['value'] not in ["Research", "Diagnostic", "Surveillance", "Validation"]:
             value['value'] = values.data['submission_type']['value']
         return value