Add initial platform module.

windelbouwman · windelbouwman · commit f2e60b24e18e · 2019-02-26T21:42:38.000+01:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/vm/Cargo.toml b/vm/Cargo.toml
@@ -18,6 +18,7 @@ serde_derive = "1.0.66"
 serde_json = "1.0.26"
 byteorder = "1.2.6"
 regex = "1"
+rustc_version_runtime = "0.1.*"
 statrs = "0.10.0"
 caseless = "0.2.1"
 unicode-segmentation = "1.2.1"
diff --git a/vm/src/stdlib/mod.rs b/vm/src/stdlib/mod.rs
@@ -3,6 +3,7 @@ mod dis;
 mod json;
 mod keyword;
 mod math;
+mod platform;
 mod pystruct;
 mod random;
 mod re;
@@ -33,6 +34,7 @@ pub fn get_module_inits() -> HashMap<String, StdlibInitFunc> {
     modules.insert("json".to_string(), Box::new(json::mk_module));
     modules.insert("keyword".to_string(), Box::new(keyword::mk_module));
     modules.insert("math".to_string(), Box::new(math::mk_module));
+    modules.insert("platform".to_string(), Box::new(platform::mk_module));
     modules.insert("re".to_string(), Box::new(re::mk_module));
     modules.insert("random".to_string(), Box::new(random::mk_module));
     modules.insert("string".to_string(), Box::new(string::mk_module));
diff --git a/vm/src/stdlib/platform.rs b/vm/src/stdlib/platform.rs
@@ -0,0 +1,29 @@
+extern crate rustc_version_runtime;
+
+use crate::pyobject::{PyContext, PyFuncArgs, PyObjectRef, PyResult};
+use crate::VirtualMachine;
+
+pub fn mk_module(ctx: &PyContext) -> PyObjectRef {
+    py_module!(ctx, "platform", {
+        "python_compiler" => ctx.new_rustfunc(platform_python_compiler),
+        "python_implementation" => ctx.new_rustfunc(platform_python_implementation),
+        "python_version" => ctx.new_rustfunc(platform_python_version),
+    })
+}
+
+fn platform_python_implementation(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args);
+    Ok(vm.new_str("RustPython".to_string()))
+}
+
+fn platform_python_version(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args);
+    // TODO: fetch version from somewhere.
+    Ok(vm.new_str("4.0.0".to_string()))
+}
+
+fn platform_python_compiler(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(vm, args);
+    let version = rustc_version_runtime::version_meta();
+    Ok(vm.new_str(format!("rustc {}", version.semver)))
+}
