28 from util.BitmapLibrary import GetBitmap

    28 from util.BitmapLibrary import GetBitmap

    29 

    29 

    30 # Customize CustomTreeItem for adding icon on item left

    30 # Customize CustomTreeItem for adding icon on item left

    31 CT.GenericTreeItem._ExtraImage = None

    31 CT.GenericTreeItem._ExtraImage = None

    32 

    32 

    33 def SetExtraImage(self, image):

    34 def SetExtraImage(self, image):

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

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

    35     self._ExtraImage = image

    36     self._ExtraImage = image

    36 

    37 

    37 CT.GenericTreeItem.SetExtraImage = SetExtraImage

    38 CT.GenericTreeItem.SetExtraImage = SetExtraImage

    38 

    39 

    39 _DefaultGetCurrentCheckedImage = CT.GenericTreeItem.GetCurrentCheckedImage

    40 _DefaultGetCurrentCheckedImage = CT.GenericTreeItem.GetCurrentCheckedImage

    40 def GetCurrentCheckedImage(self):

    43 def GetCurrentCheckedImage(self):

    41     if self._ExtraImage is not None:

    44     if self._ExtraImage is not None:

    42         return self._ExtraImage

    45         return self._ExtraImage

    43     return _DefaultGetCurrentCheckedImage(self)

    46     return _DefaultGetCurrentCheckedImage(self)

    44 CT.GenericTreeItem.GetCurrentCheckedImage = GetCurrentCheckedImage

    47 CT.GenericTreeItem.GetCurrentCheckedImage = GetCurrentCheckedImage

    45 

    49 

    46 class CustomTree(CT.CustomTreeCtrl):

    50 class CustomTree(CT.CustomTreeCtrl):

    47 

    51 

    48     def __init__(self, *args, **kwargs):

    52     def __init__(self, *args, **kwargs):

    49         CT.CustomTreeCtrl.__init__(self, *args, **kwargs)

    53         CT.CustomTreeCtrl.__init__(self, *args, **kwargs)