From e97a982d6904d7a2fc29f398e3a90ca0d79d2998 Mon Sep 17 00:00:00 2001 From: yxd92326 Date: Thu, 10 Apr 2025 16:09:47 +0100 Subject: [PATCH 1/4] Remove references to the TomographyProcessingParameters table --- src/murfey/client/multigrid_control.py | 6 --- src/murfey/server/__init__.py | 9 ----- src/murfey/server/api/__init__.py | 37 ------------------- src/murfey/server/demo_api.py | 51 -------------------------- src/murfey/util/db.py | 14 ------- src/murfey/util/models.py | 6 --- 6 files changed, 123 deletions(-) diff --git a/src/murfey/client/multigrid_control.py b/src/murfey/client/multigrid_control.py index ab1bade3d..f578b38dc 100644 --- a/src/murfey/client/multigrid_control.py +++ b/src/murfey/client/multigrid_control.py @@ -411,12 +411,6 @@ def _start_dc(self, json, from_form: bool = False): source = Path(json["source"]) context = self.analysers[source]._context if isinstance(context, TomographyContext): - if from_form: - capture_post( - f"{self._environment.url.geturl()}/clients/{self._environment.client_id}/tomography_processing_parameters", - json=json, - ) - source = Path(json["source"]) context.register_tomography_data_collections( file_extension=json["file_extension"], diff --git a/src/murfey/server/__init__.py b/src/murfey/server/__init__.py index 41e2bf1d7..31bd7a361 100644 --- a/src/murfey/server/__init__.py +++ b/src/murfey/server/__init__.py @@ -190,15 +190,6 @@ def get_job_ids(tilt_series_id: int, appid: int) -> JobIDs: ) -def get_tomo_proc_params(pj_id: int, *args) -> db.TomographyProcessingParameters: - results = murfey_db.exec( - select(db.TomographyProcessingParameters).where( - db.TomographyProcessingParameters.pj_id == pj_id - ) - ).one() - return results - - def get_tomo_preproc_params(dcg_id: int, *args) -> db.TomographyPreprocessingParameters: results = murfey_db.exec( select(db.TomographyPreprocessingParameters).where( diff --git a/src/murfey/server/api/__init__.py b/src/murfey/server/api/__init__.py index 0e786b551..dd2733538 100644 --- a/src/murfey/server/api/__init__.py +++ b/src/murfey/server/api/__init__.py @@ -70,7 +70,6 @@ SPARelionParameters, Tilt, TiltSeries, - TomographyProcessingParameters, ) from murfey.util.models import ( BLSampleImageParameters, @@ -92,7 +91,6 @@ PreprocessingParametersTomo, ProcessingJobParameters, ProcessingParametersSPA, - ProcessingParametersTomo, RegistrationMessage, RsyncerInfo, RsyncerSource, @@ -640,41 +638,6 @@ def register_tomo_preproc_params( _transport_object.send(_transport_object.feedback_queue, zocalo_message) -@router.post("/sessions/{session_id}/tomography_processing_parameters") -def register_tomo_proc_params( - session_id: MurfeySessionID, proc_params: ProcessingParametersTomo, db=murfey_db -): - log.info( - f"Registering tomography processing parameters {sanitise(proc_params.tag)}, {sanitise(proc_params.tilt_series_tag)}" - ) - collected_ids = db.exec( - select( - DataCollectionGroup, - DataCollection, - ProcessingJob, - AutoProcProgram, - ) - .where(DataCollectionGroup.session_id == session_id) - .where(DataCollectionGroup.tag == proc_params.tag) - .where(DataCollection.tag == proc_params.tilt_series_tag) - .where(DataCollection.dcg_id == DataCollectionGroup.id) - .where(ProcessingJob.dc_id == DataCollection.id) - .where(AutoProcProgram.pj_id == ProcessingJob.id) - .where(ProcessingJob.recipe == "em-tomo-preprocess") - ).one() - if not db.exec( - select(func.count(TomographyProcessingParameters.pj_id)).where( - TomographyProcessingParameters.pj_id == collected_ids[2].id - ) - ).one(): - tomogram_params = TomographyProcessingParameters( - pj_id=collected_ids[2].id, manual_tilt_offset=proc_params.manual_tilt_offset - ) - db.add(tomogram_params) - db.commit() - db.close() - - class Tag(BaseModel): tag: str diff --git a/src/murfey/server/demo_api.py b/src/murfey/server/demo_api.py index d953b87ab..105a8d4ad 100644 --- a/src/murfey/server/demo_api.py +++ b/src/murfey/server/demo_api.py @@ -58,7 +58,6 @@ Tilt, TiltSeries, TomographyPreprocessingParameters, - TomographyProcessingParameters, ) from murfey.util.models import ( ClientInfo, @@ -75,7 +74,6 @@ PreprocessingParametersTomo, ProcessingJobParameters, ProcessingParametersSPA, - ProcessingParametersTomo, RegistrationMessage, RsyncerInfo, RsyncerSource, @@ -513,57 +511,8 @@ def register_tomo_preproc_params( gain_ref=proc_params.gain_ref, motion_corr_binning=proc_params.motion_corr_binning, voltage=proc_params.voltage, - # manual_tilt_offset=proc_params.manual_tilt_offset, ) db.add(params) - if not db.exec( - select(func.count(TomographyProcessingParameters.pj_id)).where( - TomographyProcessingParameters.pj_id == collected_ids[2].id - ) - ).one(): - tomogram_params = TomographyProcessingParameters( - pj_id=collected_ids[2].id, manual_tilt_offset=proc_params.manual_tilt_offset - ) - db.add(tomogram_params) - db.commit() - db.close() - - -@router.post("/clients/{client_id}/tomography_processing_parameters") -def register_tomo_proc_params( - client_id: int, proc_params: ProcessingParametersTomo, db=murfey_db -): - client = db.exec( - select(ClientEnvironment).where(ClientEnvironment.client_id == client_id) - ).one() - session_id = client.session_id - log.info( - f"Registering tomography processing parameters {sanitise(proc_params.tag)}, {sanitise(proc_params.tilt_series_tag)}, {session_id}" - ) - collected_ids = db.exec( - select( - DataCollectionGroup, - DataCollection, - ProcessingJob, - AutoProcProgram, - ) - .where(DataCollectionGroup.session_id == session_id) - .where(DataCollectionGroup.tag == proc_params.tag) - .where(DataCollection.tag == proc_params.tilt_series_tag) - .where(DataCollection.dcg_id == DataCollectionGroup.id) - .where(ProcessingJob.dc_id == DataCollection.id) - .where(AutoProcProgram.pj_id == ProcessingJob.id) - .where(ProcessingJob.recipe == "em-tomo-preprocess") - ).one() - if not db.exec( - select(func.count(TomographyProcessingParameters.pj_id)).where( - TomographyProcessingParameters.pj_id == collected_ids[2].id - ) - ).one(): - tomogram_params = TomographyProcessingParameters( - pj_id=collected_ids[2].id, manual_tilt_offset=proc_params.manual_tilt_offset - ) - db.add(tomogram_params) db.commit() db.close() diff --git a/src/murfey/util/db.py b/src/murfey/util/db.py index c494684d9..e2b1892e7 100644 --- a/src/murfey/util/db.py +++ b/src/murfey/util/db.py @@ -448,12 +448,6 @@ class ProcessingJob(SQLModel, table=True): # type: ignore spa_feedback_parameters: List["SPAFeedbackParameters"] = Relationship( back_populates="processing_job", sa_relationship_kwargs={"cascade": "delete"} ) - tomography_processing_parameters: List["TomographyProcessingParameters"] = ( - Relationship( - back_populates="processing_job", - sa_relationship_kwargs={"cascade": "delete"}, - ) - ) ctf_parameters: List["CtfParameters"] = Relationship( back_populates="processing_job", sa_relationship_kwargs={"cascade": "delete"} ) @@ -514,14 +508,6 @@ class TomographyPreprocessingParameters(SQLModel, table=True): # type: ignore ) -class TomographyProcessingParameters(SQLModel, table=True): # type: ignore - pj_id: int = Field(primary_key=True, foreign_key="processingjob.id") - manual_tilt_offset: int - processing_job: Optional[ProcessingJob] = Relationship( - back_populates="tomography_processing_parameters" - ) - - class AutoProcProgram(SQLModel, table=True): # type: ignore id: int = Field(primary_key=True, unique=True) pj_id: int = Field(foreign_key="processingjob.id") diff --git a/src/murfey/util/models.py b/src/murfey/util/models.py index b9a5e60b3..97ed11b7d 100644 --- a/src/murfey/util/models.py +++ b/src/murfey/util/models.py @@ -372,9 +372,3 @@ class Base(BaseModel): gain_ref: Optional[str] manual_tilt_offset: float eer_fractionation: int - - -class ProcessingParametersTomo(BaseModel): - manual_tilt_offset: int - tag: str - tilt_series_tag: str From e7a1cc61bd8eaa65b96dc807c82dfd6e980cd9d0 Mon Sep 17 00:00:00 2001 From: yxd92326 Date: Thu, 10 Apr 2025 16:12:11 +0100 Subject: [PATCH 2/4] Remove manual tilt offset from client --- src/murfey/client/contexts/tomo.py | 6 ------ src/murfey/util/models.py | 2 -- 2 files changed, 8 deletions(-) diff --git a/src/murfey/client/contexts/tomo.py b/src/murfey/client/contexts/tomo.py index 072c0bf36..fb0eb53d9 100644 --- a/src/murfey/client/contexts/tomo.py +++ b/src/murfey/client/contexts/tomo.py @@ -19,7 +19,6 @@ ) from murfey.util import authorised_requests, capture_post, get_machine_config_client from murfey.util.mdoc import get_block, get_global_data, get_num_blocks -from murfey.util.tomo import midpoint logger = logging.getLogger("murfey.client.contexts.tomo") @@ -68,7 +67,6 @@ class TomographyContext(Context): ProcessingParameter( "dose_per_frame", "Dose Per Frame (e- / Angstrom^2 / frame)", default=1 ), - ProcessingParameter("manual_tilt_offset", "Tilt Offset", default=0), ProcessingParameter("gain_ref", "Gain Reference"), ProcessingParameter("eer_fractionation", "EER Fractionation", default=20), ] @@ -566,7 +564,6 @@ def gather_metadata( if environment else None ) - metadata["manual_tilt_offset"] = 0 metadata["source"] = str(self._basepath) except KeyError: return OrderedDict({}) @@ -627,9 +624,6 @@ def gather_metadata( if environment else None ) - mdoc_metadata["manual_tilt_offset"] = -midpoint( - [float(b["TiltAngle"]) for b in blocks] - ) mdoc_metadata["source"] = str(self._basepath) mdoc_metadata["tag"] = str(self._basepath) mdoc_metadata["tilt_series_tag"] = metadata_file.stem diff --git a/src/murfey/util/models.py b/src/murfey/util/models.py index 97ed11b7d..4f3b9f0d4 100644 --- a/src/murfey/util/models.py +++ b/src/murfey/util/models.py @@ -360,7 +360,6 @@ class PreprocessingParametersTomo(BaseModel): image_size_y: int pixel_size_on_image: str motion_corr_binning: int - manual_tilt_offset: float file_extension: str tag: str tilt_series_tag: str @@ -370,5 +369,4 @@ class PreprocessingParametersTomo(BaseModel): class Base(BaseModel): dose_per_frame: Optional[float] gain_ref: Optional[str] - manual_tilt_offset: float eer_fractionation: int From eaad1b663b06cc03b7031cb11e3f7f24c8a3c12e Mon Sep 17 00:00:00 2001 From: yxd92326 Date: Thu, 10 Apr 2025 16:22:05 +0100 Subject: [PATCH 3/4] Rename tomo preprocessing params to just processing params --- src/murfey/client/analyser.py | 8 ++++---- src/murfey/client/multigrid_control.py | 2 +- src/murfey/client/tui/app.py | 2 +- src/murfey/client/tui/screens.py | 12 +++++------- src/murfey/server/__init__.py | 12 ++++++------ src/murfey/server/api/__init__.py | 6 +++--- src/murfey/server/demo_api.py | 14 +++++++------- src/murfey/util/db.py | 6 +++--- src/murfey/util/models.py | 2 +- 9 files changed, 31 insertions(+), 33 deletions(-) diff --git a/src/murfey/client/analyser.py b/src/murfey/client/analyser.py index 877ed5026..104715a10 100644 --- a/src/murfey/client/analyser.py +++ b/src/murfey/client/analyser.py @@ -24,7 +24,7 @@ from murfey.client.tui.forms import FormDependency from murfey.util import Observer, get_machine_config_client from murfey.util.mdoc import get_block -from murfey.util.models import PreprocessingParametersTomo, ProcessingParametersSPA +from murfey.util.models import ProcessingParametersSPA, ProcessingParametersTomo logger = logging.getLogger("murfey.client.analyser") @@ -65,7 +65,7 @@ def __init__( self._environment = environment self._force_mdoc_metadata = force_mdoc_metadata self.parameters_model: ( - Type[ProcessingParametersSPA] | Type[PreprocessingParametersTomo] | None + Type[ProcessingParametersSPA] | Type[ProcessingParametersTomo] | None ) = None self.queue: queue.Queue = queue.Queue() @@ -183,7 +183,7 @@ def _find_context(self, file_path: Path) -> bool: if not self._context: logger.info("Acquisition software: tomo") self._context = TomographyContext("tomo", self._basepath) - self.parameters_model = PreprocessingParametersTomo + self.parameters_model = ProcessingParametersTomo return True # Files with these suffixes belong to the serial EM tomography workflow @@ -207,7 +207,7 @@ def _find_context(self, file_path: Path) -> bool: ): return False self._context = TomographyContext("serialem", self._basepath) - self.parameters_model = PreprocessingParametersTomo + self.parameters_model = ProcessingParametersTomo return True return False diff --git a/src/murfey/client/multigrid_control.py b/src/murfey/client/multigrid_control.py index f578b38dc..ed6362bd7 100644 --- a/src/murfey/client/multigrid_control.py +++ b/src/murfey/client/multigrid_control.py @@ -438,7 +438,7 @@ def _start_dc(self, json, from_form: bool = False): eer_fractionation_file = eer_response.json()["eer_fractionation_file"] json.update({"eer_fractionation_file": eer_fractionation_file}) capture_post( - f"{self._environment.url.geturl()}/sessions/{self._environment.murfey_session}/tomography_preprocessing_parameters", + f"{self._environment.url.geturl()}/sessions/{self._environment.murfey_session}/tomography_processing_parameters", json=json, ) capture_post( diff --git a/src/murfey/client/tui/app.py b/src/murfey/client/tui/app.py index 09bff5271..f2147b0da 100644 --- a/src/murfey/client/tui/app.py +++ b/src/murfey/client/tui/app.py @@ -504,7 +504,7 @@ def _start_dc(self, json, from_form: bool = False): eer_fractionation_file = eer_response.json()["eer_fractionation_file"] json.update({"eer_fractionation_file": eer_fractionation_file}) requests.post( - f"{self.app._environment.url.geturl()}/sessions/{self.app._environment.murfey_session}/tomography_preprocessing_parameters", + f"{self.app._environment.url.geturl()}/sessions/{self.app._environment.murfey_session}/tomography_processing_parameters", json=json, ) capture_post( diff --git a/src/murfey/client/tui/screens.py b/src/murfey/client/tui/screens.py index 0ba3c30a8..e68be2369 100644 --- a/src/murfey/client/tui/screens.py +++ b/src/murfey/client/tui/screens.py @@ -57,7 +57,7 @@ from murfey.client.rsync import RSyncer from murfey.client.tui.forms import FormDependency from murfey.util import capture_post, get_machine_config_client, posix_path, read_config -from murfey.util.models import PreprocessingParametersTomo, ProcessingParametersSPA +from murfey.util.models import ProcessingParametersSPA, ProcessingParametersTomo log = logging.getLogger("murfey.tui.screens") @@ -463,7 +463,7 @@ def compose(self): def _write_params( self, params: dict | None = None, - model: PreprocessingParametersTomo | ProcessingParametersSPA | None = None, + model: ProcessingParametersTomo | ProcessingParametersSPA | None = None, ): if params: try: @@ -473,9 +473,9 @@ def _write_params( for k in analyser._context.user_params + analyser._context.metadata_params: self.app.query_one("#info").write(f"{k.label}: {params.get(k.name)}") self.app._start_dc(params) - if model == PreprocessingParametersTomo: + if model == ProcessingParametersTomo: requests.post( - f"{self.app._environment.url.geturl()}/sessions/{self.app._environment.murfey_session}/tomography_preprocessing_parameters", + f"{self.app._environment.url.geturl()}/sessions/{self.app._environment.murfey_session}/tomography_processing_parameters", json=params, ) elif model == ProcessingParametersSPA: @@ -1156,9 +1156,7 @@ def on_input_changed(self, event): def on_button_pressed(self, event): if self.app._multigrid and self.app._processing_enabled: if self._context == TomographyContext: - valid = validate_form( - self._user_params, PreprocessingParametersTomo.Base - ) + valid = validate_form(self._user_params, ProcessingParametersTomo.Base) else: valid = validate_form(self._user_params, ProcessingParametersSPA.Base) if not valid: diff --git a/src/murfey/server/__init__.py b/src/murfey/server/__init__.py index 31bd7a361..9982bf67c 100644 --- a/src/murfey/server/__init__.py +++ b/src/murfey/server/__init__.py @@ -190,10 +190,10 @@ def get_job_ids(tilt_series_id: int, appid: int) -> JobIDs: ) -def get_tomo_preproc_params(dcg_id: int, *args) -> db.TomographyPreprocessingParameters: +def get_tomo_preproc_params(dcg_id: int, *args) -> db.TomographyProcessingParameters: results = murfey_db.exec( - select(db.TomographyPreprocessingParameters).where( - db.TomographyPreprocessingParameters.dcg_id == dcg_id + select(db.TomographyProcessingParameters).where( + db.TomographyProcessingParameters.dcg_id == dcg_id ) ).one() return results @@ -2537,11 +2537,11 @@ def feedback_callback(header: dict, message: dict) -> None: .where(db.ProcessingJob.recipe == "em-tomo-preprocess") ).one() if not murfey_db.exec( - select(func.count(db.TomographyPreprocessingParameters.dcg_id)).where( - db.TomographyPreprocessingParameters.dcg_id == collected_ids[0].id + select(func.count(db.TomographyProcessingParameters.dcg_id)).where( + db.TomographyProcessingParameters.dcg_id == collected_ids[0].id ) ).one(): - params = db.TomographyPreprocessingParameters( + params = db.TomographyProcessingParameters( dcg_id=collected_ids[0].id, pixel_size=float(message["pixel_size_on_image"]) * 10**10, voltage=message["voltage"], diff --git a/src/murfey/server/api/__init__.py b/src/murfey/server/api/__init__.py index dd2733538..e6ba40eda 100644 --- a/src/murfey/server/api/__init__.py +++ b/src/murfey/server/api/__init__.py @@ -88,9 +88,9 @@ GridSquareParameters, MillingParameters, PostInfo, - PreprocessingParametersTomo, ProcessingJobParameters, ProcessingParametersSPA, + ProcessingParametersTomo, RegistrationMessage, RsyncerInfo, RsyncerSource, @@ -613,9 +613,9 @@ def post_foil_hole( return register_foil_hole(session_id, gs_name, foil_hole_params, db) -@router.post("/sessions/{session_id}/tomography_preprocessing_parameters") +@router.post("/sessions/{session_id}/tomography_processing_parameters") def register_tomo_preproc_params( - session_id: MurfeySessionID, proc_params: PreprocessingParametersTomo, db=murfey_db + session_id: MurfeySessionID, proc_params: ProcessingParametersTomo, db=murfey_db ): session_processing_parameters = db.exec( select(SessionProcessingParameters).where( diff --git a/src/murfey/server/demo_api.py b/src/murfey/server/demo_api.py index 105a8d4ad..bfb82df35 100644 --- a/src/murfey/server/demo_api.py +++ b/src/murfey/server/demo_api.py @@ -57,7 +57,7 @@ SPARelionParameters, Tilt, TiltSeries, - TomographyPreprocessingParameters, + TomographyProcessingParameters, ) from murfey.util.models import ( ClientInfo, @@ -71,9 +71,9 @@ GainReference, GridSquareParameters, PostInfo, - PreprocessingParametersTomo, ProcessingJobParameters, ProcessingParametersSPA, + ProcessingParametersTomo, RegistrationMessage, RsyncerInfo, RsyncerSource, @@ -477,9 +477,9 @@ def register_spa_proc_params( db.commit() -@router.post("/sessions/{session_id}/tomography_preprocessing_parameters") +@router.post("/sessions/{session_id}/tomography_processing_parameters") def register_tomo_preproc_params( - session_id: MurfeySessionID, proc_params: PreprocessingParametersTomo, db=murfey_db + session_id: MurfeySessionID, proc_params: ProcessingParametersTomo, db=murfey_db ): log.info( f"Registering tomography preprocessing parameters {sanitise(proc_params.tag)}, {sanitise(proc_params.tilt_series_tag)}" @@ -500,11 +500,11 @@ def register_tomo_preproc_params( .where(ProcessingJob.recipe == "em-tomo-preprocess") ).one() if not db.exec( - select(func.count(TomographyPreprocessingParameters.dcg_id)).where( - TomographyPreprocessingParameters.dcg_id == collected_ids[0].id + select(func.count(TomographyProcessingParameters.dcg_id)).where( + TomographyProcessingParameters.dcg_id == collected_ids[0].id ) ).one(): - params = TomographyPreprocessingParameters( + params = TomographyProcessingParameters( dcg_id=collected_ids[0].id, pixel_size=proc_params.pixel_size_on_image, dose_per_frame=proc_params.dose_per_frame, diff --git a/src/murfey/util/db.py b/src/murfey/util/db.py index e2b1892e7..c5c65b456 100644 --- a/src/murfey/util/db.py +++ b/src/murfey/util/db.py @@ -378,7 +378,7 @@ class DataCollectionGroup(SQLModel, table=True): # type: ignore back_populates="data_collection_group", sa_relationship_kwargs={"cascade": "delete"}, ) - tomography_preprocessing_parameters: List["TomographyPreprocessingParameters"] = ( + tomography_processing_parameters: List["TomographyProcessingParameters"] = ( Relationship( back_populates="data_collection_group", sa_relationship_kwargs={"cascade": "delete"}, @@ -493,7 +493,7 @@ class SelectionStash(SQLModel, table=True): # type: ignore ) -class TomographyPreprocessingParameters(SQLModel, table=True): # type: ignore +class TomographyProcessingParameters(SQLModel, table=True): # type: ignore dcg_id: int = Field(primary_key=True, foreign_key="datacollectiongroup.id") pixel_size: float dose_per_frame: float @@ -504,7 +504,7 @@ class TomographyPreprocessingParameters(SQLModel, table=True): # type: ignore motion_corr_binning: int = 1 gain_ref: Optional[str] = None data_collection_group: Optional[DataCollectionGroup] = Relationship( - back_populates="tomography_preprocessing_parameters" + back_populates="tomography_processing_parameters" ) diff --git a/src/murfey/util/models.py b/src/murfey/util/models.py index 4f3b9f0d4..7e20c691e 100644 --- a/src/murfey/util/models.py +++ b/src/murfey/util/models.py @@ -349,7 +349,7 @@ class CompletedTiltSeries(BaseModel): rsync_source: str -class PreprocessingParametersTomo(BaseModel): +class ProcessingParametersTomo(BaseModel): dose_per_frame: Optional[float] frame_count: int tilt_axis: float From 783bab331d8256c81e75aeb91559bf4196f49c81 Mon Sep 17 00:00:00 2001 From: yxd92326 Date: Wed, 16 Apr 2025 09:08:50 +0100 Subject: [PATCH 4/4] Rename of tomo_preproc functions --- src/murfey/server/__init__.py | 6 +++--- src/murfey/server/api/__init__.py | 6 +++--- src/murfey/server/demo_api.py | 2 +- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/murfey/server/__init__.py b/src/murfey/server/__init__.py index 68d5e4b6f..92dfc44b3 100644 --- a/src/murfey/server/__init__.py +++ b/src/murfey/server/__init__.py @@ -189,7 +189,7 @@ def get_job_ids(tilt_series_id: int, appid: int) -> JobIDs: ) -def get_tomo_preproc_params(dcg_id: int, *args) -> db.TomographyProcessingParameters: +def get_tomo_proc_params(dcg_id: int, *args) -> db.TomographyProcessingParameters: results = murfey_db.exec( select(db.TomographyProcessingParameters).where( db.TomographyProcessingParameters.dcg_id == dcg_id @@ -1651,7 +1651,7 @@ def _flush_tomography_preprocessing(message: dict): .where(db.DataCollectionGroup.session_id == session_id) .where(db.DataCollectionGroup.tag == message["data_collection_group_tag"]) ).first() - proc_params = get_tomo_preproc_params(collected_ids.id) + proc_params = get_tomo_proc_params(collected_ids.id) if not proc_params: visit_name = message["visit_name"].replace("\r\n", "").replace("\n", "") logger.warning( @@ -2151,7 +2151,7 @@ def feedback_callback(header: dict, message: dict) -> None: ] tilts = get_all_tilts(relevant_tilt_series.id) ids = get_job_ids(relevant_tilt_series.id, alignment_ids[2].id) - preproc_params = get_tomo_preproc_params(ids.dcgid) + preproc_params = get_tomo_proc_params(ids.dcgid) stack_file = ( Path(message["mrc_out"]).parents[3] / "Tomograms" diff --git a/src/murfey/server/api/__init__.py b/src/murfey/server/api/__init__.py index bb6dfe062..08b1cd94c 100644 --- a/src/murfey/server/api/__init__.py +++ b/src/murfey/server/api/__init__.py @@ -42,7 +42,7 @@ get_job_ids, get_machine_config, get_microscope, - get_tomo_preproc_params, + get_tomo_proc_params, sanitise, templates, ) @@ -617,7 +617,7 @@ def post_foil_hole( @router.post("/sessions/{session_id}/tomography_processing_parameters") -def register_tomo_preproc_params( +def register_tomo_proc_params( session_id: MurfeySessionID, proc_params: ProcessingParametersTomo, db=murfey_db ): session_processing_parameters = db.exec( @@ -767,7 +767,7 @@ def register_completed_tilt_series( ] tilts = get_all_tilts(ts.id) ids = get_job_ids(ts.id, collected_ids[3].id) - preproc_params = get_tomo_preproc_params(ids.dcgid) + preproc_params = get_tomo_proc_params(ids.dcgid) first_tilt = db.exec( select(Tilt).where(Tilt.tilt_series_id == ts.id) diff --git a/src/murfey/server/demo_api.py b/src/murfey/server/demo_api.py index a8282936c..01163ec05 100644 --- a/src/murfey/server/demo_api.py +++ b/src/murfey/server/demo_api.py @@ -475,7 +475,7 @@ def register_spa_proc_params( @router.post("/sessions/{session_id}/tomography_processing_parameters") -def register_tomo_preproc_params( +def register_tomo_proc_params( session_id: MurfeySessionID, proc_params: ProcessingParametersTomo, db=murfey_db ): log.info(