From 331abafb485276e077a9372e145268d0ec8a3e70 Mon Sep 17 00:00:00 2001 From: Andrej Babic Date: Thu, 20 Aug 2020 09:25:30 +0200 Subject: [PATCH] Move bsread writer to submodule --- sf_daq_broker/writer/__init__.py | 0 sf_daq_broker/{writer.py => writer/bsread_writer.py} | 4 ++-- .../{writer_format.py => writer/bsread_writer_format.py} | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) create mode 100644 sf_daq_broker/writer/__init__.py rename sf_daq_broker/{writer.py => writer/bsread_writer.py} (98%) rename sf_daq_broker/{writer_format.py => writer/bsread_writer_format.py} (99%) diff --git a/sf_daq_broker/writer/__init__.py b/sf_daq_broker/writer/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/sf_daq_broker/writer.py b/sf_daq_broker/writer/bsread_writer.py similarity index 98% rename from sf_daq_broker/writer.py rename to sf_daq_broker/writer/bsread_writer.py index 113dfbc..f7ab76f 100644 --- a/sf_daq_broker/writer.py +++ b/sf_daq_broker/writer/bsread_writer.py @@ -9,7 +9,7 @@ import requests from bsread import source, PULL from sf_daq_broker import config, utils -from sf_daq_broker.writer_format import CompactDataBufferH5Writer +from sf_daq_broker.writer.bsread_writer_format import CompactBsreadH5Writer _logger = logging.getLogger(__name__) @@ -61,7 +61,7 @@ def write_data_to_file(parameters, json_data): output_file = parameters["output_file"] - writer = CompactDataBufferH5Writer(output_file, parameters) + writer = CompactBsreadH5Writer(output_file, parameters) writer.write_data(json_data) writer.close() diff --git a/sf_daq_broker/writer_format.py b/sf_daq_broker/writer/bsread_writer_format.py similarity index 99% rename from sf_daq_broker/writer_format.py rename to sf_daq_broker/writer/bsread_writer_format.py index bd3fba7..3805700 100644 --- a/sf_daq_broker/writer_format.py +++ b/sf_daq_broker/writer/bsread_writer_format.py @@ -11,7 +11,7 @@ from sf_daq_broker import config _logger = logging.getLogger(__name__) -class DataBufferH5Writer(object): +class BsreadH5Writer(object): def __init__(self, output_file, parameters): self.output_file = output_file self.parameters = parameters @@ -151,7 +151,7 @@ class DataBufferH5Writer(object): self.file.close() -class CompactDataBufferH5Writer(DataBufferH5Writer): +class CompactBsreadH5Writer(BsreadH5Writer): def _build_datasets_data(self, json_data):