Function build_user_block

catalog/code/:exec/python/backend.rs:206–226

Shared prelude: turns inputs into Python variables and wraps the user code in a function so `return` works. `from X import *` lines are hoisted to module scope because Python 3 disallows them inside functions.

(user_code: &str, input: &Value)

/// in a function so `return` works. `from X import *` lines are hoisted to
/// module scope because Python 3 disallows them inside functions.
fn build_user_block(user_code: &str, input: &Value) -> (String, String, String) {
    let inputs_json = serde_json::to_string(input).unwrap_or_else(|_| "{}".into());
    let (star_imports, clean_code) = extract_star_imports(user_code);
    let indented = indent_code(&clean_code, "    ");
    let var_setup = match input.as_object() {
        Some(obj) => obj.keys()
            .map(|k| {
                let safe = sanitize_identifier(k);
                format!("{} = __weft_inputs__.get({})", safe, python_str_literal(k))
            })
            .collect::<Vec<_>>()
            .join("\n"),
        None => String::new(),
    };
    let prelude = format!(
        "__weft_inputs__ = __weft_json__.loads({inputs})\n{vars}",
        inputs = python_str_literal(&inputs_json),
        vars = var_setup,
    );
    (star_imports, prelude, indented)
}

/// Build the full Python script that gets sent to E2B's /execute endpoint.
/// The function's return dict is rendered as the cell's last expression wrapped