sysmod.argv added + snippet

rmliddle · rmliddle · commit 302784df5175 · 2018-10-16T23:02:09.000+11:00
diff --git a/tests/snippets/sysmod_argv.py b/tests/snippets/sysmod_argv.py
@@ -0,0 +1,4 @@
+import sys
+
+print(sys.argv[1])
+
diff --git a/vm/src/sysmodule.rs b/vm/src/sysmodule.rs
@@ -1,11 +1,16 @@
-// use super::pyobject::{Executor, PyObject, PyObjectKind, PyObjectRef};
+use std::env;
+use super::pyobject::{DictProtocol, PyContext, PyObjectRef, PyFuncArgs, PyResult};
+use super::vm::VirtualMachine;
 
 /*
  * The magic sys module.
  */
-use std::env;
 
-use super::pyobject::{DictProtocol, PyContext, PyObjectRef};
+fn argv(ctx: &PyContext) -> PyObjectRef {
+    let argv: Vec<PyObjectRef> = env::args().map(|x| ctx.new_str(x)).collect();
+    ctx.new_list(argv)
+}
+
 
 pub fn mk_module(ctx: &PyContext) -> PyObjectRef {
     let path_list = match env::var_os("PYTHONPATH") {
@@ -20,6 +25,7 @@ pub fn mk_module(ctx: &PyContext) -> PyObjectRef {
     let sys_mod = ctx.new_module(&sys_name, ctx.new_scope(None));
     modules.set_item(&sys_name, sys_mod.clone());
     sys_mod.set_item(&"modules".to_string(), modules);
+    sys_mod.set_item(&"argv".to_string(), argv(ctx));
     sys_mod.set_item(&"path".to_string(), path);
     sys_mod
 }

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +import sys
++
 +print(sys.argv[1])
++