rustpython-vm 0.6.0

RustPython virtual machine.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use super::*;
use rustpython_compiler_core::SourceFile;

fn ensure_excepthandler_node(vm: &VirtualMachine, object: &PyObject) -> PyResult<()> {
    if vm.is_none(object)
        || !is_node_instance(
            vm,
            object,
            pyast::NodeExceptHandlerExceptHandler::static_type(),
        )?
    {
        return Err(vm.new_type_error(format!(
            "expected some sort of excepthandler, but got {}",
            object.repr(vm)?
        )));
    }
    Ok(())
}

// sum
impl Node for ast::ExceptHandler {
    fn ast_to_object(self, vm: &VirtualMachine, source_file: &SourceFile) -> PyObjectRef {
        match self {
            Self::ExceptHandler(cons) => cons.ast_to_object(vm, source_file),
        }
    }
    fn ast_from_object(
        vm: &VirtualMachine,
        source_file: &SourceFile,
        object: PyObjectRef,
    ) -> PyResult<Self> {
        ensure_excepthandler_node(vm, &object)?;
        let range = excepthandler_range_from_object(vm, source_file, &object)?;
        Ok(Self::ExceptHandler(except_handler_from_object_with_range(
            vm,
            source_file,
            &object,
            range,
        )?))
    }
}

// constructor
fn except_handler_from_object_with_range(
    vm: &VirtualMachine,
    source_file: &SourceFile,
    object: &PyObject,
    range: TextRange,
) -> PyResult<ast::ExceptHandlerExceptHandler> {
    let body: Vec<Option<ast::Stmt>> =
        get_node_list_field(vm, source_file, object, "body", "ExceptHandler")?;
    let (runtime_body, body) = runtime_stmt_list_from_values(body);
    Ok(ast::ExceptHandlerExceptHandler {
        node_index: Default::default(),
        type_: get_node_field_opt(vm, object, "type")?
            .map(|obj| Node::ast_from_object(vm, source_file, obj))
            .transpose()?,
        name: get_node_field_opt(vm, object, "name")?
            .map(|obj| Node::ast_from_object(vm, source_file, obj))
            .transpose()?,
        body,
        range,
        runtime_body,
    })
}

pub(super) fn except_handler_from_object_unvalidated_range(
    vm: &VirtualMachine,
    source_file: &SourceFile,
    object: &PyObject,
) -> PyResult<ast::ExceptHandler> {
    ensure_excepthandler_node(vm, object)?;
    let range = excepthandler_range_from_object_unvalidated(vm, source_file, object)?;
    Ok(ast::ExceptHandler::ExceptHandler(
        except_handler_from_object_with_range(vm, source_file, object, range)?,
    ))
}

impl Node for ast::ExceptHandlerExceptHandler {
    fn ast_to_object(self, vm: &VirtualMachine, source_file: &SourceFile) -> PyObjectRef {
        let Self {
            node_index: _,
            type_,
            name,
            body,
            range,
            runtime_body,
        } = self;
        let node = NodeAst
            .into_ref_with_type(
                vm,
                pyast::NodeExceptHandlerExceptHandler::static_type().to_owned(),
            )
            .unwrap();
        let dict = node.as_object().dict().unwrap();
        dict.set_item("type", type_.ast_to_object(vm, source_file), vm)
            .unwrap();
        dict.set_item("name", name.ast_to_object(vm, source_file), vm)
            .unwrap();
        let body = runtime_body.map_or_else(
            || body.ast_to_object(vm, source_file),
            |values| values.ast_to_object(vm, source_file),
        );
        dict.set_item("body", body, vm).unwrap();
        node_add_location(&dict, range, vm, source_file);
        node.into()
    }

    fn ast_from_object(
        vm: &VirtualMachine,
        source_file: &SourceFile,
        object: PyObjectRef,
    ) -> PyResult<Self> {
        let range = range_from_object(vm, source_file, &object, "ExceptHandler")?;
        except_handler_from_object_with_range(vm, source_file, &object, range)
    }
}