2007-12-24 08:54:30 +03:00
/*
Unix SMB / CIFS implementation .
Samba utility functions
Copyright ( C ) Jelmer Vernooij < jelmer @ samba . org > 2007
This program is free software ; you can redistribute it and / or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation ; either version 3 of the License , or
( at your option ) any later version .
This program is distributed in the hope that it will be useful ,
but WITHOUT ANY WARRANTY ; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the
GNU General Public License for more details .
You should have received a copy of the GNU General Public License
along with this program . If not , see < http : //www.gnu.org/licenses/>.
*/
2009-10-23 09:23:01 +04:00
# include <Python.h>
2017-04-20 16:11:58 +03:00
# include "py3compat.h"
2007-12-24 08:54:30 +03:00
# include "includes.h"
2012-12-28 18:37:14 +04:00
# include "python/modules.h"
2010-04-20 14:24:08 +04:00
# include "dynconfig/dynconfig.h"
2007-12-24 08:54:30 +03:00
2009-12-29 18:07:54 +03:00
static bool PySys_PathPrepend ( PyObject * list , const char * path )
{
2018-05-03 01:45:39 +03:00
bool ok ;
2017-04-20 16:11:58 +03:00
PyObject * py_path = PyStr_FromString ( path ) ;
2018-05-03 01:45:39 +03:00
if ( py_path = = NULL ) {
2009-12-29 18:07:54 +03:00
return false ;
2018-05-03 01:45:39 +03:00
}
ok = PyList_Insert ( list , 0 , py_path ) = = 0 ;
Py_XDECREF ( py_path ) ;
return ok ;
2009-12-29 18:07:54 +03:00
}
2011-02-21 13:47:17 +03:00
bool py_update_path ( void )
2007-12-26 01:36:36 +03:00
{
2018-05-03 01:45:39 +03:00
PyObject * mod_sys = NULL ;
PyObject * py_path = NULL ;
2009-12-29 18:07:54 +03:00
mod_sys = PyImport_ImportModule ( " sys " ) ;
if ( mod_sys = = NULL ) {
return false ;
}
py_path = PyObject_GetAttrString ( mod_sys , " path " ) ;
if ( py_path = = NULL ) {
2018-05-03 01:45:39 +03:00
goto error ;
}
2009-12-29 18:07:54 +03:00
if ( ! PyList_Check ( py_path ) ) {
2018-05-03 01:45:39 +03:00
goto error ;
2009-12-29 18:07:54 +03:00
}
2011-02-07 01:45:26 +03:00
if ( ! PySys_PathPrepend ( py_path , dyn_PYTHONDIR ) ) {
2018-05-03 01:45:39 +03:00
goto error ;
2011-01-14 09:20:01 +03:00
}
2011-02-07 01:45:26 +03:00
if ( strcmp ( dyn_PYTHONARCHDIR , dyn_PYTHONDIR ) ! = 0 ) {
if ( ! PySys_PathPrepend ( py_path , dyn_PYTHONARCHDIR ) ) {
2018-05-03 01:45:39 +03:00
goto error ;
2011-02-07 01:45:26 +03:00
}
2010-04-20 14:24:08 +04:00
}
2018-05-03 01:45:39 +03:00
Py_XDECREF ( py_path ) ;
Py_XDECREF ( mod_sys ) ;
2009-12-29 18:07:54 +03:00
return true ;
2018-05-03 01:45:39 +03:00
error :
Py_XDECREF ( py_path ) ;
Py_XDECREF ( mod_sys ) ;
return false ;
2007-12-26 01:36:36 +03:00
}