Merge pull request #315 from pallavibharadwaj/error_handling

feat(data_connector): Parameter check when calling query()

Author: peiwangdb

dataprep/connector/connector.py

@@ -14,7 +14,7 @@
 
 from ..errors import UnreachableError
 from .config_manager import config_directory, ensure_config
-from .errors import RequestError, UniversalParameterOverridden
+from .errors import RequestError, UniversalParameterOverridden, InvalidParameterError
 from .implicit_database import ImplicitDatabase, ImplicitTable
 from .int_ref import IntRef
 from .throttler import OrderedThrottler, ThrottleSession
@@ -117,6 +117,11 @@ async def query(  # pylint: disable=too-many-locals
         **where
             The additional parameters required for the query.
         """
+        allowed_params = self._impdb.tables[table].config["request"]["params"]
+        for key in where:
+            if key not in allowed_params:
+                raise InvalidParameterError(key)
+
         return await self._query_imp(table, where, _auth=_auth, _count=_count)
 
     @property

dataprep/connector/errors.py

@@ -49,3 +49,18 @@ def __init__(self, param: str, uparam: str) -> None:
 
     def __str__(self) -> str:
         return f"the parameter {self.param} is overridden by {self.uparam}"
+
+
+class InvalidParameterError(Exception):
+    """
+    The parameter used in the query is invalid
+    """
+
+    param: str
+
+    def __init__(self, param: str) -> None:
+        super().__init__()
+        self.param = param
+
+    def __str__(self) -> str:
+        return f"the parameter {self.param} is invalid, refer info method"