Function execute_duckdb_query

marimo/_data/get_datasets.py:106–127

Execute a DuckDB query and return the result. Uses connection if provided, otherwise uses duckdb.

(
    connection: duckdb.DuckDBPyConnection | None, query: str
)

def execute_duckdb_query(
    connection: duckdb.DuckDBPyConnection | None, query: str
) -> list[Any]:
    """Execute a DuckDB query and return the result. Uses connection if provided, otherwise uses duckdb."""
    try:
        if connection is None:
            import duckdb

            return duckdb.execute(query).fetchall()

        return connection.execute(query).fetchall()
    except Exception as e:
        if DependencyManager.duckdb.has():
            import duckdb

            # Connection is closed, return empty result
            if isinstance(e, duckdb.ConnectionException):
                LOGGER.debug("Skipping query on closed DuckDB connection")
                return []

        LOGGER.exception("Failed to execute DuckDB query %s", query)
        return []


def get_databases_from_duckdb(