From bccec481a1a9236bf04db490a052a6c41c07368a Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Fri, 19 Apr 2024 18:11:01 +0200 Subject: [PATCH] added get_running_detectors, get_allowed_detectors, close_pgroup endpoints; use {} instead of dict() --- slic/core/acquisition/broker/restapi.py | 28 +++++++++++++++++++------ 1 file changed, 22 insertions(+), 6 deletions(-) diff --git a/slic/core/acquisition/broker/restapi.py b/slic/core/acquisition/broker/restapi.py index bca60efb..791ebd74 100644 --- a/slic/core/acquisition/broker/restapi.py +++ b/slic/core/acquisition/broker/restapi.py @@ -20,12 +20,12 @@ def get_run_number(address, pgroup, *args, **kwargs): def retrieve(address, *args, **kwargs): response = post_request(address, "retrieve_from_buffers", *args, **kwargs) - res = dict( - run_number = int(response["run_number"]), - acq_number = int(response["acquisition_number"]), - total_acq_number = int(response["unique_acquisition_number"]), - filenames = response["files"] - ) + res = { + "run_number": int(response["run_number"]), + "acq_number": int(response["acquisition_number"]), + "total_acq_number": int(response["unique_acquisition_number"]), + "filenames": response["files"] + } return res @@ -45,6 +45,22 @@ def power_on_detector(address, detector, *args, **kwargs): response = post_request(address, "power_on_detector", params, *args, **kwargs) return response.get("message") +def get_running_detectors(address, *args, **kwargs): + params = {} + response = post_request(address, "get_running_detectors", params, *args, **kwargs) + return response.get("detectors") + +def get_allowed_detectors(address, *args, **kwargs): + params = {} + response = post_request(address, "get_allowed_detectors", params, *args, **kwargs) + return response.get("detectors") + +def close_pgroup(address, pgroup, *args, **kwargs): + params = {"pgroup": pgroup} + response = post_request(address, "close_pgroup_writing", params, *args, **kwargs) + return response.get("message") + + def post_request(address, endpoint, params, timeout=10): requrl = make_requrl(address, endpoint)