Skip to content

Commit 0d457eb

Browse files
authored
Allow security configuration path to be specified within machine configuration (#365)
Co-authored-by: @stephen-riggs
1 parent dca23bb commit 0d457eb

File tree

5 files changed

+127
-120
lines changed

5 files changed

+127
-120
lines changed

src/murfey/cli/inject_spa_processing.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
from murfey.server.ispyb import TransportManager
1212
from murfey.server.murfey_db import url
13-
from murfey.util.config import get_machine_config, get_microscope
13+
from murfey.util.config import get_machine_config, get_microscope, get_security_config
1414
from murfey.util.db import (
1515
AutoProcProgram,
1616
ClientEnvironment,
@@ -97,12 +97,13 @@ def run():
9797
os.environ["BEAMLINE"] = args.microscope
9898

9999
machine_config = get_machine_config()
100+
security_config = get_security_config()
100101
_url = url(machine_config)
101102
engine = create_engine(_url)
102103
murfey_db = Session(engine)
103104

104105
_transport_object = TransportManager(args.transport)
105-
_transport_object.feedback_queue = machine_config.feedback_queue
106+
_transport_object.feedback_queue = security_config.feedback_queue
106107

107108
query = (
108109
select(Movie)
@@ -182,7 +183,7 @@ def run():
182183
zocalo_message = {
183184
"recipes": ["em-spa-preprocess"],
184185
"parameters": {
185-
"feedback_queue": machine_config.feedback_queue,
186+
"feedback_queue": _transport_object.feedback_queue,
186187
"node_creator_queue": machine_config.node_creator_queue,
187188
"dcid": detached_ids[1],
188189
"kv": proc_params["voltage"],

src/murfey/cli/spa_ispyb_messages.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from murfey.server.ispyb import Session, TransportManager, get_session_id
2323
from murfey.server.murfey_db import url
2424
from murfey.util import db
25-
from murfey.util.config import get_machine_config, get_microscope
25+
from murfey.util.config import get_machine_config, get_microscope, get_security_config
2626

2727

2828
def run():
@@ -341,6 +341,7 @@ def run():
341341
.where(db.ProcessingJob.recipe == "em-spa-preprocess")
342342
).one()
343343
machine_config = get_machine_config()
344+
security_config = get_security_config()
344345
params = db.SPARelionParameters(
345346
pj_id=collected_ids[2].id,
346347
angpix=float(metadata["pixel_size_on_image"]) * 1e10,
@@ -377,7 +378,7 @@ def run():
377378

378379
if args.flush_preprocess:
379380
_transport_object = TransportManager(args.transport)
380-
_transport_object.feedback_queue = machine_config.feedback_queue
381+
_transport_object.feedback_queue = security_config.feedback_queue
381382
stashed_files = murfey_db.exec(
382383
select(db.PreprocessStash)
383384
.where(db.PreprocessStash.session_id == args.session_id)
@@ -407,7 +408,7 @@ def run():
407408
zocalo_message = {
408409
"recipes": ["em-spa-preprocess"],
409410
"parameters": {
410-
"feedback_queue": machine_config.feedback_queue,
411+
"feedback_queue": _transport_object.feedback_queue,
411412
"dcid": collected_ids[1].id,
412413
"kv": metadata["voltage"],
413414
"autoproc_program_id": collected_ids[3].id,

0 commit comments

Comments
 (0)