Skip to content

Commit 6b1e861

Browse files
committed
more alignment
1 parent 59974d7 commit 6b1e861

1 file changed

Lines changed: 22 additions & 10 deletions

File tree

src/murfey/util/db.py

Lines changed: 22 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -969,8 +969,12 @@ class BFactors(SQLModel, table=True): # type: ignore
969969

970970
class MotionCorrection(SQLModel, table=True): # type: ignore
971971
motionCorrectionId: int = Field(primary_key=True, unique=True)
972-
dataCollectionId: Optional[int] = Field(foreign_key="datacollection.id")
973-
autoProcProgramId: Optional[int] = Field(foreign_key="autoprocprogram.id")
972+
dataCollectionId: Optional[int] = Field(
973+
foreign_key="datacollection.dataCollectionId"
974+
)
975+
autoProcProgramId: Optional[int] = Field(
976+
foreign_key="autoprocprogram.autoProcProgramId"
977+
)
974978
imageNumber: Optional[int] = None
975979
firstFrame: Optional[int] = None
976980
lastFrame: Optional[int] = None
@@ -986,7 +990,7 @@ class MotionCorrection(SQLModel, table=True): # type: ignore
986990
fftFullPath: Optional[str] = None
987991
fftCorrectedFullPath: Optional[str] = None
988992
comments: Optional[str] = None
989-
movieId: Optional[int] = Field(foreign_key="movie.murfey_id")
993+
movieId: Optional[int] = Field(foreign_key="movie.movieId")
990994
auto_proc_program: Optional["AutoProcProgram"] = Relationship(
991995
back_populates="motion_correction"
992996
)
@@ -1008,7 +1012,9 @@ class CTF(SQLModel, table=True): # type: ignore
10081012
motionCorrectionId: Optional[int] = Field(
10091013
foreign_key="motioncorrection.motionCorrectionId"
10101014
)
1011-
autoProcProgramId: Optional[int] = Field(foreign_key="autoprocprogram.id")
1015+
autoProcProgramId: Optional[int] = Field(
1016+
foreign_key="autoprocprogram.autoProcProgramId"
1017+
)
10121018
boxSizeX: Optional[float] = None
10131019
boxSizeY: Optional[float] = None
10141020
minResolution: Optional[float] = None
@@ -1031,7 +1037,7 @@ class CTF(SQLModel, table=True): # type: ignore
10311037

10321038
class ParticlePicker(SQLModel, table=True): # type: ignore
10331039
particlePickerId: int = Field(primary_key=True, unique=True)
1034-
programId: Optional[int] = Field(foreign_key="autoprocprogram.id")
1040+
programId: Optional[int] = Field(foreign_key="autoprocprogram.autoProcProgramId")
10351041
firstMotionCorrectionId: Optional[int] = Field(
10361042
foreign_key="motioncorrection.motionCorrectionId"
10371043
)
@@ -1052,8 +1058,12 @@ class ParticlePicker(SQLModel, table=True): # type: ignore
10521058

10531059
class Tomogram(SQLModel, table=True): # type: ignore
10541060
tomogramId: int = Field(primary_key=True, unique=True)
1055-
dataCollectionId: Optional[int] = Field(foreign_key="datacollection.id")
1056-
autoProcProgramId: Optional[int] = Field(foreign_key="autoprocprogram.id")
1061+
dataCollectionId: Optional[int] = Field(
1062+
foreign_key="datacollection.dataCollectionId"
1063+
)
1064+
autoProcProgramId: Optional[int] = Field(
1065+
foreign_key="autoprocprogram.autoProcProgramId"
1066+
)
10571067
volumeFile: Optional[str] = None
10581068
stackFile: Optional[str] = None
10591069
sizeX: Optional[int] = None
@@ -1104,7 +1114,9 @@ class RelativeIceThickness(SQLModel, table=True): # type: ignore
11041114
motionCorrectionId: Optional[int] = Field(
11051115
foreign_key="motioncorrection.motionCorrectionId"
11061116
)
1107-
autoProcProgramId: Optional[int] = Field(foreign_key="autoprocprogram.id")
1117+
autoProcProgramId: Optional[int] = Field(
1118+
foreign_key="autoprocprogram.autoProcProgramId"
1119+
)
11081120
minimum: Optional[float] = None
11091121
q1: Optional[float] = None
11101122
median: Optional[float] = None
@@ -1119,7 +1131,7 @@ class RelativeIceThickness(SQLModel, table=True): # type: ignore
11191131

11201132

11211133
class TiltImageAlignment(SQLModel, table=True): # type: ignore
1122-
movieId: int = Field(foreign_key="movie.murfey_id", primary_key=True)
1134+
movieId: int = Field(foreign_key="movie.movieId", primary_key=True)
11231135
tomogramId: int = Field(foreign_key="tomogram.tomogramId", primary_key=True)
11241136
defocusU: Optional[float] = None
11251137
defocusV: Optional[float] = None
@@ -1139,7 +1151,7 @@ class ParticleClassificationGroup(SQLModel, table=True): # type: ignore
11391151
particlePickerId: Optional[int] = Field(
11401152
foreign_key="particlepicker.particlePickerId"
11411153
)
1142-
programId: Optional[int] = Field(foreign_key="autoprocprogram.id")
1154+
programId: Optional[int] = Field(foreign_key="autoprocprogram.autoProcProgramId")
11431155
type: Optional[str] = Enum("2D", "3D")
11441156
batchNumber: Optional[int] = None
11451157
numberOfParticlesPerBatch: Optional[int] = None

0 commit comments

Comments
 (0)