--- a/controls/CustomTree.py Mon Aug 14 22:30:41 2017 +0300
+++ b/controls/CustomTree.py Mon Aug 14 23:27:15 2017 +0300
@@ -30,6 +30,7 @@
# Customize CustomTreeItem for adding icon on item left
CT.GenericTreeItem._ExtraImage = None
+
def SetExtraImage(self, image):
self._type = (1 if image is not None else 0)
self._ExtraImage = image
@@ -37,12 +38,15 @@
CT.GenericTreeItem.SetExtraImage = SetExtraImage
_DefaultGetCurrentCheckedImage = CT.GenericTreeItem.GetCurrentCheckedImage
+
+
def GetCurrentCheckedImage(self):
if self._ExtraImage is not None:
return self._ExtraImage
return _DefaultGetCurrentCheckedImage(self)
CT.GenericTreeItem.GetCurrentCheckedImage = GetCurrentCheckedImage
+
class CustomTree(CT.CustomTreeCtrl):
def __init__(self, *args, **kwargs):