Skip to content

Commit 1622038

Browse files
committed
Feed s3 configuration to duckdb engine
Signed-off-by: Wei-Chun, Chang <[email protected]>
1 parent a53e770 commit 1622038

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

piperider_cli/datasource/duckdb.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,15 @@ def to_database_url(self, database):
104104
raise PipeRiderDataBaseConnectionError(self.name, self.type_name, db_path=duckdb_path)
105105
return f"duckdb:///{duckdb_path}"
106106

107+
def engine_args(self):
108+
109+
args = dict(
110+
preload_extensions=list(self.credential.get('extensions', [])),
111+
config=dict(self.credential.get('settings', {}))
112+
)
113+
114+
return dict(connect_args=args)
115+
107116
def _formalize_table_name(self, name):
108117
if not name:
109118
return f'{self.type_name}_table'

0 commit comments

Comments
 (0)