Added nbin and nchan to obs list and download

OZGrav · Mar 14, 2024 · 4d47f90 · 4d47f90
1 parent d2b8dc9
commit 4d47f90
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/psrdb/tables/observation.py b/psrdb/tables/observation.py
@@ -34,6 +34,8 @@ def __init__(self, client, logger=None):
             "beam",
             "band",
             "duration",
+            "foldNchan",
+            "foldNbin",
         ]
 
     def list(
@@ -222,7 +224,7 @@ def download(
 
         # Loop over the pulsar_fold_results and dump them as a file
         with open(output_name, "w") as f:
-            f.write("Obs ID,Pulsar Jname,UTC Start,Project Short Name,Beam #,Observing Band,Duration (s),Calibration Location\n")
+            f.write("Obs ID,Pulsar Jname,UTC Start,Project Short Name,Beam #,Observing Band,Duration (s),Nchan,Nbin,Calibration Location\n")
             for observations_dict in observations_dicts:
                 data_line = [
                     str(decode_id(observations_dict["id"])),
@@ -232,6 +234,8 @@ def download(
                     str(observations_dict["beam"]),
                     str(observations_dict["band"]),
                     str(observations_dict["duration"]),
+                    str(observations_dict["foldNchan"]),
+                    str(observations_dict["foldNbin"]),
                     str(observations_dict["calibration"]["location"]),
                 ]
                 f.write(f"{','.join(data_line)}\n")