djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

unix: Initialize sys.path from MICROPYPATH environment variable.

Browse Source 
If it's not available, "~/.micropython/lib:/usr/lib/micropython" is used
as a fallback.
This commit is contained in:
Paul Sokolovsky 2014-02-05 00:59:54 +02:00
parent e11b17c25f
commit 625d08a93e
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
unix/main.c
View File

@ -233,7 +233,41 @@ int main(int argc, char **argv) {
qstr_init(); qstr_init();
rt_init(); rt_init();
char *home = getenv("HOME");
char *path = getenv("MICROPYPATH");
if (path == NULL) {
path = "~/.micropython/lib:/usr/lib/micropython";
}
uint path_num = 0;
for (char *p = path; p != NULL; p = strchr(p, ':')) {
path_num++;
if (p != NULL) {
p++;
}
}
sys_path = mp_obj_new_list(path_num, NULL);
mp_obj_t *items;
mp_obj_list_get(sys_path, &path_num, &items);
char *p = path;
for (int i = 0; i < path_num; i++) {
char *p1 = strchr(p, ':');
if (p1 == NULL) {
p1 = p + strlen(p);
}
if (p[0] == '~' && p[1] == '/' && home != NULL) {
// Expand standalone ~ to $HOME
CHECKBUF(buf, PATH_MAX);
CHECKBUF_APPEND(buf, home, strlen(home));
CHECKBUF_APPEND(buf, p + 1, p1 - p - 1);
items[i] = MP_OBJ_NEW_QSTR(qstr_from_strn(buf, CHECKBUF_LEN(buf)));
} else {
items[i] = MP_OBJ_NEW_QSTR(qstr_from_strn(p, p1 - p));
}
p = p1 + 1;
}
mp_obj_t m_sys = mp_obj_new_module(MP_QSTR_sys); mp_obj_t m_sys = mp_obj_new_module(MP_QSTR_sys);
rt_store_attr(m_sys, MP_QSTR_path, sys_path);
mp_obj_t py_argv = mp_obj_new_list(0, NULL); mp_obj_t py_argv = mp_obj_new_list(0, NULL);
rt_store_attr(m_sys, MP_QSTR_argv, py_argv); rt_store_attr(m_sys, MP_QSTR_argv, py_argv);