astrodatadownloader: write source into manifest too

2024-03-08 11:37:44 +01:00
parent 2d4133617f
commit 37b1887326
1 changed files with 9 additions and 0 deletions
@@ -23,6 +23,7 @@ def parse_manifest(manifest):
    """
    results = deepcopy(manifest)
    filenames = []
+    sources = []
    sector_range = []
    exts = []
    for i,f in enumerate(manifest['Local Path']):
@@ -37,11 +38,19 @@ def parse_manifest(manifest):
        except:
            sector_range.append("-")

+        if("TESS" in f):
+            sources.append("TESS")
+        elif("Kepler" in f):
+            sources.append("Kepler")
+        elif("K2" in f):
+            sources.append("K2")
+
        path_parts = np.array(f.split(sep = '/'))
        filenames.append(path_parts[-1])
        exts.append(path_parts[-1][-8:])

    results.add_column(table.Column(name = "filename", data = filenames))
+    results.add_column(table.Column(name = "source", data = sources))
    results.add_column(table.Column(name = "sector", data = sector_range))
    results.add_column(table.Column(name = "fileType", data = exts))
    results.add_column(table.Column(name = "index", data = np.arange(0,len(manifest))))