Function ast_from_object

crates/vm/src/stdlib/_ast/elif_else_clause.rs:51–104

(
    vm: &VirtualMachine,
    source_file: &SourceFile,
    object: PyObjectRef,
)

}

pub(super) fn ast_from_object(
    vm: &VirtualMachine,
    source_file: &SourceFile,
    object: PyObjectRef,
) -> PyResult<ast::StmtIf> {
    let test = Node::ast_from_object(vm, source_file, get_node_field(vm, &object, "test", "If")?)?;
    let body = Node::ast_from_object(vm, source_file, get_node_field(vm, &object, "body", "If")?)?;
    let orelse: Vec<ast::Stmt> = Node::ast_from_object(
        vm,
        source_file,
        get_node_field(vm, &object, "orelse", "If")?,
    )?;
    let range = range_from_object(vm, source_file, object, "If")?;

    let elif_else_clauses = if orelse.is_empty() {
        vec![]
    } else if let [ast::Stmt::If(_)] = &*orelse {
        let Some(ast::Stmt::If(ast::StmtIf {
            node_index: _,
            range,
            test,
            body,
            mut elif_else_clauses,
        })) = orelse.into_iter().next()
        else {
            unreachable!()
        };
        elif_else_clauses.insert(
            0,
            ast::ElifElseClause {
                node_index: Default::default(),
                range,
                test: Some(*test),
                body,
            },
        );
        elif_else_clauses
    } else {
        vec![ast::ElifElseClause {
            node_index: Default::default(),
            range,
            test: None,
            body: orelse,
        }]
    };

    Ok(ast::StmtIf {
        node_index: Default::default(),
        test,
        body,
        elif_else_clauses,
        range,
    })
}