astrodatadownloader: add getStarObservations from old project

2024-03-04 16:35:17 +01:00
parent 08c9ad0014
commit f3dafa9ad1
1 changed files with 18 additions and 0 deletions
@@ -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]