diff --git a/slic/core/acquisition/broker/__init__.py b/slic/core/acquisition/broker/__init__.py index d3ded4366..86a6c227d 100644 --- a/slic/core/acquisition/broker/__init__.py +++ b/slic/core/acquisition/broker/__init__.py @@ -1,4 +1,5 @@ -from .brokerclient import BrokerClient, align_pid_left, align_pid_right, retrieve +from . import restapi +from .brokerclient import BrokerClient, align_pid_left, align_pid_right diff --git a/slic/core/acquisition/broker/brokerclient.py b/slic/core/acquisition/broker/brokerclient.py index fda9c06a4..c3cb34cf4 100644 --- a/slic/core/acquisition/broker/brokerclient.py +++ b/slic/core/acquisition/broker/brokerclient.py @@ -2,10 +2,10 @@ from time import sleep from slic.utils import xrange, tqdm_mod#, tqdm_sleep +from . import restapi from .brokerconfig import BrokerConfig, flatten_detectors from .pedestal import take_pedestal from .pids import align_pid_left, align_pid_right, aligned_pid_and_n -from .restapi import advance_run_number, retrieve, power_on_detector, get_config_pvs, set_config_pvs from .tools import get_current_pulseid @@ -33,10 +33,10 @@ class BrokerClient: def get_config_pvs(self, *args, **kwargs): - return get_config_pvs(self.address, *args, **kwargs) + return restapi.get_config_pvs(self.address, *args, **kwargs) def set_config_pvs(self, pvs, *args, **kwargs): - return set_config_pvs(self.address, pvs, *args, **kwargs) + return restapi.set_config_pvs(self.address, pvs, *args, **kwargs) def start(self): @@ -110,7 +110,7 @@ class BrokerClient: self.running = False params = self.get_config(self.run_number, start_pulseid, stop_pulseid) - res = retrieve(self.address, params, timeout=self.timeout) + res = restapi.retrieve(self.address, params, timeout=self.timeout) run_number = res["run_number"] assert run_number == self.run_number, f"received {run_number} and expected {self.run_number} run numbers not identical" #TODO: raise proper exception @@ -124,7 +124,7 @@ class BrokerClient: def next_run(self, *args, **kwargs): - self.run_number = run_number = advance_run_number(self.address, self.config.pgroup, *args, **kwargs) + self.run_number = run_number = restapi.advance_run_number(self.address, self.config.pgroup, *args, **kwargs) return run_number @@ -150,7 +150,7 @@ class BrokerClient: detectors = flatten_detectors(detectors) for d in detectors: - msg = power_on_detector(self.address, d, **kwargs) + msg = restapi.power_on_detector(self.address, d, **kwargs) print(f"{d}: {msg}") diff --git a/slic/core/acquisition/sfacquisition.py b/slic/core/acquisition/sfacquisition.py index 9a0e846b7..9480158e5 100644 --- a/slic/core/acquisition/sfacquisition.py +++ b/slic/core/acquisition/sfacquisition.py @@ -9,7 +9,7 @@ from .baseacquisition import BaseAcquisition from .sfpaths import SwissFELPaths from .bschannels import BSChannels -from .broker import BrokerClient, align_pid_left, align_pid_right, retrieve +from .broker import BrokerClient, align_pid_left, align_pid_right, restapi from .detcfg import DetectorConfig @@ -120,7 +120,7 @@ class SFAcquisition(BaseAcquisition): if not is_continuous(pulseids): params["selected_pulse_ids"] = pulseids - res = retrieve(client.address, params, timeout=client.timeout) + res = restapi.retrieve(client.address, params, timeout=client.timeout) res_run_number = res["run_number"] assert res_run_number == run_number, f"received {res_run_number} and expected {run_number} run numbers not identical"