Fix T58964: drivers_remove fails w/ missing paths

This commit is contained in:
Campbell Barton 2019-01-07 15:27:59 +11:00
parent abe32d2a35
commit 07287ceda1
Notes: blender-bot 2023-02-14 08:39:23 +01:00
Referenced by issue #58964, Python: "drivers_remove" not working and crashing with invalid drivers
1 changed files with 116 additions and 30 deletions

View File

@ -33,6 +33,7 @@
#include "BLI_utildefines.h"
#include "BLI_string.h"
#include "BLI_string_utils.h"
#include "DNA_scene_types.h"
#include "DNA_anim_types.h"
@ -61,9 +62,9 @@
#include "../generic/python_utildefines.h"
/* for keyframes and drivers */
static int pyrna_struct_anim_args_parse(
static int pyrna_struct_anim_args_parse_ex(
PointerRNA *ptr, const char *error_prefix, const char *path,
const char **path_full, int *index)
const char **path_full, int *index, bool *r_path_no_validate)
{
const bool is_idbase = RNA_struct_is_ID(ptr->type);
PropertyRNA *prop;
@ -101,38 +102,47 @@ static int pyrna_struct_anim_args_parse(
}
if (prop == NULL) {
if (r_path_no_validate) {
*r_path_no_validate = true;
return -1;
}
PyErr_Format(PyExc_TypeError,
"%.200s property \"%s\" not found",
error_prefix, path);
return -1;
}
if (!RNA_property_animateable(&r_ptr, prop)) {
PyErr_Format(PyExc_TypeError,
"%.200s property \"%s\" not animatable",
error_prefix, path);
return -1;
}
if (RNA_property_array_check(prop) == 0) {
if ((*index) == -1) {
*index = 0;
}
else {
PyErr_Format(PyExc_TypeError,
"%.200s index %d was given while property \"%s\" is not an array",
error_prefix, *index, path);
return -1;
}
if (r_path_no_validate) {
/* Don't touch the index. */
}
else {
int array_len = RNA_property_array_length(&r_ptr, prop);
if ((*index) < -1 || (*index) >= array_len) {
if (!RNA_property_animateable(&r_ptr, prop)) {
PyErr_Format(PyExc_TypeError,
"%.200s index out of range \"%s\", given %d, array length is %d",
error_prefix, path, *index, array_len);
"%.200s property \"%s\" not animatable",
error_prefix, path);
return -1;
}
if (RNA_property_array_check(prop) == 0) {
if ((*index) == -1) {
*index = 0;
}
else {
PyErr_Format(PyExc_TypeError,
"%.200s index %d was given while property \"%s\" is not an array",
error_prefix, *index, path);
return -1;
}
}
else {
int array_len = RNA_property_array_length(&r_ptr, prop);
if ((*index) < -1 || (*index) >= array_len) {
PyErr_Format(PyExc_TypeError,
"%.200s index out of range \"%s\", given %d, array length is %d",
error_prefix, path, *index, array_len);
return -1;
}
}
}
if (is_idbase) {
@ -152,6 +162,68 @@ static int pyrna_struct_anim_args_parse(
return 0;
}
static int pyrna_struct_anim_args_parse(
PointerRNA *ptr, const char *error_prefix, const char *path,
const char **path_full, int *index)
{
return pyrna_struct_anim_args_parse_ex(ptr, error_prefix, path, path_full, index, NULL);
}
/**
* Unlike #pyrna_struct_anim_args_parse \a path_full may be copied from \a path.
*/
static int pyrna_struct_anim_args_parse_no_resolve(
PointerRNA *ptr, const char *error_prefix, const char *path,
const char **path_full)
{
const bool is_idbase = RNA_struct_is_ID(ptr->type);
if (is_idbase) {
*path_full = path;
return 0;
}
else {
char *path_prefix = RNA_path_from_ID_to_struct(ptr);
if (path_prefix == NULL) {
PyErr_Format(PyExc_TypeError,
"%.200s could not make path for type %s",
error_prefix, RNA_struct_identifier(ptr->type));
return -1;
}
if (*path == '[') {
*path_full = BLI_string_joinN(path_prefix, path);
}
else {
*path_full = BLI_string_join_by_sep_charN('.', path_prefix, path);
}
MEM_freeN(path_prefix);
}
return 0;
}
static int pyrna_struct_anim_args_parse_no_resolve_fallback(
PointerRNA *ptr, const char *error_prefix, const char *path,
const char **path_full, int *index)
{
bool path_unresolved = false;
if (pyrna_struct_anim_args_parse_ex(
ptr, error_prefix, path,
path_full, index, &path_unresolved) == -1)
{
if (path_unresolved == true) {
if (pyrna_struct_anim_args_parse_no_resolve(
ptr, error_prefix, path, path_full) == -1)
{
return -1;
}
}
else {
return -1;
}
}
return 0;
}
/* internal use for insert and delete */
static int pyrna_struct_keyframe_parse(
PointerRNA *ptr, PyObject *args, PyObject *kw, const char *parse_str, const char *error_prefix,
@ -162,14 +234,19 @@ static int pyrna_struct_keyframe_parse(
const char *path;
/* note, parse_str MUST start with 's|ifsO!' */
if (!PyArg_ParseTupleAndKeywords(args, kw, parse_str, (char **)kwlist, &path, index, cfra, group_name,
&PySet_Type, &pyoptions))
if (!PyArg_ParseTupleAndKeywords(
args, kw, parse_str, (char **)kwlist, &path, index, cfra, group_name,
&PySet_Type, &pyoptions))
{
return -1;
}
if (pyrna_struct_anim_args_parse(ptr, error_prefix, path, path_full, index) == -1)
if (pyrna_struct_anim_args_parse(
ptr, error_prefix, path,
path_full, index) == -1)
{
return -1;
}
if (*cfra == FLT_MAX)
*cfra = CTX_data_scene(BPy_GetContext())->r.cfra;
@ -419,7 +496,10 @@ PyObject *pyrna_struct_driver_add(BPy_StructRNA *self, PyObject *args)
if (!PyArg_ParseTuple(args, "s|i:driver_add", &path, &index))
return NULL;
if (pyrna_struct_anim_args_parse(&self->ptr, "bpy_struct.driver_add():", path, &path_full, &index) == -1) {
if (pyrna_struct_anim_args_parse(
&self->ptr, "bpy_struct.driver_add():", path,
&path_full, &index) == -1)
{
return NULL;
}
else {
@ -490,10 +570,14 @@ PyObject *pyrna_struct_driver_remove(BPy_StructRNA *self, PyObject *args)
PYRNA_STRUCT_CHECK_OBJ(self);
if (!PyArg_ParseTuple(args, "s|i:driver_remove", &path, &index))
if (!PyArg_ParseTuple(args, "s|i:driver_remove", &path, &index)) {
return NULL;
}
if (pyrna_struct_anim_args_parse(&self->ptr, "bpy_struct.driver_remove():", path, &path_full, &index) == -1) {
if (pyrna_struct_anim_args_parse_no_resolve_fallback(
&self->ptr, "bpy_struct.driver_remove():", path,
&path_full, &index) == -1)
{
return NULL;
}
else {
@ -504,7 +588,9 @@ PyObject *pyrna_struct_driver_remove(BPy_StructRNA *self, PyObject *args)
result = ANIM_remove_driver(&reports, (ID *)self->ptr.id.data, path_full, index, 0);
MEM_freeN((void *)path_full);
if (path != path_full) {
MEM_freeN((void *)path_full);
}
if (BPy_reports_to_error(&reports, PyExc_RuntimeError, true) == -1)
return NULL;