Function extract_imports

plugins/ast/python/src/lib.rs:325–356

(body: &[Statement])

// ─── imports ───

fn extract_imports(body: &[Statement]) -> Vec<String> {
    let mut imports = Vec::new();

    for statement in body {
        match statement {
            Statement::Import(import) => {
                for alias in &import.names {
                    imports.push(alias.name.to_string());
                }
            }

            Statement::ImportFrom(import) => {
                let module = import
                    .module
                    .as_ref()
                    .map(ToString::to_string)
                    .unwrap_or_default();

                for alias in &import.names {
                    if module.is_empty() {
                        imports.push(alias.name.to_string());
                    } else {
                        imports.push(format!("{}.{}", module, alias.name));
                    }
                }
            }
            _ => {}
        }
    }

    imports
}

// ─── schemas ───