From 9177e19efae191df8df7d9a1c07b093c4e96a960 Mon Sep 17 00:00:00 2001
From: Andrej730 <azhilenkov@gmail.com>
Date: Fri, 4 Oct 2024 16:14:15 +0500
Subject: [PATCH] Reset proposed property fields after adding custom property
 #5539

---
 src/bonsai/bonsai/core/pset.py | 1 +
 src/bonsai/bonsai/tool/pset.py | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/src/bonsai/bonsai/core/pset.py b/src/bonsai/bonsai/core/pset.py
index c96e5177ff..5a059f4833 100644
--- a/src/bonsai/bonsai/core/pset.py
+++ b/src/bonsai/bonsai/core/pset.py
@@ -94,6 +94,7 @@ def add_proposed_prop(
 ) -> Union[None, str]:
     props = pset.get_pset_props(obj_name, obj_type)
     res = pset.add_proposed_property(name, pset.cast_string_to_primitive(value), props)
+    pset.reset_proposed_property_fields(props)
     return res
 
 
diff --git a/src/bonsai/bonsai/tool/pset.py b/src/bonsai/bonsai/tool/pset.py
index dcfa111872..c547330f3c 100644
--- a/src/bonsai/bonsai/tool/pset.py
+++ b/src/bonsai/bonsai/tool/pset.py
@@ -327,3 +327,10 @@ class Pset(bonsai.core.tool.Pset):
                 return value
             except:
                 return value
+
+    @classmethod
+    def reset_proposed_property_fields(cls, props: bpy.types.PropertyGroup) -> None:
+        reset_props = ("prop_name", "prop_value")
+        bl_rna_props = props.bl_rna.properties
+        for prop_name in reset_props:
+            setattr(props, prop_name, bl_rna_props[prop_name].default)