Fix up query default args value

Author: auxten

chdb/__init__.py

@@ -1,6 +1,13 @@
 import sys
 import os
 
+
+# If any UDF is defined, the path of the UDF will be set to this variable
+# and the path will be deleted when the process exits
+# UDF config path will be f"{g_udf_path}/udf_config.xml"
+# UDF script path will be f"{g_udf_path}/{func_name}.py"
+g_udf_path = ""
+
 chdb_version = (0, 6, 0)
 if sys.version_info[:2] >= (3, 7):
     # get the path of the current file
@@ -32,27 +39,30 @@ def to_arrowTable(res):
         import pyarrow as pa
         import pandas
     except ImportError as e:
-        print(f'ImportError: {e}')
+        print(f"ImportError: {e}")
         print('Please install pyarrow and pandas via "pip install pyarrow pandas"')
-        raise ImportError('Failed to import pyarrow or pandas') from None
+        raise ImportError("Failed to import pyarrow or pandas") from None
     if len(res) == 0:
         return pa.Table.from_batches([], schema=pa.schema([]))
     return pa.RecordBatchFileReader(res.bytes()).read_all()
 
 
 # return pandas dataframe
 def to_df(r):
-    """"convert arrow table to Dataframe"""
+    """convert arrow table to Dataframe"""
     t = to_arrowTable(r)
     return t.to_pandas(use_threads=True)
 
 
 # wrap _chdb functions
-def query(sql, output_format="CSV", path=None, udf_path=None):
+def query(sql, output_format="CSV", path="", udf_path=""):
+    global g_udf_path
+    if udf_path != "":
+        g_udf_path = udf_path
     lower_output_format = output_format.lower()
     if lower_output_format == "dataframe":
-        return to_df(_chdb.query(sql, "Arrow", path=path, udf_path=udf_path))
-    elif lower_output_format == 'arrowtable':
-        return to_arrowTable(_chdb.query(sql, "Arrow", path=path, udf_path=udf_path))
+        return to_df(_chdb.query(sql, "Arrow", path=path, udf_path=g_udf_path))
+    elif lower_output_format == "arrowtable":
+        return to_arrowTable(_chdb.query(sql, "Arrow", path=path, udf_path=g_udf_path))
     else:
-        return _chdb.query(sql, output_format, path=path, udf_path=udf_path)
+        return _chdb.query(sql, output_format, path=path, udf_path=g_udf_path)