Skip to content

Commit 8e02e7f

Browse files
authored
Fixes for SPA flushing (#441)
Use the recipes from Murfey configuration rather than hardcoding them. Also, add foil hole ID to preprocessing message during flush.
1 parent 1c5667d commit 8e02e7f

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

src/murfey/server/__init__.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1947,6 +1947,7 @@ def _flush_spa_preprocessing(message: dict):
19471947
machine_config = get_machine_config(instrument_name=instrument_name)[
19481948
instrument_name
19491949
]
1950+
recipe_name = machine_config.recipes.get("em-spa-preprocess", "em-spa-preprocess")
19501951
collected_ids = murfey_db.exec(
19511952
select(
19521953
db.DataCollectionGroup,
@@ -1959,7 +1960,7 @@ def _flush_spa_preprocessing(message: dict):
19591960
.where(db.DataCollection.dcg_id == db.DataCollectionGroup.id)
19601961
.where(db.ProcessingJob.dc_id == db.DataCollection.id)
19611962
.where(db.AutoProcProgram.pj_id == db.ProcessingJob.id)
1962-
.where(db.ProcessingJob.recipe == "em-spa-preprocess")
1963+
.where(db.ProcessingJob.recipe == recipe_name)
19631964
).one()
19641965
params = murfey_db.exec(
19651966
select(db.SPARelionParameters, db.SPAFeedbackParameters)
@@ -1973,7 +1974,7 @@ def _flush_spa_preprocessing(message: dict):
19731974
f"No SPA processing parameters found for client processing job ID {collected_ids[2].id}"
19741975
)
19751976
raise ValueError(
1976-
"No processing parameters were foudn in the database when flushing SPA preprocessing"
1977+
"No processing parameters were found in the database when flushing SPA preprocessing"
19771978
)
19781979

19791980
murfey_ids = _murfey_id(
@@ -2000,7 +2001,7 @@ def _flush_spa_preprocessing(message: dict):
20002001
)
20012002
murfey_db.add(movie)
20022003
zocalo_message: dict = {
2003-
"recipes": ["em-spa-preprocess"],
2004+
"recipes": [recipe_name],
20042005
"parameters": {
20052006
"node_creator_queue": machine_config.node_creator_queue,
20062007
"dcid": collected_ids[1].id,
@@ -2020,6 +2021,7 @@ def _flush_spa_preprocessing(message: dict):
20202021
"particle_diameter": proc_params.particle_diameter or 0,
20212022
"fm_int_file": f.eer_fractionation_file,
20222023
"do_icebreaker_jobs": default_spa_parameters.do_icebreaker_jobs,
2024+
"foil_hole_id": f.foil_hole_id,
20232025
},
20242026
}
20252027
if _transport_object:
@@ -2075,6 +2077,8 @@ def _flush_tomography_preprocessing(message: dict):
20752077
)
20762078
return
20772079

2080+
recipe_name = machine_config.get("em-tomo-preprocess", "em-tomo-preprocess")
2081+
20782082
for f in stashed_files:
20792083
collected_ids = murfey_db.exec(
20802084
select(
@@ -2089,7 +2093,7 @@ def _flush_tomography_preprocessing(message: dict):
20892093
.where(db.DataCollection.tag == f.tag)
20902094
.where(db.ProcessingJob.dc_id == db.DataCollection.id)
20912095
.where(db.AutoProcProgram.pj_id == db.ProcessingJob.id)
2092-
.where(db.ProcessingJob.recipe == "em-tomo-preprocess")
2096+
.where(db.ProcessingJob.recipe == recipe_name)
20932097
).one()
20942098
detached_ids = [c.id for c in collected_ids]
20952099

@@ -2105,7 +2109,7 @@ def _flush_tomography_preprocessing(message: dict):
21052109
)
21062110
murfey_db.add(movie)
21072111
zocalo_message: dict = {
2108-
"recipes": ["em-tomo-preprocess"],
2112+
"recipes": [recipe_name],
21092113
"parameters": {
21102114
"node_creator_queue": machine_config.node_creator_queue,
21112115
"dcid": detached_ids[1],

0 commit comments

Comments
 (0)