mirror of
https://github.com/paulscherrerinstitute/sf_daq_broker.git
synced 2026-04-30 14:32:23 +02:00
Dmitry Ozerov
64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
import argparse
|
|
import requests
|
|
import logging
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
def create_broker_request(channels, output_file, start_pulse_id, stop_pulse_id, rest_api_host, compact):
|
|
|
|
_logger.info("Broker hostname: %s", rest_api_host)
|
|
|
|
parameters = {"general/created": "just now",
|
|
"general/user": "root",
|
|
"general/process": "manual_broker_test.py",
|
|
"general/instrument": "bsread",
|
|
"output_file": output_file,
|
|
"output_file_format": "compact" if compact else None}
|
|
|
|
_logger.info("Sending request with parameters: %s", parameters)
|
|
requests.post("%s/parameters" % rest_api_host, json=parameters)
|
|
|
|
_logger.info("Sending start_pulse_id: %d", start_pulse_id)
|
|
requests.put("%s/start_pulse_id/%d" % (rest_api_host, start_pulse_id))
|
|
|
|
_logger.info("Sending stop_pulse_id: %d", stop_pulse_id)
|
|
requests.put("%s/stop_pulse_id/%d" % (rest_api_host, stop_pulse_id))
|
|
|
|
_logger.info("Request sent. Soon check for the output file %s.", output_file)
|
|
|
|
|
|
def run():
|
|
parser = argparse.ArgumentParser(description='Manual broker test')
|
|
parser.add_argument("channels_file", type=str, help="JSON file with channels to buffer.")
|
|
parser.add_argument("output_file", type=str, help="Where to write the output file.")
|
|
parser.add_argument("start_pulse_id", type=int, help="Start pulse_id to write.")
|
|
parser.add_argument("stop_pulse_id", type=int, help="Stop pulse_id to write.")
|
|
parser.add_argument("--host", help="Hostname of broker rest API\nFormat: hostname:port", default="localhost:1002")
|
|
parser.add_argument("--compact", action='store_true', help="broker rest API\nFormat: hostname:port",
|
|
default="localhost:1002")
|
|
parser.add_argument("--log_level", default="INFO",
|
|
choices=['CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG'], help="Log level to use.")
|
|
|
|
args = parser.parse_args()
|
|
|
|
logging.basicConfig(level=args.log_level, format='[%(levelname)s] %(message)s')
|
|
|
|
with open(args.channels_file) as input_file:
|
|
file_lines = input_file.readlines()
|
|
channels = [channel.strip() for channel in file_lines
|
|
if not channel.strip().startswith("#") and channel.strip()]
|
|
|
|
create_broker_request(
|
|
channels=channels,
|
|
output_file=args.output_file,
|
|
start_pulse_id=args.start_pulse_id,
|
|
stop_pulse_id=args.stop_pulse_id,
|
|
rest_api_host="http://%s" % args.host,
|
|
compact=args.compact
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|