diff options
author | Jelmer Vernooij <jelmer@samba.org> | 2009-12-29 16:07:54 +0100 |
---|---|---|
committer | Jelmer Vernooij <jelmer@ganieda.vernstok.nl> | 2009-12-29 16:26:20 +0100 |
commit | 588b3e61812978f73d2708ec37da30726ac8026e (patch) | |
tree | 0181a51f24ace87a41ac09e27cb54d8ce3a6f6c4 /source4/scripting | |
parent | e2c4d8281d726716a00cfe2e3e0352777fc8b66f (diff) | |
download | samba-588b3e61812978f73d2708ec37da30726ac8026e.tar.gz samba-588b3e61812978f73d2708ec37da30726ac8026e.tar.bz2 samba-588b3e61812978f73d2708ec37da30726ac8026e.zip |
python: When updating sys.path to include the Samba python path, avoid throwing away the changes made by site.py.
Diffstat (limited to 'source4/scripting')
-rw-r--r-- | source4/scripting/python/modules.c | 43 | ||||
-rw-r--r-- | source4/scripting/python/modules.h | 2 |
2 files changed, 41 insertions, 4 deletions
diff --git a/source4/scripting/python/modules.c b/source4/scripting/python/modules.c index 5365c5007d..198b3ccf12 100644 --- a/source4/scripting/python/modules.c +++ b/source4/scripting/python/modules.c @@ -61,10 +61,47 @@ void py_load_samba_modules(void) } } -void py_update_path(const char *bindir) +static bool PySys_PathPrepend(PyObject *list, const char *path) +{ + PyObject *py_path = PyString_FromString(path); + if (py_path == NULL) + return false; + + return (PyList_Insert(list, 0, py_path) == 0); +} + +bool py_update_path(const char *bindir) { char *newpath; - asprintf(&newpath, "%s/python:%s/../scripting/python:%s", bindir, bindir, Py_GetPath()); - PySys_SetPath(newpath); + PyObject *mod_sys, *py_path; + + mod_sys = PyImport_ImportModule("sys"); + if (mod_sys == NULL) { + return false; + } + + py_path = PyObject_GetAttrString(mod_sys, "path"); + if (py_path == NULL) { + return false; + } + + if (!PyList_Check(py_path)) { + return false; + } + + asprintf(&newpath, "%s/../scripting/python", bindir); + if (!PySys_PathPrepend(py_path, newpath)) { + free(newpath); + return false; + } free(newpath); + + asprintf(&newpath, "%s/python", bindir); + if (!PySys_PathPrepend(py_path, newpath)) { + free(newpath); + return false; + } + free(newpath); + + return true; } diff --git a/source4/scripting/python/modules.h b/source4/scripting/python/modules.h index 6b242ee257..4d1067cdd4 100644 --- a/source4/scripting/python/modules.h +++ b/source4/scripting/python/modules.h @@ -21,7 +21,7 @@ #define __SAMBA_PYTHON_MODULES_H__ void py_load_samba_modules(void); -void py_update_path(const char *bindir); +bool py_update_path(const char *bindir); #define py_iconv_convenience(mem_ctx) smb_iconv_convenience_init(mem_ctx, "ASCII", PyUnicode_GetDefaultEncoding(), true) |