Eliminate empty log files forever

Author: matteobachetti

hendrics/binary.py

@@ -278,25 +278,25 @@ def main_presto(args=None):
 
     if args.energy_interval is None:
         args.energy_interval = [None, None]
-    with log.log_to_file("HENbinary.log"):
-        for f in args.files:
-            print(f)
-            outfile = f.replace(HEN_FILE_EXTENSION, ".dat")
-            ftype, contents = get_file_type(f)
-            if ftype == "lc":
-                lcinfo = save_lc_to_binary(contents, outfile)
-            elif ftype == "events":
-                if args.deorbit_par is not None:
-                    contents = deorbit_events(contents, args.deorbit_par)
-                lcinfo = save_events_to_binary(
-                    contents,
-                    outfile,
-                    bin_time=bintime,
-                    emin=args.energy_interval[0],
-                    emax=args.energy_interval[1],
-                )
-            else:
-                raise ValueError("File type not recognized")
-
-            info = get_header_info(contents)
-            save_inf(lcinfo, info, f.replace(HEN_FILE_EXTENSION, ".inf"))
+
+    for f in args.files:
+        print(f)
+        outfile = f.replace(HEN_FILE_EXTENSION, ".dat")
+        ftype, contents = get_file_type(f)
+        if ftype == "lc":
+            lcinfo = save_lc_to_binary(contents, outfile)
+        elif ftype == "events":
+            if args.deorbit_par is not None:
+                contents = deorbit_events(contents, args.deorbit_par)
+            lcinfo = save_events_to_binary(
+                contents,
+                outfile,
+                bin_time=bintime,
+                emin=args.energy_interval[0],
+                emax=args.energy_interval[1],
+            )
+        else:
+            raise ValueError("File type not recognized")
+
+        info = get_header_info(contents)
+        save_inf(lcinfo, info, f.replace(HEN_FILE_EXTENSION, ".inf"))

hendrics/calibrate.py

@@ -219,21 +219,20 @@ def main(args=None):
         args.loglevel = "DEBUG"
 
     log.setLevel(args.loglevel)
-    with log.log_to_file("HENcalibrate.log"):
-        funcargs = []
-        for i_f, f in enumerate(files):
-            outname = f
-            if args.overwrite is False:
-                label = "_calib"
-                if args.rough:
-                    label = "_rough_calib"
-                outname = f.replace(get_file_extension(f), label + HEN_FILE_EXTENSION)
-            funcargs.append([f, outname, args.rmf, args.rough])
-
-        if os.name == "nt" or args.nproc == 1:
-            [_calib_wrap(fa) for fa in funcargs]
-        else:
-            pool = Pool(processes=args.nproc)
-            for i in pool.imap_unordered(_calib_wrap, funcargs):
-                pass
-            pool.close()
+    funcargs = []
+    for i_f, f in enumerate(files):
+        outname = f
+        if args.overwrite is False:
+            label = "_calib"
+            if args.rough:
+                label = "_rough_calib"
+            outname = f.replace(get_file_extension(f), label + HEN_FILE_EXTENSION)
+        funcargs.append([f, outname, args.rmf, args.rough])
+
+    if os.name == "nt" or args.nproc == 1:
+        [_calib_wrap(fa) for fa in funcargs]
+    else:
+        pool = Pool(processes=args.nproc)
+        for i in pool.imap_unordered(_calib_wrap, funcargs):
+            pass
+        pool.close()

hendrics/colors.py

@@ -45,45 +45,44 @@ def main(args=None):
 
     log.setLevel(args.loglevel)
 
