Save .isDefault from EnumerationMember issue #2896

1 files changed, 3 insertions, 2 deletions

diff --git a/python/types.py b/python/types.py
index fae9d611..03ee3733 100644
--- a/python/types.py
+++ b/python/types.py
@@ -1199,6 +1199,7 @@ class StructureBuilder(TypeBuilder):
 class EnumerationMember:
 	name: str
 	value: Optional[int] = None
+	isDefault: Optional[bool] = None
 
 	def __repr__(self):
 		value = f"{self.value:#x}" if self.value is not None else "auto()"
@@ -1258,7 +1259,7 @@ class EnumerationBuilder(TypeBuilder):
 		try:
 			for i in range(count.value):
 				result.append(
-				    EnumerationMember(members[i].name, members[i].value if not members[i].isDefault else None)
+				    EnumerationMember(members[i].name, members[i].value if not members[i].isDefault else None, members[i].isDefault)
 				)
 			return result
 		finally:
@@ -2099,7 +2100,7 @@ class EnumerationType(IntegerType):
 		assert members is not None, "core.BNGetEnumerationMembers returned None"
 		result = []
 		for i in range(0, count.value):
-			result.append(EnumerationMember(members[i].name, members[i].value))
+			result.append(EnumerationMember(members[i].name, members[i].value, members[i].isDefault))
 		core.BNFreeEnumerationMemberList(members, count.value)
 		return result