blackened

km3io/utils/extract_dom_rates.py

@@ -15,21 +15,25 @@ import os
 import numpy as np
 import h5py
 
+
 def get_doms_rates(slices, frame):
-    raw_rates = [km3io.online.get_rate(getattr(slices.rates[frame],f"ch{i}")) for i in range(31)]
+    raw_rates = [
+        km3io.online.get_rate(getattr(slices.rates[frame], f"ch{i}")) for i in range(31)
+    ]
     return np.sum(raw_rates, axis=0)
 
+
 def main():
     arguments = docopt(__doc__)
 
     data = {}
     for key in arguments:
-        data[key.replace('-','')] = arguments[key]
+        data[key.replace("-", "")] = arguments[key]
 
-    reader = km3io.OnlineReader(data['input_file'])
+    reader = km3io.OnlineReader(data["input_file"])
     summary_slices = reader.summaryslices
 
-    detector = km3pipe.hardware.Detector(data['detector_file'])
+    detector = km3pipe.hardware.Detector(data["detector_file"])
 
     n_slices = len(summary_slices.slices)
 
@@ -37,18 +41,25 @@ def main():
     for dom in detector.dom_ids:
         doms_rates[dom] = np.zeros(n_slices)
 
-    for i in range (n_slices):
+    for i in range(n_slices):
         rates = get_doms_rates(summary_slices, i)
         dom_ids = summary_slices.slices[i].dom_id
-        for j in range(len(dom_ids)-1):
+        for j in range(len(dom_ids) - 1):
             doms_rates[dom_ids[j]][i] = rates[j]
 
-    f = h5py.File(data['output_file'], "a")
-    f.create_dataset("frame_indices",data=np.array(summary_slices.headers.frame_index))
-    f.create_dataset("frame_times"  ,data=np.array(summary_slices.headers.UTC_seconds * 1e9 + summary_slices.headers.UTC_16nanosecondcycles * 16))
+    f = h5py.File(data["output_file"], "a")
+    f.create_dataset("frame_indices", data=np.array(summary_slices.headers.frame_index))
+    f.create_dataset(
+        "frame_times",
+        data=np.array(
+            summary_slices.headers.UTC_seconds * 1e9
+            + summary_slices.headers.UTC_16nanosecondcycles * 16
+        ),
+    )
 
     for key, value in doms_rates.items():
         f.create_dataset(str(key), data=value)
 
+
 if __name__ == "__main__":
     main()

setup.py

@@ -36,7 +36,12 @@ setup(
     install_requires=read_requirements("install"),
     extras_require={kind: read_requirements(kind) for kind in ["dev"]},
     python_requires=">=3.6",
-    entry_points={"console_scripts": ["KPrintTree=km3io.utils.kprinttree:main","ExtractDomRates=km3io.utils.ExtractDomRates:main"]},
+    entry_points={
+        "console_scripts": [
+            "KPrintTree=km3io.utils.kprinttree:main",
+            "ExtractDomRates=km3io.utils.extract_dom_rates:main",
+        ]
+    },
     classifiers=[
         "Intended Audience :: Developers",
         "Intended Audience :: Science/Research",