add preferredQueryEngine to run the query on Upsolver instead of the default Athena

Author: tanyshak

upsolver/client/query.py

@@ -5,6 +5,10 @@
 
 ExecutionResult = list
 
+preferredQueryEngine = {"displayName": "Upsolver",
+                            "engine": {
+                                "clazz": "GetAvailableQueryEnginesUseCase$QueryEngine"}}
+
 class RestQueryApi():
     def __init__(self, requester: Requester, poller_builder: ResponsePollerBuilder):
         self.requester = requester
@@ -19,7 +23,7 @@ def execute(self, query: str, timeout_sec: float) -> Iterator[ExecutionResult]:
 
         (data, next_path) = poller(
             self.requester,
-            self.requester.post('query', json={'sql': query})
+            self.requester.post('query', json={'sql': query, 'preferredQueryEngine': preferredQueryEngine})
         )
         yield data