From a2b8dad469a0f7fa16d73cccfdbd37c9d5d5c10d Mon Sep 17 00:00:00 2001
From: Campbell Barton <ideasman42@gmail.com>
Date: Fri, 30 Jul 2021 16:03:47 +1000
Subject: PyAPI: support accessing the original value for RNA enum parsing

Needed in siturations when the input argument is needed for exception messages.
---
 source/blender/python/intern/bpy_rna.c | 3 +++
 source/blender/python/intern/bpy_rna.h | 7 +++++++
 2 files changed, 10 insertions(+)

(limited to 'source/blender/python/intern')

diff --git a/source/blender/python/intern/bpy_rna.c b/source/blender/python/intern/bpy_rna.c
index e45cd02a5bf..759630b73ac 100644
--- a/source/blender/python/intern/bpy_rna.c
+++ b/source/blender/python/intern/bpy_rna.c
@@ -832,6 +832,8 @@ int pyrna_enum_value_parse_string(PyObject *o, void *p)
           parse_data->items, identifier, &parse_data->value, "enum identifier") == -1) {
     return 0;
   }
+
+  parse_data->value_orig = o;
   parse_data->is_set = true;
   return 1;
 }
@@ -851,6 +853,7 @@ int pyrna_enum_bitfield_parse_set(PyObject *o, void *p)
           parse_data->items, o, &parse_data->value, "enum identifier set") == -1) {
     return 0;
   }
+  parse_data->value_orig = o;
   parse_data->is_set = true;
   return 1;
 }
diff --git a/source/blender/python/intern/bpy_rna.h b/source/blender/python/intern/bpy_rna.h
index 23a6b19c22c..24dbad53eb3 100644
--- a/source/blender/python/intern/bpy_rna.h
+++ b/source/blender/python/intern/bpy_rna.h
@@ -219,6 +219,13 @@ int pyrna_set_to_enum_bitfield(const struct EnumPropertyItem *items,
  */
 struct BPy_EnumProperty_Parse {
   const EnumPropertyItem *items;
+  /**
+   * Set when the value was successfully parsed.
+   * Useful if the input ever needs to be included in an error message.
+   * (if the value is not supported under certain conditions).
+   */
+  PyObject *value_orig;
+
   int value;
   bool is_set;
 };
-- 
cgit v1.2.3