refactor: Pass kwargs into query class #268

- This allows the `COUNT` keyword to be specified during a search API endpoint
ral-facilities · Jan 13, 2022 · 34fa7d9 · 34fa7d9
1 parent e0b54ea
commit 34fa7d9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/datagateway_api/src/search_api/query.py b/datagateway_api/src/search_api/query.py
@@ -4,13 +4,15 @@
 
 
 class SearchAPIQuery:
-    def __init__(self, panosc_entity_name):
+    def __init__(self, panosc_entity_name, **kwargs):
         self.panosc_entity_name = panosc_entity_name
         self.icat_entity_name = mappings.mappings[panosc_entity_name][
             "base_icat_entity"
         ]
 
-        self.icat_query = ICATQuery(SessionHandler.client, self.icat_entity_name)
+        self.icat_query = ICATQuery(
+            SessionHandler.client, self.icat_entity_name, **kwargs,
+        )
 
     def __repr__(self):
         return (