Function populate_singletons

crates/vm/src/stdlib/_ast/pyast.rs:1789–1813

(vm: &VirtualMachine, module: &Py<PyModule>)

}

fn populate_singletons(vm: &VirtualMachine, module: &Py<PyModule>) {
    let instance_attr = vm.ctx.intern_str("_instance");
    const SINGLETON_TYPES: &[&str] = &[
        // expr_context
        "Load", "Store", "Del", // boolop
        "And", "Or", // operator
        "Add", "Sub", "Mult", "MatMult", "Div", "Mod", "Pow", "LShift", "RShift", "BitOr",
        "BitXor", "BitAnd", "FloorDiv", // unaryop
        "Invert", "Not", "UAdd", "USub", // cmpop
        "Eq", "NotEq", "Lt", "LtE", "Gt", "GtE", "Is", "IsNot", "In", "NotIn",
    ];

    for &class_name in SINGLETON_TYPES {
        let class = module
            .get_attr(class_name, vm)
            .unwrap_or_else(|_| panic!("AST class '{class_name}' not found in module"));
        let Some(type_obj) = class.downcast_ref::<PyType>() else {
            continue;
        };
        let instance = vm
            .ctx
            .new_base_object(type_obj.to_owned(), Some(vm.ctx.new_dict()));
        type_obj.set_attr(instance_attr, instance);
    }
}

fn force_ast_module_name(vm: &VirtualMachine, module: &Py<PyModule>) {
    let ast_name = vm.ctx.new_str("ast");