PyAPI: add imports arg to BPY_execute_string
authorCampbell Barton <ideasman42@gmail.com>
Fri, 9 Nov 2018 23:55:04 +0000 (10:55 +1100)
committerCampbell Barton <ideasman42@gmail.com>
Fri, 9 Nov 2018 23:55:04 +0000 (10:55 +1100)
Allows for avoiding `__import__` in expressions,
was already supported for BPY_execute_string_as_* API calls.

source/blender/editors/space_script/script_edit.c
source/blender/freestyle/intern/system/PythonInterpreter.h
source/blender/python/BPY_extern.h
source/blender/python/intern/bpy_interface.c
source/blender/windowmanager/intern/wm_files.c
source/creator/creator_args.c

index 6bfb51d07c62dd9a397d5b28014554571febe5da..5a761d1cabf466f556f21ff4d57e14c18c7ac6c6 100644 (file)
@@ -125,7 +125,7 @@ static int script_reload_exec(bContext *C, wmOperator *op)
        /* TODO, this crashes on netrender and keying sets, need to look into why
         * disable for now unless running in debug mode */
        WM_cursor_wait(1);
-       BPY_execute_string(C, "__import__('bpy').utils.load_scripts(reload_scripts=True)");
+       BPY_execute_string(C, (const char *[]){"bpy", NULL}, "bpy.utils.load_scripts(reload_scripts=True)");
        WM_cursor_wait(0);
        WM_event_add_notifier(C, NC_WINDOW, NULL);
        return OPERATOR_FINISHED;
index 4f5e94ef7a0db13a93af7a831128f117188b729a..cb49e7718a0239a3f6dcbac877db43d19bee1597 100644 (file)
@@ -112,7 +112,7 @@ public:
 
                BKE_reports_clear(reports);
 
-               if (!BPY_execute_string(_context, str.c_str())) {
+               if (!BPY_execute_string(_context, NULL, str.c_str())) {
                        BPy_errors_to_report(reports);
                        cerr << "\nError executing Python script from PythonInterpreter::interpretString" << endl;
                        cerr << "Name: " << name << endl;
index 5f43f0bf8857ee8f79f99aa3c1a5ed1d9c7479f3..955b1c4e4ede5dc0961fcfe1091c39fad0233b3a 100644 (file)
@@ -80,8 +80,8 @@ bool  BPY_execute_string_as_number(struct bContext *C, const char *imports[], con
 bool   BPY_execute_string_as_intptr(struct bContext *C, const char *imports[], const char *expr, const bool verbose, intptr_t *r_value);
 bool   BPY_execute_string_as_string(struct bContext *C, const char *imports[], const char *expr, const bool verbose, char **r_value);
 
-bool   BPY_execute_string_ex(struct bContext *C, const char *expr, bool use_eval);
-bool   BPY_execute_string(struct bContext *C, const char *expr);
+bool   BPY_execute_string_ex(struct bContext *C, const char *imports[], const char *expr, bool use_eval);
+bool   BPY_execute_string(struct bContext *C, const char *imports[], const char *expr);
 
 void   BPY_text_free_code(struct Text *text);
 void   BPY_modules_update(struct bContext *C); // XXX - annoying, need this for pointers that get out of date
index 123c938b921c8425e77aa7d4d19fbc2a28dca864..0c7f3a7f68d267c1c00569085d39105ea5e4378f 100644 (file)
@@ -692,7 +692,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;
@@ -714,13 +716,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 {
@@ -734,9 +741,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)
index 87ca7310971a3748f5026421099bcfc53d7de48c..ef951010ba587a2865870039388e102a53e627d7 100644 (file)
@@ -473,10 +473,14 @@ static void wm_file_read_post(bContext *C, const bool is_startup_file, const boo
                        if (use_userdef) {
                                /* Only run when we have a template path found. */
                                if (BKE_appdir_app_template_any()) {
-                                       BPY_execute_string(C, "__import__('bl_app_template_utils').reset()");
+                                       BPY_execute_string(
+                                               C, (const char *[]){"bl_app_template_utils", NULL},
+                                               "bl_app_template_utils.reset()");
                                }
                                /* sync addons, these may have changed from the defaults */
-                               BPY_execute_string(C, "__import__('addon_utils').reset_all()");
+                               BPY_execute_string(
+                                       C, (const char *[]){"addon_utils", NULL},
+                                       "addon_utils.reset_all()");
                        }
                        BPY_python_reset(C);
                        addons_loaded = true;
index f8fe57b05a22a52dfdaba196b529ffedaaf9d159..74bcc129f20888f4c374c3a2e7e6ca99d0505aff 100644 (file)
@@ -1815,7 +1815,7 @@ static int arg_handle_python_expr_run(int argc, const char **argv, void *data)
        /* workaround for scripts not getting a bpy.context.scene, causes internal errors elsewhere */
        if (argc > 1) {
                bool ok;
-               BPY_CTX_SETUP(ok = BPY_execute_string_ex(C, argv[1], false));
+               BPY_CTX_SETUP(ok = BPY_execute_string_ex(C, NULL, argv[1], false));
                if (!ok && app_state.exit_code_on_error.python) {
                        printf("\nError: script failed, expr: '%s', exiting.\n", argv[1]);
                        exit(app_state.exit_code_on_error.python);
@@ -1841,7 +1841,7 @@ static int arg_handle_python_console_run(int UNUSED(argc), const char **argv, vo
 #ifdef WITH_PYTHON
        bContext *C = data;
 
-       BPY_CTX_SETUP(BPY_execute_string(C, "__import__('code').interact()"));
+       BPY_CTX_SETUP(BPY_execute_string(C, (const char *[]){"code", NULL}, "code.interact()"));
 
        return 0;
 #else
@@ -1897,7 +1897,7 @@ static int arg_handle_addons_set(int argc, const char **argv, void *data)
                BLI_snprintf(str, slen, script_str, argv[1]);
 
                BLI_assert(strlen(str) + 1 == slen);
-               BPY_CTX_SETUP(BPY_execute_string_ex(C, str, false));
+               BPY_CTX_SETUP(BPY_execute_string_ex(C, NULL, str, false));
                free(str);
 #else
                UNUSED_VARS(argv, data);