Update duckdb.py

buremba · Oct 20, 2024 · 890ad48 · 890ad48
1 parent 392eff8
commit 890ad48
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/universql/warehouse/duckdb.py b/universql/warehouse/duckdb.py
@@ -268,9 +268,14 @@ def __init__(self, context: dict, query_id: str, credentials: dict, compute: dic
         if context.get('max_cache_size') != "0":
             duck_config['max_temp_directory_size'] = context.get('max_cache_size')
         self.account = parse_snowflake_account(context.get('account'))
+        database_path = self.context.get('database_path')
+        if database_path is not None:
+            database = os.path.join(database_path, "main.duckdb")
+        else:
+            database = ':memory:'
 
         try:
-            self.duckdb = duckdb.connect(":memory:", config=duck_config)
+            self.duckdb = duckdb.connect(database, config=duck_config)
         except duckdb.InvalidInputException as e:
             raise QueryError(f"Unable to spin up DuckDB with config {duck_config}: {e}")
         DuckDBFunctions.register(self.duckdb)