unix: Implement sys.argv.

pfalcon · pfalcon · commit fe2690da0a1b · 2014-01-20T01:00:28.000+02:00
diff --git a/tests/io/argv.py b/tests/io/argv.py
@@ -0,0 +1,2 @@
+import sys
+print(sys.argv)
diff --git a/unix/main.c b/unix/main.c
@@ -216,6 +216,10 @@ int main(int argc, char **argv) {
     qstr_init();
     rt_init();
 
+    mp_obj_t m_sys = mp_obj_new_module(qstr_from_str_static("sys"));
+    mp_obj_t py_argv = mp_obj_new_list(0, NULL);
+    rt_store_attr(m_sys, qstr_from_str_static("argv"), py_argv);
+
     rt_store_name(qstr_from_str_static("test"), test_obj_new(42));
     rt_store_name(qstr_from_str_static("open"), (mp_obj_t)&mp_builtin_open_obj);
     rawsocket_init();
@@ -254,7 +258,11 @@ int main(int argc, char **argv) {
                     return usage();
                 }
             } else {
+                for (int i = a; i < argc; i++) {
+                    rt_list_append(py_argv, MP_OBJ_NEW_QSTR(qstr_from_strn_copy(argv[i], strlen(argv[i]))));
+                }
                 do_file(argv[a]);
+                break;
             }
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -216,6 +216,10 @@ int main(int argc, char **argv) {`
`216`	`216`	`qstr_init();`
`217`	`217`	`rt_init();`
`218`	`218`
	`219`	`+ mp_obj_t m_sys = mp_obj_new_module(qstr_from_str_static("sys"));`
	`220`	`+ mp_obj_t py_argv = mp_obj_new_list(0, NULL);`
	`221`	`+ rt_store_attr(m_sys, qstr_from_str_static("argv"), py_argv);`
	`222`	`+`
`219`	`223`	`rt_store_name(qstr_from_str_static("test"), test_obj_new(42));`
`220`	`224`	`rt_store_name(qstr_from_str_static("open"), (mp_obj_t)&mp_builtin_open_obj);`
`221`	`225`	`rawsocket_init();`
`@@ -254,7 +258,11 @@ int main(int argc, char **argv) {`
`254`	`258`	`return usage();`
`255`	`259`	`}`
`256`	`260`	`} else {`
	`261`	`+ for (int i = a; i < argc; i++) {`
	`262`	`+ rt_list_append(py_argv, MP_OBJ_NEW_QSTR(qstr_from_strn_copy(argv[i], strlen(argv[i]))));`
	`263`	`+ }`
`257`	`264`	`do_file(argv[a]);`
	`265`	`+ break;`
`258`	`266`	`}`
`259`	`267`	`}`
`260`	`268`	`}`