Error when using invalid keywords in criteria queries

Author: snbianco

astroquery/mast/discovery_portal.py

@@ -6,6 +6,7 @@
 This module contains various methods for querying the MAST Discovery Portal API.
 """
 
+import difflib
 import warnings
 import uuid
 import json
@@ -20,7 +21,7 @@
 from ..query import BaseQuery
 from ..utils import async_to_sync
 from ..utils.class_or_instance import class_or_instance
-from ..exceptions import InputWarning, NoResultsWarning, RemoteServiceError
+from ..exceptions import InputWarning, InvalidQueryError, NoResultsWarning, RemoteServiceError
 
 from . import conf, utils
 
@@ -406,8 +407,10 @@ def build_filter_set(self, column_config_name, service_name=None, **filters):
             # Get the column type and separator
             col_info = caom_col_config.get(colname)
             if not col_info:
-                warnings.warn("Filter {} does not exist. This filter will be skipped.".format(colname), InputWarning)
-                continue
+                closest_match = difflib.get_close_matches(colname, caom_col_config.keys(), n=1)[0]
+                error_msg = f"Filter '{colname}' does not exist. Did you mean '{closest_match}'?" if closest_match \
+                    else f"Filter '{colname}' does not exist."
+                raise InvalidQueryError(error_msg)
 
             colType = "discrete"
             if (col_info.get("vot.datatype", col_info.get("type")) in ("double", "float", "numeric")) \

astroquery/mast/tests/test_mast_remote.py

@@ -274,6 +274,17 @@ def test_observations_query_criteria(self):
                                              intentType="calibration")
         assert (result["intentType"] == "calibration").all()
 
+    def test_observations_query_criteria_invalid_keyword(self):
+        # attempt to make a criteria query with invalid keyword
+        with pytest.raises(InvalidQueryError) as err_no_alt:
+            Observations.query_criteria_count(not_a_keyword='TESS')
+        assert 'Filter not_a_keyword does not exist.' in str(err_no_alt.value)
+
+        # keyword is close enough for difflib to offer alternative
+        with pytest.raises(InvalidQueryError) as err_with_alt:
+            Observations.query_criteria_count(oops_collection='TESS')
+        assert 'obs_collection' in str(err_with_alt.value)
+
     # count functions
     def test_observations_query_region_count(self):
         maxRes = Observations.query_criteria_count()
@@ -881,6 +892,17 @@ def check_result(result, exp_vals):
         assert isinstance(result, Table)
         assert result['distance'][0] <= result['distance'][1]
 
+    def test_catalogs_query_criteria_invalid_keyword(self):
+        # attempt to make a criteria query with invalid keyword
+        with pytest.raises(InvalidQueryError) as err_no_alt:
+            Catalogs.query_criteria(catalog='tic', not_a_keyword='TESS')
+        assert 'Filter not_a_keyword does not exist.' in str(err_no_alt.value)
+
+        # keyword is close enough for difflib to offer alternative
+        with pytest.raises(InvalidQueryError) as err_with_alt:
+            Catalogs.query_criteria(catalog='ctl', objectType="STAR")
+        assert 'objType' in str(err_with_alt.value)
+
     def test_catalogs_query_hsc_matchid_async(self):
         catalogData = Catalogs.query_object("M10",
                                             radius=.001,