PyAPI: support retrieving the exception when running a string
- Optionally get the error as a single line. - Support access the error as an allocated string. - PyC_ExceptionBuffer_Simple was always printing to the `stdout` while PyC_ExceptionBuffer didn't, now either print to the output. Without this, callers are unable to do anything with the error string.
This commit is contained in:
		| @@ -107,8 +107,9 @@ bool user_string_to_number(bContext *C, | |||||||
|                            const char *str, |                            const char *str, | ||||||
|                            const struct UnitSettings *unit, |                            const struct UnitSettings *unit, | ||||||
|                            int type, |                            int type, | ||||||
|                            const char *error_prefix, |                            double *r_value, | ||||||
|                            double *r_value); |                            const bool use_single_line_error, | ||||||
|  |                            char **r_error); | ||||||
|  |  | ||||||
| /** \} */ | /** \} */ | ||||||
|  |  | ||||||
|   | |||||||
| @@ -52,6 +52,7 @@ | |||||||
| #include "BKE_context.h" | #include "BKE_context.h" | ||||||
| #include "BKE_idprop.h" | #include "BKE_idprop.h" | ||||||
| #include "BKE_main.h" | #include "BKE_main.h" | ||||||
|  | #include "BKE_report.h" | ||||||
| #include "BKE_scene.h" | #include "BKE_scene.h" | ||||||
| #include "BKE_screen.h" | #include "BKE_screen.h" | ||||||
| #include "BKE_unit.h" | #include "BKE_unit.h" | ||||||
| @@ -2912,7 +2913,14 @@ char *ui_but_string_get_dynamic(uiBut *but, int *r_str_size) | |||||||
| static bool ui_number_from_string_units( | static bool ui_number_from_string_units( | ||||||
|     bContext *C, const char *str, const int unit_type, const UnitSettings *unit, double *r_value) |     bContext *C, const char *str, const int unit_type, const UnitSettings *unit, double *r_value) | ||||||
| { | { | ||||||
|   return user_string_to_number(C, str, unit, unit_type, UI_NUMBER_EVAL_ERROR_PREFIX, r_value); |   char *error = NULL; | ||||||
|  |   const bool ok = user_string_to_number(C, str, unit, unit_type, r_value, true, &error); | ||||||
|  |   if (error) { | ||||||
|  |     ReportList *reports = CTX_wm_reports(C); | ||||||
|  |     BKE_reportf(reports, RPT_ERROR, "%s: %s", UI_NUMBER_EVAL_ERROR_PREFIX, error); | ||||||
|  |     MEM_freeN(error); | ||||||
|  |   } | ||||||
|  |   return ok; | ||||||
| } | } | ||||||
|  |  | ||||||
| static bool ui_number_from_string_units_with_but(bContext *C, | static bool ui_number_from_string_units_with_but(bContext *C, | ||||||
| @@ -2929,7 +2937,11 @@ static bool ui_number_from_string(bContext *C, const char *str, double *r_value) | |||||||
| { | { | ||||||
|   bool ok; |   bool ok; | ||||||
| #ifdef WITH_PYTHON | #ifdef WITH_PYTHON | ||||||
|   ok = BPY_run_string_as_number(C, NULL, str, UI_NUMBER_EVAL_ERROR_PREFIX, r_value); |   struct BPy_RunErrInfo err_info = { | ||||||
|  |       .reports = CTX_wm_reports(C), | ||||||
|  |       .report_prefix = UI_NUMBER_EVAL_ERROR_PREFIX, | ||||||
|  |   }; | ||||||
|  |   ok = BPY_run_string_as_number(C, NULL, str, &err_info, r_value); | ||||||
| #else | #else | ||||||
|   UNUSED_VARS(C); |   UNUSED_VARS(C); | ||||||
|   *r_value = atof(str); |   *r_value = atof(str); | ||||||
|   | |||||||
| @@ -399,7 +399,7 @@ static void ui_but_user_menu_add(bContext *C, uiBut *but, bUserMenu *um) | |||||||
|                    "'%s').label", |                    "'%s').label", | ||||||
|                    idname); |                    idname); | ||||||
|           char *expr_result = NULL; |           char *expr_result = NULL; | ||||||
|           if (BPY_run_string_as_string(C, expr_imports, expr, __func__, &expr_result)) { |           if (BPY_run_string_as_string(C, expr_imports, expr, NULL, &expr_result)) { | ||||||
|             STRNCPY(drawstr, expr_result); |             STRNCPY(drawstr, expr_result); | ||||||
|             MEM_freeN(expr_result); |             MEM_freeN(expr_result); | ||||||
|           } |           } | ||||||
|   | |||||||
| @@ -428,7 +428,7 @@ static uiTooltipData *ui_tooltip_data_from_tool(bContext *C, uiBut *but, bool is | |||||||
|     if (has_valid_context == false) { |     if (has_valid_context == false) { | ||||||
|       expr_result = BLI_strdup(has_valid_context_error); |       expr_result = BLI_strdup(has_valid_context_error); | ||||||
|     } |     } | ||||||
|     else if (BPY_run_string_as_string(C, expr_imports, expr, __func__, &expr_result)) { |     else if (BPY_run_string_as_string(C, expr_imports, expr, NULL, &expr_result)) { | ||||||
|       if (STREQ(expr_result, "")) { |       if (STREQ(expr_result, "")) { | ||||||
|         MEM_freeN(expr_result); |         MEM_freeN(expr_result); | ||||||
|         expr_result = NULL; |         expr_result = NULL; | ||||||
| @@ -485,7 +485,7 @@ static uiTooltipData *ui_tooltip_data_from_tool(bContext *C, uiBut *but, bool is | |||||||
|     if (has_valid_context == false) { |     if (has_valid_context == false) { | ||||||
|       expr_result = BLI_strdup(has_valid_context_error); |       expr_result = BLI_strdup(has_valid_context_error); | ||||||
|     } |     } | ||||||
|     else if (BPY_run_string_as_string(C, expr_imports, expr, __func__, &expr_result)) { |     else if (BPY_run_string_as_string(C, expr_imports, expr, NULL, &expr_result)) { | ||||||
|       if (STREQ(expr_result, ".")) { |       if (STREQ(expr_result, ".")) { | ||||||
|         MEM_freeN(expr_result); |         MEM_freeN(expr_result); | ||||||
|         expr_result = NULL; |         expr_result = NULL; | ||||||
| @@ -589,7 +589,7 @@ static uiTooltipData *ui_tooltip_data_from_tool(bContext *C, uiBut *but, bool is | |||||||
|         if (has_valid_context == false) { |         if (has_valid_context == false) { | ||||||
|           shortcut = BLI_strdup(has_valid_context_error); |           shortcut = BLI_strdup(has_valid_context_error); | ||||||
|         } |         } | ||||||
|         else if (BPY_run_string_as_intptr(C, expr_imports, expr, __func__, &expr_result)) { |         else if (BPY_run_string_as_intptr(C, expr_imports, expr, NULL, &expr_result)) { | ||||||
|           if (expr_result != 0) { |           if (expr_result != 0) { | ||||||
|             wmKeyMap *keymap = (wmKeyMap *)expr_result; |             wmKeyMap *keymap = (wmKeyMap *)expr_result; | ||||||
|             LISTBASE_FOREACH (wmKeyMapItem *, kmi, &keymap->items) { |             LISTBASE_FOREACH (wmKeyMapItem *, kmi, &keymap->items) { | ||||||
| @@ -654,7 +654,7 @@ static uiTooltipData *ui_tooltip_data_from_tool(bContext *C, uiBut *but, bool is | |||||||
|         /* pass */ |         /* pass */ | ||||||
|       } |       } | ||||||
|       else if (BPY_run_string_as_string_and_size( |       else if (BPY_run_string_as_string_and_size( | ||||||
|                    C, expr_imports, expr, __func__, &expr_result, &expr_result_len)) { |                    C, expr_imports, expr, NULL, &expr_result, &expr_result_len)) { | ||||||
|         /* pass. */ |         /* pass. */ | ||||||
|       } |       } | ||||||
|     } |     } | ||||||
| @@ -731,7 +731,7 @@ static uiTooltipData *ui_tooltip_data_from_tool(bContext *C, uiBut *but, bool is | |||||||
|     if (has_valid_context == false) { |     if (has_valid_context == false) { | ||||||
|       /* pass */ |       /* pass */ | ||||||
|     } |     } | ||||||
|     else if (BPY_run_string_as_intptr(C, expr_imports, expr, __func__, &expr_result)) { |     else if (BPY_run_string_as_intptr(C, expr_imports, expr, NULL, &expr_result)) { | ||||||
|       if (expr_result != 0) { |       if (expr_result != 0) { | ||||||
|         { |         { | ||||||
|           uiTooltipField *field = text_field_add(data, |           uiTooltipField *field = text_field_add(data, | ||||||
|   | |||||||
| @@ -29,6 +29,7 @@ | |||||||
| #include "BLT_translation.h" | #include "BLT_translation.h" | ||||||
|  |  | ||||||
| #include "BKE_context.h" | #include "BKE_context.h" | ||||||
|  | #include "BKE_report.h" | ||||||
| #include "BKE_scene.h" | #include "BKE_scene.h" | ||||||
| #include "BKE_unit.h" | #include "BKE_unit.h" | ||||||
|  |  | ||||||
| @@ -283,10 +284,15 @@ bool user_string_to_number(bContext *C, | |||||||
|                            const char *str, |                            const char *str, | ||||||
|                            const UnitSettings *unit, |                            const UnitSettings *unit, | ||||||
|                            int type, |                            int type, | ||||||
|                            const char *error_prefix, |                            double *r_value, | ||||||
|                            double *r_value) |                            const bool use_single_line_error, | ||||||
|  |                            char **r_error) | ||||||
| { | { | ||||||
| #ifdef WITH_PYTHON | #ifdef WITH_PYTHON | ||||||
|  |   struct BPy_RunErrInfo err_info = { | ||||||
|  |       .use_single_line_error = use_single_line_error, | ||||||
|  |       .r_string = r_error, | ||||||
|  |   }; | ||||||
|   double unit_scale = BKE_scene_unit_scale(unit, type, 1.0); |   double unit_scale = BKE_scene_unit_scale(unit, type, 1.0); | ||||||
|   if (BKE_unit_string_contains_unit(str, type)) { |   if (BKE_unit_string_contains_unit(str, type)) { | ||||||
|     char str_unit_convert[256]; |     char str_unit_convert[256]; | ||||||
| @@ -294,10 +300,10 @@ bool user_string_to_number(bContext *C, | |||||||
|     BKE_unit_replace_string( |     BKE_unit_replace_string( | ||||||
|         str_unit_convert, sizeof(str_unit_convert), str, unit_scale, unit->system, type); |         str_unit_convert, sizeof(str_unit_convert), str, unit_scale, unit->system, type); | ||||||
|  |  | ||||||
|     return BPY_run_string_as_number(C, NULL, str_unit_convert, error_prefix, r_value); |     return BPY_run_string_as_number(C, NULL, str_unit_convert, &err_info, r_value); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   int success = BPY_run_string_as_number(C, NULL, str, error_prefix, r_value); |   int success = BPY_run_string_as_number(C, NULL, str, &err_info, r_value); | ||||||
|   *r_value = BKE_unit_apply_preferred_unit(unit, type, *r_value); |   *r_value = BKE_unit_apply_preferred_unit(unit, type, *r_value); | ||||||
|   *r_value /= unit_scale; |   *r_value /= unit_scale; | ||||||
|   return success; |   return success; | ||||||
| @@ -577,10 +583,19 @@ bool handleNumInput(bContext *C, NumInput *n, const wmEvent *event) | |||||||
|   if (n->str[0]) { |   if (n->str[0]) { | ||||||
|     const float val_prev = n->val[idx]; |     const float val_prev = n->val[idx]; | ||||||
|     Scene *sce = CTX_data_scene(C); |     Scene *sce = CTX_data_scene(C); | ||||||
|  |     char *error = NULL; | ||||||
|  |  | ||||||
|     double val; |     double val; | ||||||
|     int success = user_string_to_number( |     int success = user_string_to_number( | ||||||
|         C, n->str, &sce->unit, n->unit_type[idx], IFACE_("Numeric input evaluation"), &val); |         C, n->str, &sce->unit, n->unit_type[idx], &val, false, &error); | ||||||
|  |  | ||||||
|  |     if (error) { | ||||||
|  |       ReportList *reports = CTX_wm_reports(C); | ||||||
|  |       printf("%s\n", error); | ||||||
|  |       BKE_report(reports, RPT_ERROR, error); | ||||||
|  |       BKE_report(reports, RPT_ERROR, IFACE_("Numeric input evaluation")); | ||||||
|  |       MEM_freeN(error); | ||||||
|  |     } | ||||||
|  |  | ||||||
|     if (success) { |     if (success) { | ||||||
|       n->val[idx] = (float)val; |       n->val[idx] = (float)val; | ||||||
|   | |||||||
| @@ -42,27 +42,43 @@ bool BPY_run_text(struct bContext *C, | |||||||
| bool BPY_run_string_exec(struct bContext *C, const char *imports[], const char *expr); | bool BPY_run_string_exec(struct bContext *C, const char *imports[], const char *expr); | ||||||
| bool BPY_run_string_eval(struct bContext *C, const char *imports[], const char *expr); | bool BPY_run_string_eval(struct bContext *C, const char *imports[], const char *expr); | ||||||
|  |  | ||||||
|  | /** | ||||||
|  |  * \note When this struct is passed in as NULL, | ||||||
|  |  * print errors to the `stdout` and clear. | ||||||
|  |  */ | ||||||
|  | struct BPy_RunErrInfo { | ||||||
|  |   /** Brief text, single line (can show this in status bar for e.g.). */ | ||||||
|  |   bool use_single_line_error; | ||||||
|  |  | ||||||
|  |   /** Report with optional prefix (when non-NULL). */ | ||||||
|  |   struct ReportList *reports; | ||||||
|  |   const char *report_prefix; | ||||||
|  |  | ||||||
|  |   /** Allocated exception text (assign when non-NULL). */ | ||||||
|  |   char **r_string; | ||||||
|  | }; | ||||||
|  |  | ||||||
| /* Run, evaluating to fixed type result. */ | /* Run, evaluating to fixed type result. */ | ||||||
| bool BPY_run_string_as_number(struct bContext *C, | bool BPY_run_string_as_number(struct bContext *C, | ||||||
|                               const char *imports[], |                               const char *imports[], | ||||||
|                               const char *expr, |                               const char *expr, | ||||||
|                               const char *report_prefix, |                               struct BPy_RunErrInfo *err_info, | ||||||
|                               double *r_value); |                               double *r_value); | ||||||
| bool BPY_run_string_as_intptr(struct bContext *C, | bool BPY_run_string_as_intptr(struct bContext *C, | ||||||
|                               const char *imports[], |                               const char *imports[], | ||||||
|                               const char *expr, |                               const char *expr, | ||||||
|                               const char *report_prefix, |                               struct BPy_RunErrInfo *err_info, | ||||||
|                               intptr_t *r_value); |                               intptr_t *r_value); | ||||||
| bool BPY_run_string_as_string_and_size(struct bContext *C, | bool BPY_run_string_as_string_and_size(struct bContext *C, | ||||||
|                                        const char *imports[], |                                        const char *imports[], | ||||||
|                                        const char *expr, |                                        const char *expr, | ||||||
|                                        const char *report_prefix, |                                        struct BPy_RunErrInfo *err_info, | ||||||
|                                        char **r_value, |                                        char **r_value, | ||||||
|                                        size_t *r_value_size); |                                        size_t *r_value_size); | ||||||
| bool BPY_run_string_as_string(struct bContext *C, | bool BPY_run_string_as_string(struct bContext *C, | ||||||
|                               const char *imports[], |                               const char *imports[], | ||||||
|                               const char *expr, |                               const char *expr, | ||||||
|                               const char *report_prefix, |                               struct BPy_RunErrInfo *err_info, | ||||||
|                               char **r_value); |                               char **r_value); | ||||||
|  |  | ||||||
| #ifdef __cplusplus | #ifdef __cplusplus | ||||||
|   | |||||||
| @@ -733,7 +733,6 @@ PyObject *PyC_ExceptionBuffer_Simple(void) | |||||||
|  |  | ||||||
|   PyErr_Restore(error_type, error_value, error_traceback); |   PyErr_Restore(error_type, error_value, error_traceback); | ||||||
|  |  | ||||||
|   PyErr_Print(); |  | ||||||
|   PyErr_Clear(); |   PyErr_Clear(); | ||||||
|   return string_io_buf; |   return string_io_buf; | ||||||
| } | } | ||||||
|   | |||||||
| @@ -31,6 +31,7 @@ | |||||||
|  |  | ||||||
| #include "BKE_context.h" | #include "BKE_context.h" | ||||||
| #include "BKE_main.h" | #include "BKE_main.h" | ||||||
|  | #include "BKE_report.h" | ||||||
| #include "BKE_text.h" | #include "BKE_text.h" | ||||||
|  |  | ||||||
| #include "DNA_text_types.h" | #include "DNA_text_types.h" | ||||||
| @@ -294,13 +295,47 @@ bool BPY_run_string_exec(bContext *C, const char *imports[], const char *expr) | |||||||
|  * in code that doesn't deal with Python data-types. |  * in code that doesn't deal with Python data-types. | ||||||
|  * \{ */ |  * \{ */ | ||||||
|  |  | ||||||
|  | static void run_string_handle_error(struct BPy_RunErrInfo *err_info) | ||||||
|  | { | ||||||
|  |   if (err_info == NULL) { | ||||||
|  |     PyErr_Print(); | ||||||
|  |     PyErr_Clear(); | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   /* Signal to do nothing. */ | ||||||
|  |   if (!(err_info->reports || err_info->r_string)) { | ||||||
|  |     PyErr_Clear(); | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   PyObject *py_err_str = err_info->use_single_line_error ? PyC_ExceptionBuffer_Simple() : | ||||||
|  |                                                            PyC_ExceptionBuffer(); | ||||||
|  |   const char *err_str = py_err_str ? PyUnicode_AsUTF8(py_err_str) : "Unable to extract exception"; | ||||||
|  |  | ||||||
|  |   if (err_info->reports != NULL) { | ||||||
|  |     if (err_info->report_prefix) { | ||||||
|  |       BKE_reportf(err_info->reports, RPT_ERROR, "%s: %s", err_info->report_prefix, err_str); | ||||||
|  |     } | ||||||
|  |     else { | ||||||
|  |       BKE_report(err_info->reports, RPT_ERROR, err_str); | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   if (err_info->r_string != NULL) { | ||||||
|  |     *err_info->r_string = BLI_strdup(err_str); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   Py_XDECREF(py_err_str); | ||||||
|  | } | ||||||
|  |  | ||||||
| /** | /** | ||||||
|  * \return success |  * \return success | ||||||
|  */ |  */ | ||||||
| bool BPY_run_string_as_number(bContext *C, | bool BPY_run_string_as_number(bContext *C, | ||||||
|                               const char *imports[], |                               const char *imports[], | ||||||
|                               const char *expr, |                               const char *expr, | ||||||
|                               const char *report_prefix, |                               struct BPy_RunErrInfo *err_info, | ||||||
|                               double *r_value) |                               double *r_value) | ||||||
| { | { | ||||||
|   PyGILState_STATE gilstate; |   PyGILState_STATE gilstate; | ||||||
| @@ -320,12 +355,7 @@ bool BPY_run_string_as_number(bContext *C, | |||||||
|   ok = PyC_RunString_AsNumber(imports, expr, "<expr as number>", r_value); |   ok = PyC_RunString_AsNumber(imports, expr, "<expr as number>", r_value); | ||||||
|  |  | ||||||
|   if (ok == false) { |   if (ok == false) { | ||||||
|     if (report_prefix != NULL) { |     run_string_handle_error(err_info); | ||||||
|       BPy_errors_to_report_ex(CTX_wm_reports(C), report_prefix, false, false); |  | ||||||
|     } |  | ||||||
|     else { |  | ||||||
|       PyErr_Clear(); |  | ||||||
|     } |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   bpy_context_clear(C, &gilstate); |   bpy_context_clear(C, &gilstate); | ||||||
| @@ -339,7 +369,7 @@ bool BPY_run_string_as_number(bContext *C, | |||||||
| bool BPY_run_string_as_string_and_size(bContext *C, | bool BPY_run_string_as_string_and_size(bContext *C, | ||||||
|                                        const char *imports[], |                                        const char *imports[], | ||||||
|                                        const char *expr, |                                        const char *expr, | ||||||
|                                        const char *report_prefix, |                                        struct BPy_RunErrInfo *err_info, | ||||||
|                                        char **r_value, |                                        char **r_value, | ||||||
|                                        size_t *r_value_size) |                                        size_t *r_value_size) | ||||||
| { | { | ||||||
| @@ -357,12 +387,7 @@ bool BPY_run_string_as_string_and_size(bContext *C, | |||||||
|   ok = PyC_RunString_AsStringAndSize(imports, expr, "<expr as str>", r_value, r_value_size); |   ok = PyC_RunString_AsStringAndSize(imports, expr, "<expr as str>", r_value, r_value_size); | ||||||
|  |  | ||||||
|   if (ok == false) { |   if (ok == false) { | ||||||
|     if (report_prefix != NULL) { |     run_string_handle_error(err_info); | ||||||
|       BPy_errors_to_report_ex(CTX_wm_reports(C), false, false, report_prefix); |  | ||||||
|     } |  | ||||||
|     else { |  | ||||||
|       PyErr_Clear(); |  | ||||||
|     } |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   bpy_context_clear(C, &gilstate); |   bpy_context_clear(C, &gilstate); | ||||||
| @@ -373,12 +398,11 @@ bool BPY_run_string_as_string_and_size(bContext *C, | |||||||
| bool BPY_run_string_as_string(bContext *C, | bool BPY_run_string_as_string(bContext *C, | ||||||
|                               const char *imports[], |                               const char *imports[], | ||||||
|                               const char *expr, |                               const char *expr, | ||||||
|                               const char *report_prefix, |                               struct BPy_RunErrInfo *err_info, | ||||||
|                               char **r_value) |                               char **r_value) | ||||||
| { | { | ||||||
|   size_t value_dummy_size; |   size_t value_dummy_size; | ||||||
|   return BPY_run_string_as_string_and_size( |   return BPY_run_string_as_string_and_size(C, imports, expr, err_info, r_value, &value_dummy_size); | ||||||
|       C, imports, expr, report_prefix, r_value, &value_dummy_size); |  | ||||||
| } | } | ||||||
|  |  | ||||||
| /** | /** | ||||||
| @@ -389,7 +413,7 @@ bool BPY_run_string_as_string(bContext *C, | |||||||
| bool BPY_run_string_as_intptr(bContext *C, | bool BPY_run_string_as_intptr(bContext *C, | ||||||
|                               const char *imports[], |                               const char *imports[], | ||||||
|                               const char *expr, |                               const char *expr, | ||||||
|                               const char *report_prefix, |                               struct BPy_RunErrInfo *err_info, | ||||||
|                               intptr_t *r_value) |                               intptr_t *r_value) | ||||||
| { | { | ||||||
|   BLI_assert(r_value && expr); |   BLI_assert(r_value && expr); | ||||||
| @@ -406,12 +430,7 @@ bool BPY_run_string_as_intptr(bContext *C, | |||||||
|   ok = PyC_RunString_AsIntPtr(imports, expr, "<expr as intptr>", r_value); |   ok = PyC_RunString_AsIntPtr(imports, expr, "<expr as intptr>", r_value); | ||||||
|  |  | ||||||
|   if (ok == false) { |   if (ok == false) { | ||||||
|     if (report_prefix != NULL) { |     run_string_handle_error(err_info); | ||||||
|       BPy_errors_to_report_ex(CTX_wm_reports(C), report_prefix, false, false); |  | ||||||
|     } |  | ||||||
|     else { |  | ||||||
|       PyErr_Clear(); |  | ||||||
|     } |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   bpy_context_clear(C, &gilstate); |   bpy_context_clear(C, &gilstate); | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user