diff --git a/src/murfey/util/db.py b/src/murfey/util/db.py index 56b891d78..86d1d6f2a 100644 --- a/src/murfey/util/db.py +++ b/src/murfey/util/db.py @@ -379,6 +379,12 @@ class ProcessingJob(SQLModel, table=True): # type: ignore refine3ds: List["Refine3D"] = Relationship( back_populates="processing_job", sa_relationship_kwargs={"cascade": "delete"} ) + bfactor_parameters: List["BFactorParameters"] = Relationship( + back_populates="processing_job", sa_relationship_kwargs={"cascade": "delete"} + ) + bfactors: List["BFactors"] = Relationship( + back_populates="processing_job", sa_relationship_kwargs={"cascade": "delete"} + ) class PreprocessStash(SQLModel, table=True): # type: ignore @@ -810,6 +816,9 @@ class BFactorParameters(SQLModel, table=True): # type: ignore class_number: int mask_file: str run: bool = True + processing_job: Optional[ProcessingJob] = Relationship( + back_populates="bfactor_parameters" + ) class BFactors(SQLModel, table=True): # type: ignore @@ -817,6 +826,7 @@ class BFactors(SQLModel, table=True): # type: ignore pj_id: int = Field(primary_key=True, foreign_key="processingjob.processingJobId") number_of_particles: int resolution: float + processing_job: Optional[ProcessingJob] = Relationship(back_populates="bfactors") class MotionCorrection(SQLModel, table=True): # type: ignore