Merge pull request #29 from stackql/feature/refactor

added additional constructor args

Author: jeffreyaven

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## v3.2.5 (2023-12-07)
+
+### Updates
+
+ * included `app_root` and `execution_concurrency_limit` options in `StackQL` constructor
+
 ## v3.2.4 (2023-10-24)
 
 ### Updates

README.rst

@@ -193,4 +193,4 @@ To publish the package to PyPI, run the following command:
 
 ::
 
-    twine upload dist/pystackql-3.2.4.tar.gz
+    twine upload dist/pystackql-3.2.5.tar.gz

docs/source/conf.py

@@ -26,7 +26,7 @@
 # The short X.Y version
 version = ''
 # The full version, including alpha/beta/rc tags
-release = '3.2.4'
+release = '3.2.5'
 
 
 # -- General configuration ---------------------------------------------------

pystackql/stackql.py

@@ -39,6 +39,12 @@ class StackQL:
 	:param download_dir: The download directory for the StackQL executable 
 		(defaults to `site.getuserbase()`, not supported in `server_mode`)
 	:type download_dir: str, optional
+	:param app_root: Application config and cache root path
+		(defaults to `{cwd}/.stackql`)
+	:type app_root: str, optional	
+	:param execution_concurrency_limit: Concurrency limit for query execution
+		(defaults to `1`, set to `-1` for unlimited)
+	:type execution_concurrency_limit: int, optional	
 	:param api_timeout: API timeout 
 		(defaults to `45`, not supported in `server_mode`)
 	:type api_timeout: int, optional
@@ -204,6 +210,8 @@ def __init__(self,
 				 server_address='127.0.0.1', 
 				 server_port=5466, 
 				 download_dir=None, 
+				 app_root=None,
+				 execution_concurrency_limit=1,
 				 output='dict',
 				 custom_auth=None,
 				 sep=',', 
@@ -290,6 +298,17 @@ def __init__(self,
 					_setup(self.download_dir, this_os)
 					self.version, self.sha = _get_version(self.bin_path)
 
+			# if app_root is set, use it
+			if app_root is not None:
+				self.app_root = app_root
+				self.params.append("--approot")
+				self.params.append(app_root)
+
+			# set execution_concurrency_limit
+			self.execution_concurrency_limit = execution_concurrency_limit
+			self.params.append("--execution.concurrency.limit")
+			self.params.append(str(execution_concurrency_limit))
+
 			# if custom_auth is set, use it
 			if custom_auth is not None:
 				authobj, authstr = _format_auth(custom_auth)

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name='pystackql',
-    version='3.2.4',
+    version='3.2.5',
     description='A Python interface for StackQL',
     long_description=readme,
     author='Jeffrey Aven',