From 37b1887326db306cf5ce870458c467f9dd31e17b Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Fri, 8 Mar 2024 11:37:44 +0100 Subject: [PATCH] astrodatadownloader: write source into manifest too --- main/astrodatadownloader/astrodatadownloader.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/main/astrodatadownloader/astrodatadownloader.py b/main/astrodatadownloader/astrodatadownloader.py index a5425c8..b15e357 100644 --- a/main/astrodatadownloader/astrodatadownloader.py +++ b/main/astrodatadownloader/astrodatadownloader.py @@ -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))))