add async to query method

Author: danfrankj

omniduct/databases/base.py

@@ -10,6 +10,7 @@
 import sqlparse
 from decorator import decorator
 from jinja2 import StrictUndefined, Template
+from concurrent.futures import ThreadPoolExecutor
 
 from . import cursor_formatters
 from omniduct.caches.base import cached_method
@@ -18,6 +19,7 @@
 from omniduct.utils.docs import quirk_docs
 from omniduct.utils.magics import MagicsProvider, process_line_arguments, process_line_cell_arguments
 
+
 logging.getLogger('requests').setLevel(logging.WARNING)
 
 
@@ -236,15 +238,21 @@ def query(self, statement, format=None, format_opts={}, **kwargs):
         Returns:
             The results of the query formatted as nominated.
         """
-        cursor = self.execute(statement, async=False, template=False, **kwargs)
+        async = kwargs.get('async', False)
+        cursor = self.execute(statement, template=False, **kwargs)
+
+        def finish(cursor):
+            if self._cursor_empty(cursor):
+                return None
+            # Some DBAPI2 cursor implementations error if attempting to extract
+            # data from an empty cursor, and if so, we simply return None.
+            formatter = self._get_formatter(format, cursor, **format_opts)
+            return formatter.dump()
 
-        # Some DBAPI2 cursor implementations error if attempting to extract
-        # data from an empty cursor, and if so, we simply return None.
-        if self._cursor_empty(cursor):
-            return None
+        if not async:
+            return finish(cursor)
 
-        formatter = self._get_formatter(format, cursor, **format_opts)
-        return formatter.dump()
+        return ThreadPoolExecutor(max_workers=1).submit(finish, cursor)  # returns a future
 
     def stream(self, statement, format=None, format_opts={}, batch=None, **kwargs):
         """