Skip to content

Commit 5be9a35

Browse files
committed
Suppress unwanted warnings
1 parent 50b7aa7 commit 5be9a35

3 files changed

Lines changed: 54 additions & 30 deletions

File tree

src/murfey/server/api/auth.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -110,11 +110,8 @@ async def submit_to_auth_endpoint(
110110
Helper function to forward incoming requests to an authentication server
111111
to verify that they are allowed to inspect the
112112
"""
113-
return None
114113
if security_config.auth_type == "none":
115-
return None
116-
117-
114+
return {"valid": True}
118115

119116
# Forward only essentials auth-related headers
120117
headers = {
@@ -194,10 +191,9 @@ async def validate_instrument_token(
194191
"""
195192
Used by the backend routers to check the incoming instrument server token.
196193
"""
197-
return None
198194
if security_config.instrument_auth_type == "none":
199195
return None
200-
196+
201197
try:
202198
# Validate using auth URL if provided
203199
if security_config.instrument_auth_url:

src/murfey/server/api/workflow.py

Lines changed: 38 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
BLSubSample,
1616
)
1717
from pydantic import BaseModel
18-
from sqlalchemy.exc import OperationalError
18+
from sqlalchemy.exc import NoResultFound, OperationalError
1919
from sqlmodel import col, select
2020
from 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")
242242
def 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")
360363
def 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")
575585
def 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

599613
class Source(BaseModel):
@@ -604,7 +618,10 @@ class Source(BaseModel):
604618
"/visits/{visit_name}/sessions/{session_id}/flush_tomography_processing"
605619
)
606620
def 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")
940959
async 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):
10251047
def 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

src/murfey/server/ispyb.py

Lines changed: 14 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -41,17 +41,21 @@
4141
log = logging.getLogger("murfey.server.ispyb")
4242
security_config = get_security_config()
4343

44-
try:
45-
ISPyBSession = sessionmaker(
46-
bind=create_engine(
47-
url(credentials=security_config.ispyb_credentials),
48-
connect_args={"use_pure": True},
44+
if security_config.ispyb_credentials:
45+
try:
46+
ISPyBSession = sessionmaker(
47+
bind=create_engine(
48+
url(credentials=security_config.ispyb_credentials),
49+
connect_args={"use_pure": True},
50+
)
4951
)
50-
)
51-
log.info("Loaded ISPyB database session")
52-
# Catch all errors associated with loading ISPyB database
53-
except Exception:
54-
log.error("Error loading ISPyB session", exc_info=True)
52+
log.info("Loaded ISPyB database session")
53+
# Catch all errors associated with loading ISPyB database
54+
except Exception:
55+
log.error("Error loading ISPyB session", exc_info=True)
56+
ISPyBSession = lambda: None
57+
else:
58+
log.info("No ISPyB credentials set, using local database", exc_info=True)
5559
ISPyBSession = lambda: None
5660

5761

0 commit comments

Comments
 (0)