Skip to content

Commit 22b6ab7

Browse files
committed
Add processing parameters to tables
1 parent 694a0a4 commit 22b6ab7

File tree

2 files changed

+26
-2
lines changed

2 files changed

+26
-2
lines changed

src/murfey/server/api/__init__.py

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -478,6 +478,19 @@ def _parse(ps, i, dcg_id):
478478
def register_spa_proc_params(
479479
session_id: MurfeySessionID, proc_params: ProcessingParametersSPA, db=murfey_db
480480
):
481+
session_processing_parameters = db.exec(
482+
select(SessionProcessingParameters).where(
483+
SessionProcessingParameters.session_id == session_id
484+
)
485+
).all()
486+
if session_processing_parameters:
487+
proc_params.gain_ref = session_processing_parameters[0].gain_ref
488+
proc_params.dose_per_frame = session_processing_parameters[0].dose_per_frame
489+
proc_params.eer_fractionation_file = session_processing_parameters[
490+
0
491+
].eer_fractionation_file
492+
proc_params.symmetry = session_processing_parameters[0].symmetry
493+
481494
zocalo_message = {
482495
"register": "spa_processing_parameters",
483496
**dict(proc_params),
@@ -606,6 +619,18 @@ def post_foil_hole(
606619
def register_tomo_preproc_params(
607620
session_id: MurfeySessionID, proc_params: PreprocessingParametersTomo, db=murfey_db
608621
):
622+
session_processing_parameters = db.exec(
623+
select(SessionProcessingParameters).where(
624+
SessionProcessingParameters.session_id == session_id
625+
)
626+
).all()
627+
if session_processing_parameters:
628+
proc_params.gain_ref = session_processing_parameters[0].gain_ref
629+
proc_params.dose_per_frame = session_processing_parameters[0].dose_per_frame
630+
proc_params.eer_fractionation_file = session_processing_parameters[
631+
0
632+
].eer_fractionation_file
633+
609634
zocalo_message = {
610635
"register": "tomography_processing_parameters",
611636
**dict(proc_params),
@@ -1247,7 +1272,6 @@ async def request_tomography_preprocessing(
12471272
proc_file.eer_fractionation_file = processing_job_parameters[
12481273
0
12491274
].eer_fractionation_file
1250-
proc_file.mc_binning = processing_job_parameters[0].motion_corr_binning
12511275

12521276
zocalo_message: dict = {
12531277
"recipes": [recipe_name],

src/murfey/util/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ class ProcessingParametersSPA(BaseModel):
226226
boxsize: Optional[int]
227227
downscale: bool
228228
small_boxsize: Optional[int]
229-
eer_fractionation: int
229+
eer_fractionation_file: str = ""
230230
particle_diameter: Optional[float]
231231
magnification: Optional[int] = None
232232
total_exposed_dose: Optional[float] = None

0 commit comments

Comments
 (0)