Function process_py_import_from

internal/cbm/extract_imports.c:259–304

Process a single Python import_from_statement node (from X import Y [as Z]).

// Process a single Python import_from_statement node (from X import Y [as Z]).
static void process_py_import_from(CBMExtractCtx *ctx, TSNode node) {
    CBMArena *a = ctx->arena;
    // `from __future__ import annotations` is a dedicated node type whose first
    // child is the literal `__future__` keyword (an identifier, not a
    // dotted_name).  Emit a single import for `__future__` and return.
    if (strcmp(ts_node_type(node), "future_import_statement") == 0) {
        CBMImport imp = {.local_name = cbm_arena_strdup(a, "__future__"),
                         .module_path = cbm_arena_strdup(a, "__future__")};
        cbm_imports_push(&ctx->result->imports, a, imp);
        return;
    }
    TSNode module_node = resolve_py_module_node(node);
    char *mod_path =
        ts_node_is_null(module_node) ? NULL : cbm_node_text(a, module_node, ctx->source);

    uint32_t nc = ts_node_child_count(node);
    bool emitted = false;
    for (uint32_t j = 0; j < nc; j++) {
        TSNode child = ts_node_child(node, j);
        const char *ck = ts_node_type(child);
        if (strcmp(ck, "identifier") == 0 || strcmp(ck, "dotted_name") == 0) {
            if (!ts_node_is_null(module_node) &&
                ts_node_start_byte(child) == ts_node_start_byte(module_node)) {
                continue;
            }
            emit_py_import_from_name(ctx, child, mod_path);
            emitted = true;
        } else if (strcmp(ck, "aliased_import") == 0) {
            emit_py_aliased_import(ctx, child, mod_path);
            emitted = true;
        } else if (strcmp(ck, "wildcard_import") == 0) {
            // `from os.path import *` — the module itself is the import.
            if (mod_path && mod_path[0]) {
                CBMImport imp = {.local_name = path_last(a, mod_path), .module_path = mod_path};
                cbm_imports_push(&ctx->result->imports, a, imp);
                emitted = true;
            }
        }
    }
    // Defensive: a from-import with a module but no recognized name child
    // (grammar variant) still records the module as an import.
    if (!emitted && mod_path && mod_path[0]) {
        CBMImport imp = {.local_name = path_last(a, mod_path), .module_path = mod_path};
        cbm_imports_push(&ctx->result->imports, a, imp);
    }
}

static void parse_python_imports(CBMExtractCtx *ctx) {
    TSTreeCursor cursor = ts_tree_cursor_new(ctx->root);