Function execute_duckdb_sql

marimo/_sql/utils.py:147–200

Execute a parameterized DuckDB query with kernel globals context. Like wrapped_sql, but uses connection.execute() to support parameterized queries ($1, $2, ...) for safe value interpolation.

(
    query: str,
    params: list[Any],
    connection: duckdb.DuckDBPyConnection | None = None,
)

def execute_duckdb_sql(
    query: str,
    params: list[Any],
    connection: duckdb.DuckDBPyConnection | None = None,
) -> duckdb.DuckDBPyConnection:
    """Execute a parameterized DuckDB query with kernel globals context.

    Like wrapped_sql, but uses connection.execute() to support
    parameterized queries ($1, $2, ...) for safe value interpolation.
    """
    DependencyManager.duckdb.require("to execute sql")
    import duckdb

    if connection is None:
        connection = cast(duckdb.DuckDBPyConnection, duckdb)

    try:
        ctx = get_context()
    except ContextNotInitializedError:
        result = _try_wasm_duckdb(
            "execute", query, connection, globals(), params
        )
        if result is _NO_WASM_DUCKDB_RESULT:
            # No WASM rewrite was needed; preserve DuckDB's parameterized path.
            return connection.execute(query, params)
        return cast(duckdb.DuckDBPyConnection, result)
    else:
        install_connection = (
            ctx.execution_context.with_connection
            if ctx.execution_context is not None
            else nullcontext
        )
        with install_connection(connection):
            result = _try_wasm_duckdb(
                "execute",
                query,
                connection,
                ctx.globals,
                params,
            )
            if result is _NO_WASM_DUCKDB_RESULT:
                # Run in kernel globals so parameterized SQL can scan user data.
                value = eval(
                    "connection.execute(query, params)",
                    ctx.globals,
                    {
                        "query": query,
                        "params": params,
                        "connection": connection,
                    },
                )
            else:
                value = result
            return cast(duckdb.DuckDBPyConnection, value)


def try_convert_to_polars(