From f3dafa9ad1e286077837bfde8ad85ed2a7a3a637 Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Mon, 4 Mar 2024 16:35:17 +0100 Subject: [PATCH] astrodatadownloader: add getStarObservations from old project --- astrodatadownloader/astrodatadownloader.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/astrodatadownloader/astrodatadownloader.py b/astrodatadownloader/astrodatadownloader.py index 8b13789..a99808b 100644 --- a/astrodatadownloader/astrodatadownloader.py +++ b/astrodatadownloader/astrodatadownloader.py @@ -1 +1,19 @@ +from astroquery.mast import Observations +def getStarObservations(starName: str, sequences: list[int]) -> str: + obs = Observations.query_object(objectname=starName, radius="0 deg") + + if(len(sequences) > 1): + obsWantedFilter = ((obs['dataproduct_type'] == 'timeseries') & + (obs['obs_collection'] == 'TESS') & + (obs['sequence_number'] >= max(sequences)) & + (obs['sequence_number'] <= min(sequences))) + elif(len(sequences) == 1): + obsWantedFilter = ((obs['dataproduct_type'] == 'timeseries') & + (obs['obs_collection'] == 'TESS') & + (obs['sequence_number'] == sequences[0])) + else: + obsWantedFilter = ((obs['dataproduct_type'] == 'timeseries') & + (obs['obs_collection'] == 'TESS')) + + return obs[obsWantedFilter] \ No newline at end of file