From 2adeb2aa26ae9b1c86666b586068383a2559074c Mon Sep 17 00:00:00 2001 From: Ivan Lin <35328415+ivanonair@users.noreply.github.com> Date: Thu, 20 Feb 2025 13:41:05 +0800 Subject: [PATCH] Update duckdb.py Allow duckdb to accept external argument for configurations. --- sqeleton/databases/duckdb.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sqeleton/databases/duckdb.py b/sqeleton/databases/duckdb.py index 154997d..7eeba8d 100644 --- a/sqeleton/databases/duckdb.py +++ b/sqeleton/databases/duckdb.py @@ -164,7 +164,8 @@ def close(self): def create_connection(self): ddb = import_duckdb() try: - return ddb.connect(self._args["filepath"]) + ddb_config = {key: value for key, value in self._args.items() if key not in ["driver", "filepath"]} + return ddb.connect(self._args["filepath"],config=ddb_config) except ddb.OperationalError as e: raise ConnectError(*e.args) from e