From 393b1a8663c068da6f8d8b91814ae105d65592c5 Mon Sep 17 00:00:00 2001
From: Thibault Barnouin <thibault.barnouin@astro.unistra.fr>
Date: Fri, 28 Jun 2024 11:51:34 +0200
Subject: [PATCH] treat no results from query as exception

---
 package/lib/query.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/package/lib/query.py b/package/lib/query.py
index 5c244c5..2372bd4 100755
--- a/package/lib/query.py
+++ b/package/lib/query.py
@@ -10,6 +10,9 @@ from astropy.table import unique, Column
 from astropy.time import Time, TimeDelta
 import astropy.units as u
 import numpy as np
+from astroquery.exceptions import NoResultsWarning
+from warnings import filterwarnings
+filterwarnings("error", category=NoResultsWarning)
 
 
 def divide_proposal(products):
@@ -191,5 +194,6 @@ if __name__ == "__main__":
     parser.add_argument('-o', '--output_dir', metavar='directory_path', required=False,
                         help='output directory path for the data products', type=str, default="./data")
     args = parser.parse_args()
+    print(args.target)
     prodpaths = retrieve_products(target=args.target, proposal_id=args.proposal_id)
     print(prodpaths)