SGCMarkus/flaredetector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

astrodatadownloader: write source into manifest too

Browse Source
This commit is contained in:
SGCMarkus
2024-03-08 11:37:44 +01:00
parent 2d4133617f
commit 37b1887326
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/astrodatadownloader/astrodatadownloader.py
+9
View File
@@ -23,6 +23,7 @@ def parse_manifest(manifest):
""" """
results = deepcopy(manifest) results = deepcopy(manifest)
filenames = [] filenames = []
sources = []
sector_range = [] sector_range = []
exts = [] exts = []
for i,f in enumerate(manifest['Local Path']): for i,f in enumerate(manifest['Local Path']):
@@ -37,11 +38,19 @@ def parse_manifest(manifest):
except: except:
sector_range.append("-") 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 = '/')) path_parts = np.array(f.split(sep = '/'))
filenames.append(path_parts[-1]) filenames.append(path_parts[-1])
exts.append(path_parts[-1][-8:]) exts.append(path_parts[-1][-8:])
results.add_column(table.Column(name = "filename", data = filenames)) 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 = "sector", data = sector_range))
results.add_column(table.Column(name = "fileType", data = exts)) results.add_column(table.Column(name = "fileType", data = exts))
results.add_column(table.Column(name = "index", data = np.arange(0,len(manifest)))) results.add_column(table.Column(name = "index", data = np.arange(0,len(manifest))))