Skip to content

Commit af61be2

Browse files
committed
Get 'feedback_queue' from security config or transport object instead of machine config
1 parent 1236de2 commit af61be2

File tree

4 files changed

+25
-14
lines changed

4 files changed

+25
-14
lines changed

src/murfey/server/demo_api.py

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
from murfey.server.api import MurfeySessionID
4141
from murfey.server.api.auth import validate_token
4242
from murfey.server.murfey_db import murfey_db
43-
from murfey.util.config import MachineConfig, from_file
43+
from murfey.util.config import MachineConfig, from_file, security_from_file
4444
from murfey.util.db import (
4545
AutoProcProgram,
4646
ClientEnvironment,
@@ -962,9 +962,26 @@ def flush_spa_processing(
962962
return
963963

964964
detached_ids = [c.id for c in collected_ids]
965-
instrument_name = (
966-
db.exec(select(Session).where(Session.id == session_id)).one().instrument_name
967-
)
965+
try:
966+
instrument_name = (
967+
db.exec(select(Session).where(Session.id == session_id))
968+
.one()
969+
.instrument_name
970+
)
971+
except Exception:
972+
log.error(
973+
f"Unable to find a Murfey session associated with session ID {session_id}"
974+
)
975+
return
976+
977+
# Load the security config
978+
security_config_file = machine_config[instrument_name].security_configuration_path
979+
if not security_config_file:
980+
log.error(
981+
f"No security configuration file set for instrument {instrument_name!r}"
982+
)
983+
return
984+
security_config = security_from_file(security_config_file)
968985

969986
murfey_ids = _murfey_id(
970987
detached_ids[3], db, number=2 * len(stashed_files), close=False
@@ -986,7 +1003,7 @@ def flush_spa_processing(
9861003
zocalo_message = {
9871004
"recipes": ["em-spa-preprocess"],
9881005
"parameters": {
989-
"feedback_queue": machine_config[instrument_name].feedback_queue,
1006+
"feedback_queue": security_config.feedback_queue,
9901007
"node_creator_queue": machine_config[
9911008
instrument_name
9921009
].node_creator_queue,

src/murfey/workflows/clem/align_and_merge.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def submit_cluster_request(
3737

3838
# Load feedback queue
3939
machine_config = get_machine_config()[instrument_name]
40-
feedback_queue: str = machine_config.feedback_queue
40+
feedback_queue: str = messenger.feedback_queue
4141

4242
# Work out session directory from file path
4343
processed_folder = machine_config.processed_directory_name

src/murfey/workflows/clem/process_raw_lifs.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,6 @@
66
from pathlib import Path
77
from typing import Optional
88

9-
from murfey.util.config import get_machine_config
10-
119
try:
1210
from murfey.server.ispyb import TransportManager # Session
1311
except AttributeError:
@@ -43,8 +41,7 @@ def zocalo_cluster_request(
4341
)
4442

4543
# Load machine config to get the feedback queue
46-
machine_config = get_machine_config()
47-
feedback_queue = machine_config[instrument_name].feedback_queue
44+
feedback_queue: str = messenger.feedback_queue
4845

4946
# Send the message
5047
# The keys under "parameters" will populate all the matching fields in {}

src/murfey/workflows/clem/process_raw_tiffs.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,6 @@
66
from pathlib import Path
77
from typing import Optional
88

9-
from murfey.util.config import get_machine_config
10-
119
try:
1210
from murfey.server.ispyb import TransportManager # Session
1311
except AttributeError:
@@ -50,8 +48,7 @@ def zocalo_cluster_request(
5048
metadata = tiff_list[0].parent / "Metadata" / (series_name + ".xlif")
5149

5250
# Load machine config to get the feedback queue
53-
machine_config = get_machine_config()
54-
feedback_queue = machine_config[instrument_name].feedback_queue
51+
feedback_queue: str = messenger.feedback_queue
5552

5653
messenger.send(
5754
"processing_recipe",

0 commit comments

Comments
 (0)