bugfix [#24087] Blender can not install add-ons unless running with root priviledges
[blender.git] / source / blender / python / intern / bpy.c
index 7933c2a..69b5fac 100644 (file)
@@ -41,6 +41,8 @@
 #include "../generic/blf_api.h"
 #include "../generic/IDProp.h"
 
+#include "AUD_PyInit.h"
+
 static char bpy_script_paths_doc[] =
 ".. function:: script_paths()\n"
 "\n"
@@ -65,7 +67,7 @@ PyObject *bpy_script_paths(PyObject *self)
 static char bpy_blend_paths_doc[] =
 ".. function:: blend_paths(absolute=False)\n"
 "\n"
-"   Returns a list of paths associated with this blend file.\n"
+"   Returns a list of paths to external files referenced by the loaded .blend file.\n"
 "\n"
 "   :arg absolute: When true the paths returned are made absolute.\n"
 "   :type absolute: boolean\n"
@@ -110,8 +112,52 @@ static PyObject *bpy_blend_paths(PyObject * self, PyObject *args, PyObject *kw)
        return list;
 }
 
-static PyMethodDef meth_bpy_script_paths[] = {{ "script_paths", (PyCFunction)bpy_script_paths, METH_NOARGS, bpy_script_paths_doc}};
-static PyMethodDef meth_bpy_blend_paths[] = {{ "blend_paths", (PyCFunction)bpy_blend_paths, METH_VARARGS|METH_KEYWORDS, bpy_blend_paths_doc}};
+
+static char bpy_user_resource_doc[] =
+".. function:: user_resource(type, subdir)\n"
+"\n"
+"   Returns a list of paths to external files referenced by the loaded .blend file.\n"
+"\n"
+"   :arg type: Resource type in ['DATAFILES', 'CONFIG', 'SCRIPTS', 'AUTOSAVE'].\n"
+"   :type type: string\n"
+"   :arg subdir: Optional subdirectory.\n"
+"   :type subdir: string\n"
+"   :return: a path.\n"
+"   :rtype: string\n";
+static PyObject *bpy_user_resource(PyObject * self, PyObject *args, PyObject *kw)
+{
+       char *type;
+       char *subdir= NULL;
+       int folder_id;
+       static char *kwlist[] = {"type", "subdir", NULL};
+
+       char *path;
+
+       if (!PyArg_ParseTupleAndKeywords(args, kw, "s|s:user_resource", kwlist, &type, &subdir))
+               return NULL;
+       
+       /* stupid string compare */
+       if     (!strcmp(type, "DATAFILES"))     folder_id= BLENDER_USER_DATAFILES;
+       else if(!strcmp(type, "CONFIG"))        folder_id= BLENDER_USER_CONFIG;
+       else if(!strcmp(type, "SCRIPTS"))       folder_id= BLENDER_USER_SCRIPTS;
+       else if(!strcmp(type, "AUTOSAVE"))      folder_id= BLENDER_USER_AUTOSAVE;
+       else {
+               PyErr_SetString(PyExc_ValueError, "invalid resource argument");
+               return NULL;
+       }
+       
+       /* same logic as BLI_get_folder_create(), but best leave it up to the script author to create */
+       path= BLI_get_folder(folder_id, subdir);
+
+       if (!path)
+               path = BLI_get_user_folder_notest(folder_id, subdir);
+
+       return PyUnicode_FromString(path ? path : "");
+}
+
+static PyMethodDef meth_bpy_script_paths = {"script_paths", (PyCFunction)bpy_script_paths, METH_NOARGS, bpy_script_paths_doc};
+static PyMethodDef meth_bpy_blend_paths = {"blend_paths", (PyCFunction)bpy_blend_paths, METH_VARARGS|METH_KEYWORDS, bpy_blend_paths_doc};
+static PyMethodDef meth_bpy_user_resource = {"user_resource", (PyCFunction)bpy_user_resource, METH_VARARGS|METH_KEYWORDS, bpy_user_resource_doc};
 
 static void bpy_import_test(char *modname)
 {
@@ -153,18 +199,20 @@ void BPy_init_modules( void )
        BGL_Init();
        BLF_Init();
        IDProp_Init_Types();
-
+       AUD_initPython();
 
        mod = PyModule_New("_bpy");
 
        /* add the module so we can import it */
-       PyDict_SetItemString(PySys_GetObject("modules"), "_bpy", mod);
+       PyDict_SetItemString(PyImport_GetModuleDict(), "_bpy", mod);
        Py_DECREF(mod);
 
        /* run first, initializes rna types */
        BPY_rna_init();
 
        PyModule_AddObject( mod, "types", BPY_rna_types() ); /* needs to be first so bpy_types can run */
+       PyModule_AddObject(mod, "StructMetaIDProp", (PyObject *)&pyrna_struct_meta_idprop_Type); /* metaclass for idprop types, bpy_types.py needs access */
+                       
        bpy_import_test("bpy_types");
        PyModule_AddObject( mod, "data", BPY_rna_module() ); /* imports bpy_types by running this */
        bpy_import_test("bpy_types");
@@ -182,8 +230,9 @@ void BPy_init_modules( void )
        PyModule_AddObject(mod, "context", (PyObject *)bpy_context_module);
 
        /* utility func's that have nowhere else to go */
-       PyModule_AddObject(mod, meth_bpy_script_paths->ml_name, (PyObject *)PyCFunction_New(meth_bpy_script_paths, NULL));
-       PyModule_AddObject(mod, meth_bpy_blend_paths->ml_name, (PyObject *)PyCFunction_New(meth_bpy_blend_paths, NULL));
+       PyModule_AddObject(mod, meth_bpy_script_paths.ml_name, (PyObject *)PyCFunction_New(&meth_bpy_script_paths, NULL));
+       PyModule_AddObject(mod, meth_bpy_blend_paths.ml_name, (PyObject *)PyCFunction_New(&meth_bpy_blend_paths, NULL));
+       PyModule_AddObject(mod, meth_bpy_user_resource.ml_name, (PyObject *)PyCFunction_New(&meth_bpy_user_resource, NULL));
 
        /* add our own modules dir, this is a python package */
        bpy_import_test("bpy");