Function wrapped_sql

marimo/_sql/utils.py:96–144

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

def wrapped_sql(
    query: str,
    connection: duckdb.DuckDBPyConnection | None,
) -> duckdb.DuckDBPyRelation:
    DependencyManager.duckdb.require("to execute sql")
    import duckdb

    # In Python globals() are scoped to modules; since this function
    # is in a different module than user code, globals() doesn't return
    # the kernel globals, it just returns this module's global namespace.
    #
    # However, duckdb needs access to the kernel's globals. For this reason,
    # we manually exec duckdb and provide it with the kernel's globals.
    if connection is None:
        connection = cast(duckdb.DuckDBPyConnection, duckdb)

    try:
        ctx = get_context()
    except ContextNotInitializedError:
        result = _try_wasm_duckdb("sql", query, connection, globals())
        if result is _NO_WASM_DUCKDB_RESULT:
            # No WASM rewrite was needed; use DuckDB's normal SQL path.
            relation = connection.sql(query=query)
        else:
            relation = cast(duckdb.DuckDBPyRelation, 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(
                "sql",
                query,
                connection,
                ctx.globals,
            )
            if result is _NO_WASM_DUCKDB_RESULT:
                # Run in kernel globals so DuckDB replacement scans see user data.
                relation = eval(
                    "connection.sql(query=query)",
                    ctx.globals,
                    {"query": query, "connection": connection},
                )
            else:
                relation = cast(duckdb.DuckDBPyRelation, result)

    return relation


def execute_duckdb_sql(