diff -r c02818d7e29f -r 7e61baa047f0 controls/CustomTree.py --- 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):