Merge branch 'master' into blender2.8
[blender.git] / source / blender / python / intern / bpy_interface.c
index d915c13fe548ecb22acbde702966c988ea25ed52..8ac1d807e3850ecf54f89c939518c89d11f1ac8a 100644 (file)
@@ -693,7 +693,9 @@ bool BPY_execute_string_as_intptr(
        return ok;
 }
 
-bool BPY_execute_string_ex(bContext *C, const char *expr, bool use_eval)
+bool BPY_execute_string_ex(
+        bContext *C, const char *imports[],
+        const char *expr, bool use_eval)
 {
        BLI_assert(expr);
        PyGILState_STATE gilstate;
@@ -715,13 +717,18 @@ bool BPY_execute_string_ex(bContext *C, const char *expr, bool use_eval)
        bmain_back = bpy_import_main_get();
        bpy_import_main_set(CTX_data_main(C));
 
-       retval = PyRun_String(expr, use_eval ? Py_eval_input : Py_file_input, py_dict, py_dict);
+       if (imports && (!PyC_NameSpace_ImportArray(py_dict, imports))) {
+               Py_DECREF(py_dict);
+               retval = NULL;
+       }
+       else {
+               retval = PyRun_String(expr, use_eval ? Py_eval_input : Py_file_input, py_dict, py_dict);
+       }
 
        bpy_import_main_set(bmain_back);
 
        if (retval == NULL) {
                ok = false;
-
                BPy_errors_to_report(CTX_wm_reports(C));
        }
        else {
@@ -735,9 +742,11 @@ bool BPY_execute_string_ex(bContext *C, const char *expr, bool use_eval)
        return ok;
 }
 
-bool BPY_execute_string(bContext *C, const char *expr)
+bool BPY_execute_string(
+        bContext *C, const char *imports[],
+        const char *expr)
 {
-       return BPY_execute_string_ex(C, expr, true);
+       return BPY_execute_string_ex(C, imports, expr, true);
 }
 
 void BPY_modules_load_user(bContext *C)