mirror of
https://github.com/paulscherrerinstitute/sf_daq_buffer.git
synced 2026-04-23 03:22:43 +02:00
Dmitry Ozerov
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import argparse
|
|
import json
|
|
import os
|
|
import datetime
|
|
from shutil import copyfile
|
|
|
|
PEDESTAL_DIRECTORY="/sf/jungfrau/data/pedestal"
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument("file_pedestal", type=str)
|
|
parser.add_argument("json_run", type=str)
|
|
parser.add_argument("detector", type=str)
|
|
parser.add_argument("json_stream", type=str)
|
|
|
|
args = parser.parse_args()
|
|
|
|
with open(args.json_run, "r") as run_file:
|
|
data = json.load(run_file)
|
|
|
|
request_time=datetime.datetime.strptime(data["request_time"], '%Y-%m-%d %H:%M:%S.%f')
|
|
|
|
if not os.path.isdir(f'{PEDESTAL_DIRECTORY}/{args.detector}'):
|
|
os.mkdir(f'{PEDESTAL_DIRECTORY}/{args.detector}')
|
|
|
|
out_name = f'{PEDESTAL_DIRECTORY}/{args.detector}/{request_time.strftime("%Y%m%d_%H%M%S")}.h5'
|
|
copyfile(args.file_pedestal, out_name)
|
|
|
|
print(f'Copied resulting pedestal file {args.file_pedestal} to {out_name}')
|
|
|
|
if not os.path.exists(args.json_stream):
|
|
print(f'stream file {args.json_stream} does not exists, exiting')
|
|
exit()
|
|
|
|
with open(args.json_stream, "r") as stream_file:
|
|
det = json.load(stream_file)
|
|
|
|
print(f'Changing in stream file {args.json_stream} pedestal from {det["pedestal_file"]} to {out_name}')
|
|
|
|
det["pedestal_file"] = out_name
|
|
|
|
with open(args.json_stream, "w") as write_file:
|
|
json.dump(det, write_file, indent=4)
|
|
|