1515 BLSubSample ,
1616)
1717from pydantic import BaseModel
18- from sqlalchemy .exc import OperationalError
18+ from sqlalchemy .exc import NoResultFound , OperationalError
1919from sqlmodel import col , select
2020from werkzeug .utils import secure_filename
2121
@@ -240,7 +240,10 @@ class DCParameters(BaseModel):
240240
241241@router .post ("/visits/{visit_name}/sessions/{session_id}/start_data_collection" )
242242def start_dc (
243- visit_name : str , session_id : MurfeySessionID , dc_params : DCParameters , db : Session = murfey_db
243+ visit_name : str ,
244+ session_id : MurfeySessionID ,
245+ dc_params : DCParameters ,
246+ db : Session = murfey_db ,
244247):
245248 ispyb_proposal_code = visit_name [:2 ]
246249 ispyb_proposal_number = visit_name .split ("-" )[0 ][2 :]
@@ -358,7 +361,9 @@ def register_proc(
358361
359362@spa_router .post ("/sessions/{session_id}/spa_processing_parameters" )
360363def register_spa_proc_params (
361- session_id : MurfeySessionID , proc_params : ProcessingParametersSPA , db : Session = murfey_db
364+ session_id : MurfeySessionID ,
365+ proc_params : ProcessingParametersSPA ,
366+ db : Session = murfey_db ,
362367):
363368 session_processing_parameters = db .exec (
364369 select (SessionProcessingParameters ).where (
@@ -379,7 +384,9 @@ def register_spa_proc_params(
379384 "session_id" : session_id ,
380385 }
381386 if murfey .server ._transport_object :
382- murfey .server ._transport_object .send (murfey .server ._transport_object .feedback_queue , zocalo_message )
387+ murfey .server ._transport_object .send (
388+ murfey .server ._transport_object .feedback_queue , zocalo_message
389+ )
383390 else :
384391 logger .error (
385392 f"Pre-processing was requested for { session_id } "
@@ -402,7 +409,9 @@ def flush_spa_processing(
402409 "tag" : tag .tag ,
403410 }
404411 if murfey .server ._transport_object :
405- murfey .server ._transport_object .send (murfey .server ._transport_object .feedback_queue , zocalo_message )
412+ murfey .server ._transport_object .send (
413+ murfey .server ._transport_object .feedback_queue , zocalo_message
414+ )
406415 return
407416
408417
@@ -431,14 +440,12 @@ async def request_spa_preprocessing(
431440 proc_file : SPAProcessFile ,
432441 db : Session = murfey_db ,
433442):
434- help = ""
435443 instrument_name = (
436444 db .exec (select (Session ).where (Session .id == session_id )).one ().instrument_name
437445 )
438446 machine_config = get_machine_config (instrument_name = instrument_name )[
439447 instrument_name
440448 ]
441- print ("Useful stuff " , Path (proc_file .path ), visit_name )
442449 mrc_out = motion_corrected_mrc (Path (proc_file .path ), visit_name , machine_config )
443450 try :
444451 collected_ids = db .exec (
@@ -471,12 +478,15 @@ async def request_spa_preprocessing(
471478 .one ()[0 ]
472479 .id
473480 )
481+ except NoResultFound :
482+ logger .warning ("No foil hole ID found" )
483+ foil_hole_id = None
474484 except Exception as e :
475485 logger .warning (
476486 f"Foil hole ID not found for foil hole { sanitise (str (proc_file .foil_hole_id ))} : { e } " ,
477487 exc_info = True ,
478488 )
479- foil_hole_id = None
489+ foil_hole_id = None
480490 if proc_params :
481491 detached_ids = [c .id for c in collected_ids ]
482492
@@ -573,7 +583,9 @@ async def request_spa_preprocessing(
573583
574584@tomo_router .post ("/sessions/{session_id}/tomography_processing_parameters" )
575585def register_tomo_proc_params (
576- session_id : MurfeySessionID , proc_params : ProcessingParametersTomo , db : Session = murfey_db
586+ session_id : MurfeySessionID ,
587+ proc_params : ProcessingParametersTomo ,
588+ db : Session = murfey_db ,
577589):
578590 session_processing_parameters = db .exec (
579591 select (SessionProcessingParameters ).where (
@@ -593,7 +605,9 @@ def register_tomo_proc_params(
593605 "session_id" : session_id ,
594606 }
595607 if murfey .server ._transport_object :
596- murfey .server ._transport_object .send (murfey .server ._transport_object .feedback_queue , zocalo_message )
608+ murfey .server ._transport_object .send (
609+ murfey .server ._transport_object .feedback_queue , zocalo_message
610+ )
597611
598612
599613class Source (BaseModel ):
@@ -604,7 +618,10 @@ class Source(BaseModel):
604618 "/visits/{visit_name}/sessions/{session_id}/flush_tomography_processing"
605619)
606620def flush_tomography_processing (
607- visit_name : str , session_id : MurfeySessionID , rsync_source : Source , db : Session = murfey_db
621+ visit_name : str ,
622+ session_id : MurfeySessionID ,
623+ rsync_source : Source ,
624+ db : Session = murfey_db ,
608625):
609626 zocalo_message = {
610627 "register" : "flush_tomography_preprocess" ,
@@ -613,7 +630,9 @@ def flush_tomography_processing(
613630 "data_collection_group_tag" : rsync_source .rsync_source ,
614631 }
615632 if murfey .server ._transport_object :
616- murfey .server ._transport_object .send (murfey .server ._transport_object .feedback_queue , zocalo_message )
633+ murfey .server ._transport_object .send (
634+ murfey .server ._transport_object .feedback_queue , zocalo_message
635+ )
617636 return
618637
619638
@@ -938,7 +957,10 @@ class TiltInfo(BaseModel):
938957
939958@tomo_router .post ("/visits/{visit_name}/sessions/{session_id}/tilt" )
940959async def register_tilt (
941- visit_name : str , session_id : MurfeySessionID , tilt_info : TiltInfo , db : Session = murfey_db
960+ visit_name : str ,
961+ session_id : MurfeySessionID ,
962+ tilt_info : TiltInfo ,
963+ db : Session = murfey_db ,
942964):
943965 def _add_tilt ():
944966 tilt_series_id = (
@@ -1025,7 +1047,9 @@ class BLSampleParameters(BaseModel):
10251047def register_sample (visit_name : str , sample_params : BLSampleParameters ) -> dict :
10261048 record = BLSample ()
10271049 if murfey .server ._transport_object :
1028- return murfey .server ._transport_object .do_insert_sample (record , sample_params .sample_group_id )
1050+ return murfey .server ._transport_object .do_insert_sample (
1051+ record , sample_params .sample_group_id
1052+ )
10291053 return {"success" : False }
10301054
10311055
0 commit comments