Skip to content

Commit ff2f088

Browse files
Saul Pwansoncpcloud
Saul Pwanson
authored andcommitted
feat(api): ibis.connect() default to duckdb for parquet/csv extensions
1 parent e13946b commit ff2f088

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

ibis/backends/base/__init__.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -686,3 +686,24 @@ def _(
686686
con = getattr(ibis, backend).connect(**kwargs)
687687
con.register(f"{extension}://{filename}")
688688
return con
689+
690+
691+
@_connect.register(
692+
r"(?P<filename>.+\.(?P<extension>parquet|csv))",
693+
priority=8,
694+
)
695+
def _(
696+
_: str,
697+
*,
698+
filename: str,
699+
extension: str,
700+
**kwargs: Any,
701+
) -> BaseBackend:
702+
"""Connect to `duckdb` and register a parquet or csv file.
703+
704+
Examples
705+
--------
706+
>>> con = ibis.connect("relative/path/to/data.csv")
707+
>>> con = ibis.connect("relative/path/to/more/data.parquet")
708+
"""
709+
return _connect(f"duckdb://{filename}", **kwargs)

0 commit comments

Comments
 (0)