RustPython-RustPython/vm/src/stdlib/mod.rs at 4cf2672876d41b761d75e8ee87ea6a6f548f6a88 · sthagen/RustPython-RustPython · GitHub

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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
pub mod array;
#[cfg(feature = "rustpython-parser")]
pub(crate) mod ast;
mod binascii;
mod codecs;
mod collections;
mod csv;
mod dis;
mod errno;
mod functools;
mod hashlib;
mod imp;
mod itertools;
mod json;
#[cfg(feature = "rustpython-parser")]
mod keyword;
mod marshal;
mod math;
mod platform;
mod pystruct;
mod random;
mod re;
#[cfg(not(target_arch = "wasm32"))]
pub mod socket;
mod string;
#[cfg(feature = "rustpython-compiler")]
mod symtable;
mod thread;
mod time_module;
#[cfg(feature = "rustpython-parser")]
mod tokenize;
mod unicodedata;
mod warnings;
mod weakref;
use std::collections::HashMap;

use crate::vm::VirtualMachine;

#[cfg(not(target_arch = "wasm32"))]
pub mod io;
#[cfg(not(target_arch = "wasm32"))]
mod multiprocessing;
#[cfg(not(target_arch = "wasm32"))]
mod os;
#[cfg(all(unix, not(any(target_os = "android", target_os = "redox"))))]
mod pwd;
#[cfg(not(target_arch = "wasm32"))]
mod select;
#[cfg(not(target_arch = "wasm32"))]
pub mod signal;
#[cfg(not(target_arch = "wasm32"))]
mod subprocess;
#[cfg(windows)]
mod winapi;
#[cfg(not(target_arch = "wasm32"))]
mod zlib;

use crate::pyobject::PyObjectRef;

pub type StdlibInitFunc = Box<dyn Fn(&VirtualMachine) -> PyObjectRef>;

pub fn get_module_inits() -> HashMap<String, StdlibInitFunc> {
    #[allow(unused_mut)]
    let mut modules = hashmap! {
        "array".to_string() => Box::new(array::make_module) as StdlibInitFunc,
        "binascii".to_string() => Box::new(binascii::make_module),
        "dis".to_string() => Box::new(dis::make_module),
        "_codecs".to_string() => Box::new(codecs::make_module),
        "_collections".to_string() => Box::new(collections::make_module),
        "_csv".to_string() => Box::new(csv::make_module),
        "_functools".to_string() => Box::new(functools::make_module),
        "errno".to_string() => Box::new(errno::make_module),
        "hashlib".to_string() => Box::new(hashlib::make_module),
        "itertools".to_string() => Box::new(itertools::make_module),
        "json".to_string() => Box::new(json::make_module),
        "marshal".to_string() => Box::new(marshal::make_module),
        "math".to_string() => Box::new(math::make_module),
        "platform".to_string() => Box::new(platform::make_module),
        "regex_crate".to_string() => Box::new(re::make_module),
        "random".to_string() => Box::new(random::make_module),
        "_string".to_string() => Box::new(string::make_module),
        "struct".to_string() => Box::new(pystruct::make_module),
        "_thread".to_string() => Box::new(thread::make_module),
        "time".to_string() => Box::new(time_module::make_module),
        "_weakref".to_string() => Box::new(weakref::make_module),
        "_imp".to_string() => Box::new(imp::make_module),
        "unicodedata".to_string() => Box::new(unicodedata::make_module),
        "_warnings".to_string() => Box::new(warnings::make_module),
    };

    // Insert parser related modules:
    #[cfg(feature = "rustpython-parser")]
    {
        modules.insert(
            "_ast".to_string(),
            Box::new(ast::make_module) as StdlibInitFunc,
        );
        modules.insert("keyword".to_string(), Box::new(keyword::make_module));
        modules.insert("tokenize".to_string(), Box::new(tokenize::make_module));
    }

    // Insert compiler related modules:
    #[cfg(feature = "rustpython-compiler")]
    {
        modules.insert("symtable".to_string(), Box::new(symtable::make_module));
    }

    // disable some modules on WASM
    #[cfg(not(target_arch = "wasm32"))]
    {
        modules.insert("_io".to_string(), Box::new(io::make_module));
        modules.insert("_os".to_string(), Box::new(os::make_module));
        modules.insert("_socket".to_string(), Box::new(socket::make_module));
        modules.insert(
            "_multiprocessing".to_string(),
            Box::new(multiprocessing::make_module),
        );
        modules.insert("signal".to_string(), Box::new(signal::make_module));
        modules.insert("select".to_string(), Box::new(select::make_module));
        modules.insert("_subprocess".to_string(), Box::new(subprocess::make_module));
        modules.insert("zlib".to_string(), Box::new(zlib::make_module));
    }

    // Unix-only
    #[cfg(all(unix, not(any(target_os = "android", target_os = "redox"))))]
    {
        modules.insert("pwd".to_string(), Box::new(pwd::make_module));
    }

    // Windows-only
    #[cfg(windows)]
    {
        modules.insert("_winapi".to_string(), Box::new(winapi::make_module));
    }

    modules
}