-    with log.log_to_file("HENcolors.log"):
-        energies = [
-            [args.energies[0], args.energies[1]],
-            [args.energies[2], args.energies[3]],
-        ]
-        if args.outfile is not None and len(files) > 1:
-            raise ValueError("Specify --output only when processing " "a single file")
-        for f in files:
-            events = load_events(f)
-            if not args.use_pi and events.energy is None:
-                raise ValueError(
-                    f"Energy information not found in file {f}. "
-                    "Use --use-pi if you want to use PI channels "
-                    "instead."
-                )
-            h_starts, h_stops, colors, color_errs = events.get_color_evolution(
-                energy_ranges=energies, segment_size=args.bintime, use_pi=args.use_pi
+    energies = [
+        [args.energies[0], args.energies[1]],
+        [args.energies[2], args.energies[3]],
+    ]
+    if args.outfile is not None and len(files) > 1:
+        raise ValueError("Specify --output only when processing " "a single file")
+    for f in files:
+        events = load_events(f)
+        if not args.use_pi and events.energy is None:
+            raise ValueError(
+                f"Energy information not found in file {f}. "
+                "Use --use-pi if you want to use PI channels "
+                "instead."
             )
+        h_starts, h_stops, colors, color_errs = events.get_color_evolution(
+            energy_ranges=energies, segment_size=args.bintime, use_pi=args.use_pi
+        )
 
-            time = (h_starts + h_stops) / 2
+        time = (h_starts + h_stops) / 2
 
-            scolor = Lightcurve(
-                time=time,
-                counts=colors,
-                err=color_errs,
-                input_counts=False,
-                err_dist="gauss",
-                gti=events.gti,
-                dt=args.bintime,
-                skip_checks=True,
-            )
+        scolor = Lightcurve(
+            time=time,
+            counts=colors,
+            err=color_errs,
+            input_counts=False,
+            err_dist="gauss",
+            gti=events.gti,
+            dt=args.bintime,
+            skip_checks=True,
+        )
 
-            if args.outfile is None:
-                label = "_E_"
-                if args.use_pi:
-                    label = "_PI_"
-                label += "{3:g}-{2:g}_over_{1:g}-{0:g}".format(*args.energies)
-                args.outfile = hen_root(f) + label + HEN_FILE_EXTENSION
-            scolor.e_intervals = np.asarray([float(k) for k in args.energies])
-            scolor.use_pi = args.use_pi
-            save_lcurve(scolor, args.outfile, lctype="Color")
-            print(args.outfile)
+        if args.outfile is None:
+            label = "_E_"
+            if args.use_pi:
+                label = "_PI_"
+            label += "{3:g}-{2:g}_over_{1:g}-{0:g}".format(*args.energies)
+            args.outfile = hen_root(f) + label + HEN_FILE_EXTENSION
+        scolor.e_intervals = np.asarray([float(k) for k in args.energies])
+        scolor.use_pi = args.use_pi
+        save_lcurve(scolor, args.outfile, lctype="Color")
+        print(args.outfile)

hendrics/create_gti.py

@@ -188,34 +188,31 @@ def main(args=None):
         args.loglevel = "DEBUG"
 
     log.setLevel(args.loglevel)
-    with log.log_to_file("HENcreategti.log"):
-        filter_expr = args.filter
-        if filter_expr is None and args.apply_gti is None:
-            sys.exit(
-                "Please specify filter expression (-f option) or input " "GTI file (-a option)"
-            )
-
-        for fname in files:
-            if args.apply_gti is not None:
-                data = load_data(args.apply_gti)
-                gti = data["gti"]
-            else:
-                gti = create_gti(
-                    fname,
-                    filter_expr,
-                    safe_interval=args.safe_interval,
-                    minimum_length=args.minimum_length,
-                )
-            if args.create_only:
-                continue
-            if args.overwrite:
-                outname = fname
-            else:
-                # Use default
-                outname = None
-            apply_gti(
+    filter_expr = args.filter
+    if filter_expr is None and args.apply_gti is None:
+        sys.exit("Please specify filter expression (-f option) or input " "GTI file (-a option)")
+
+    for fname in files:
+        if args.apply_gti is not None:
+            data = load_data(args.apply_gti)
+            gti = data["gti"]
+        else:
+            gti = create_gti(
                 fname,
-                gti,
-                outname=outname,
+                filter_expr,
+                safe_interval=args.safe_interval,
                 minimum_length=args.minimum_length,
             )
+        if args.create_only:
+            continue
+        if args.overwrite:
+            outname = fname
+        else:
+            # Use default
+            outname = None
+        apply_gti(
+            fname,
+            gti,
+            outname=outname,
+            minimum_length=args.minimum_length,
+        )

hendrics/efsearch.py

@@ -1875,14 +1875,12 @@ def _common_main(args, func):
 
 def main_efsearch(args=None):
     """Main function called by the `HENefsearch` command line script."""
-    with log.log_to_file("HENefsearch.log"):
-        return _common_main(args, epoch_folding_search)
+    return _common_main(args, epoch_folding_search)
 
 
 def main_zsearch(args=None):
     """Main function called by the `HENzsearch` command line script."""
-    with log.log_to_file("HENzsearch.log"):
-        return _common_main(args, z_n_search)
+    return _common_main(args, z_n_search)
 
 
 def z2_vs_pf(event_list, deadtime=0.0, ntrials=100, outfile=None, N=2):

hendrics/exposure.py

@@ -353,30 +353,29 @@ def main(args=None):
         args.loglevel = "DEBUG"
 
     log.setLevel(args.loglevel)
-    with log.log_to_file("HENexposure.log"):
-        lc_file = args.lcfile
-        uf_file = args.uffile
+    lc_file = args.lcfile
+    uf_file = args.uffile
 
-        outroot = _assign_value_if_none(args.outroot, hen_root(lc_file))
+    outroot = _assign_value_if_none(args.outroot, hen_root(lc_file))
 
-        outname = outroot + "_lccorr" + HEN_FILE_EXTENSION
+    outname = outroot + "_lccorr" + HEN_FILE_EXTENSION
 
-        outfile = correct_lightcurve(lc_file, uf_file, outname)
+    outfile = correct_lightcurve(lc_file, uf_file, outname)
 
-        # outdata = load_data(outfile)
-        _, outdata = get_file_type(outfile, raw_data=False)
-        time = outdata.time
-        lc = outdata.counts
-        expo = outdata.expo
-        gti = outdata.gti
+    # outdata = load_data(outfile)
+    _, outdata = get_file_type(outfile, raw_data=False)
+    time = outdata.time
+    lc = outdata.counts
+    expo = outdata.expo
+    gti = outdata.gti
 
-        try:
-            _plot_corrected_light_curve(time, lc * expo, expo, gti, outroot)
-            _plot_dead_time_from_uf(uf_file, outroot)
-        except Exception as e:
-            warnings.warn(str(e))
+    try:
+        _plot_corrected_light_curve(time, lc * expo, expo, gti, outroot)
+        _plot_dead_time_from_uf(uf_file, outroot)
+    except Exception as e:
+        warnings.warn(str(e))
 
-        if args.plot:
-            import matplotlib.pyplot as plt
+    if args.plot:
+        import matplotlib.pyplot as plt
 
-            plt.show()
+        plt.show()

hendrics/exvar.py

@@ -66,31 +66,28 @@ def main(args=None):
         args.loglevel = "DEBUG"
 
     log.setLevel(args.loglevel)
-    with log.log_to_file("HENexcvar.log"):
-        filelist = []
-        for fname in args.files:
-            lcurve = load_lcurve(fname)
-            if args.norm == "fvar":
-                start, stop, res = lcurve.analyze_segments(
-                    fvar, args.chunk_length, args.fraction_step
-                )
-            elif args.norm == "excvar":
-                start, stop, res = lcurve.analyze_segments(
-                    excvar_none, args.chunk_length, args.fraction_step
-                )
-            elif args.norm == "norm_excvar":
-                start, stop, res = lcurve.analyze_segments(
-                    excvar_norm, args.chunk_length, args.fraction_step
-                )
-            else:
-                raise ValueError("Normalization must be fvar, norm_excvar " "or excvar")
-            var, var_err = res
-            out = hen_root(fname) + "_" + args.norm + ".qdp"
-            save_as_qdp(
-                [(start + stop) / 2, var],
-                [(stop - start) / 2, var_err],
-                filename=out,
+    filelist = []
+    for fname in args.files:
+        lcurve = load_lcurve(fname)
+        if args.norm == "fvar":
+            start, stop, res = lcurve.analyze_segments(fvar, args.chunk_length, args.fraction_step)
+        elif args.norm == "excvar":
+            start, stop, res = lcurve.analyze_segments(
+                excvar_none, args.chunk_length, args.fraction_step
             )
-            filelist.append(out)
+        elif args.norm == "norm_excvar":
+            start, stop, res = lcurve.analyze_segments(
+                excvar_norm, args.chunk_length, args.fraction_step
+            )
+        else:
+            raise ValueError("Normalization must be fvar, norm_excvar " "or excvar")
+        var, var_err = res
+        out = hen_root(fname) + "_" + args.norm + ".qdp"
+        save_as_qdp(
+            [(start + stop) / 2, var],
+            [(stop - start) / 2, var_err],
+            filename=out,
+        )
+        filelist.append(out)
 
     return filelist