33 

    33 

    34 def SetExtraImage(self, image):

    34 def SetExtraImage(self, image):

    35     self._type = (1 if image is not None else 0)

    35     self._type = (1 if image is not None else 0)

    36     self._ExtraImage = image

    36     self._ExtraImage = image

    37 

    37 

    38 CT.GenericTreeItem.SetExtraImage = SetExtraImage

    39 CT.GenericTreeItem.SetExtraImage = SetExtraImage

    39 

    40 

    40 _DefaultGetCurrentCheckedImage = CT.GenericTreeItem.GetCurrentCheckedImage

    41 _DefaultGetCurrentCheckedImage = CT.GenericTreeItem.GetCurrentCheckedImage

    41 

    42 

    42 

    43 

    43 def GetCurrentCheckedImage(self):

    44 def GetCurrentCheckedImage(self):

    44     if self._ExtraImage is not None:

    45     if self._ExtraImage is not None:

    45         return self._ExtraImage

    46         return self._ExtraImage

    46     return _DefaultGetCurrentCheckedImage(self)

    47     return _DefaultGetCurrentCheckedImage(self)

    47 CT.GenericTreeItem.GetCurrentCheckedImage = GetCurrentCheckedImage

    50 CT.GenericTreeItem.GetCurrentCheckedImage = GetCurrentCheckedImage

    48 

    51 

    49 

    52 

    50 class CustomTree(CT.CustomTreeCtrl):

    53 class CustomTree(CT.CustomTreeCtrl):

    51 

    54