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
use crate::bytecode::FrozenModule;
use std::collections::HashMap;

pub fn get_module_inits() -> HashMap<String, FrozenModule> {
    let mut modules = HashMap::new();

    macro_rules! ext_modules {
        ($($t:tt)*) => {
            modules.extend(py_compile_bytecode!($($t)*));
        };
    }

    ext_modules!(
        source = "initialized = True; print(\"Hello world!\")\n",
        module_name = "__hello__",
    );

    // Python modules that the vm calls into, but are not actually part of the stdlib. They could
    // in theory be implemented in Rust, but are easiest to do in Python for one reason or another.
    // Includes _importlib_bootstrap and _importlib_bootstrap_external
    ext_modules!(dir = "Lib/python_builtins/");

    #[cfg(not(feature = "freeze-stdlib"))]
    {
        // core stdlib Python modules that the vm calls into, but are still used in Python
        // application code, e.g. copyreg
        ext_modules!(dir = "Lib/core_modules/");
    }
    // if we're on freeze-stdlib, the core stdlib modules will be included anyway
    #[cfg(feature = "freeze-stdlib")]
    {
        ext_modules!(dir = "../Lib/");
    }

    modules
}