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