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

use crate::bytecode;

/// `Py_single_input` / `Py_file_input` / `Py_eval_input` / `Py_func_type_input`.
#[repr(i32)]
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
pub(crate) enum CompileStart {
    Single = 256,
    File = 257,
    Eval = 258,
    FuncType = 345,
}

impl CompileStart {
    pub(crate) const fn from_i32(v: i32) -> Option<Self> {
        match v {
            256 => Some(Self::Single),
            257 => Some(Self::File),
            258 => Some(Self::Eval),
            345 => Some(Self::FuncType),
            _ => None,
        }
    }

    pub(crate) const fn as_i32(self) -> i32 {
        self as i32
    }
}

bitflags::bitflags! {
    /// `PyCF_*` compiler flags together with the `__future__` `CO_FUTURE_*`
    /// bits, mirroring `PyCompilerFlags.cf_flags`.
    ///
    /// Caveat emptor: these flags are undocumented on purpose and depending on
    /// their effect outside the standard library is **unsupported**.
    #[derive(Copy, Clone, Debug, PartialEq, Eq)]
    pub(crate) struct CompilerFlags: i32 {
        const SOURCE_IS_UTF8 = 0x0100;
        const DONT_IMPLY_DEDENT = 0x0200;
        const ONLY_AST = 0x0400;
        const IGNORE_COOKIE = 0x0800;
        const TYPE_COMMENTS = 0x1000;
        const ALLOW_TOP_LEVEL_AWAIT = 0x2000;
        const ALLOW_INCOMPLETE_INPUT = 0x4000;
        const OPTIMIZED_AST = 0x8000 | Self::ONLY_AST.bits();

        // __future__ flags - sync with Lib/__future__.py and Include/cpython/compile.h.
        const NESTED = 0x0010;
        const FUTURE_DIVISION = 0x20000;
        const FUTURE_ABSOLUTE_IMPORT = 0x40000;
        const FUTURE_WITH_STATEMENT = 0x80000;
        const FUTURE_PRINT_FUNCTION = 0x100000;
        const FUTURE_UNICODE_LITERALS = 0x200000;
        const FUTURE_BARRY_AS_BDFL = 0x400000;
        const FUTURE_GENERATOR_STOP = 0x800000;
        const FUTURE_ANNOTATIONS = 0x1000000;
    }
}

impl CompilerFlags {
    const FUTURE_MASK: Self = Self::FUTURE_DIVISION
        .union(Self::FUTURE_ABSOLUTE_IMPORT)
        .union(Self::FUTURE_WITH_STATEMENT)
        .union(Self::FUTURE_PRINT_FUNCTION)
        .union(Self::FUTURE_UNICODE_LITERALS)
        .union(Self::FUTURE_BARRY_AS_BDFL)
        .union(Self::FUTURE_GENERATOR_STOP)
        .union(Self::FUTURE_ANNOTATIONS);
    const MASK_OBSOLETE: Self = Self::NESTED;
    const COMPILE_MASK: Self = Self::ONLY_AST
        .union(Self::ALLOW_TOP_LEVEL_AWAIT)
        .union(Self::TYPE_COMMENTS)
        .union(Self::DONT_IMPLY_DEDENT)
        .union(Self::ALLOW_INCOMPLETE_INPUT)
        .union(Self::OPTIMIZED_AST);
    pub(crate) const ALLOWED_FLAGS: Self = Self::FUTURE_MASK
        .union(Self::MASK_OBSOLETE)
        .union(Self::COMPILE_MASK);
}

// Python-visible `ast.PyCF_*` attribute values. The flags cross the
// `compile()` boundary as a plain `int`, so the exposed surface stays `i32`.
pub(crate) const PY_CF_SOURCE_IS_UTF8: i32 = CompilerFlags::SOURCE_IS_UTF8.bits();
pub(crate) const PY_CF_DONT_IMPLY_DEDENT: i32 = CompilerFlags::DONT_IMPLY_DEDENT.bits();
pub(crate) const PY_CF_ONLY_AST: i32 = CompilerFlags::ONLY_AST.bits();
pub(crate) const PY_CF_IGNORE_COOKIE: i32 = CompilerFlags::IGNORE_COOKIE.bits();
pub(crate) const PY_CF_TYPE_COMMENTS: i32 = CompilerFlags::TYPE_COMMENTS.bits();
pub(crate) const PY_CF_ALLOW_TOP_LEVEL_AWAIT: i32 = CompilerFlags::ALLOW_TOP_LEVEL_AWAIT.bits();
pub(crate) const PY_CF_ALLOW_INCOMPLETE_INPUT: i32 = CompilerFlags::ALLOW_INCOMPLETE_INPUT.bits();
pub(crate) const PY_CF_OPTIMIZED_AST: i32 = CompilerFlags::OPTIMIZED_AST.bits();

pub(crate) fn compile_future_features_from_flags(flags: i32) -> bytecode::CodeFlags {
    bytecode::CodeFlags::from_bits_truncate(flags as u32 & bytecode::CodeFlags::FUTURE_MASK.bits())
}