wxPython/samples/wxPIA_book/Chapter-15/tree_virtual.py

   1 import wx
   2 import data
   3
   4 class TestFrame(wx.Frame):
   5     def __init__(self):
   6         wx.Frame.__init__(self, None, title="virtual tree with icons", size=(400,500))
   7         il = wx.ImageList(16,16)
   8         self.fldridx = il.Add(
   9             wx.ArtProvider.GetBitmap(wx.ART_FOLDER, wx.ART_OTHER, (16,16)))
  10         self.fldropenidx = il.Add(
  11             wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN,   wx.ART_OTHER, (16,16)))
  12         self.fileidx = il.Add(
  13             wx.ArtProvider.GetBitmap(wx.ART_NORMAL_FILE, wx.ART_OTHER, (16,16)))
  14         self.tree = wx.TreeCtrl(self)
  15         self.tree.AssignImageList(il)
  16         root = self.tree.AddRoot("wx.Object")
  17         self.tree.SetItemImage(root, self.fldridx,
  18                                wx.TreeItemIcon_Normal)
  19         self.tree.SetItemImage(root, self.fldropenidx,
  20                                wx.TreeItemIcon_Expanded)
  21
  22         # Instead of adding nodes for the whole tree, just attach some
  23         # data to the root node so that it can find and add its child
  24         # nodes when it is expanded, and mark it as having children so
  25         # it will be expandable.
  26         self.tree.SetItemPyData(root, data.tree)
  27         self.tree.SetItemHasChildren(root, True)
  28
  29         # Bind some interesting events
  30         self.Bind(wx.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded, self.tree)
  31         self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed, self.tree)
  32         self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged, self.tree)
  33         self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivated, self.tree)
  34
  35         self.Bind(wx.EVT_TREE_ITEM_EXPANDING, self.OnItemExpanding, self.tree)
  36         self.tree.Expand(root)
  37
  38
  39     def AddTreeNodes(self, parentItem):
  40         """
  41         Add nodes for just the children of the parentItem
  42         """
  43         items = self.tree.GetItemPyData(parentItem)
  44         for item in items:
  45             if type(item) == str:
  46                 # a leaf node
  47                 newItem = self.tree.AppendItem(parentItem, item)
  48                 self.tree.SetItemImage(newItem, self.fileidx,
  49                                        wx.TreeItemIcon_Normal)
  50             else:
  51                 # this item has children
  52                 newItem = self.tree.AppendItem(parentItem, item[0])
  53                 self.tree.SetItemImage(newItem, self.fldridx,
  54                                        wx.TreeItemIcon_Normal)
  55                 self.tree.SetItemImage(newItem, self.fldropenidx,
  56                                        wx.TreeItemIcon_Expanded)
  57                 self.tree.SetItemPyData(newItem, item[1])
  58                 self.tree.SetItemHasChildren(newItem, True)
  59
  60
  61
  62     def GetItemText(self, item):
  63         if item:
  64             return self.tree.GetItemText(item)
  65         else:
  66             return ""
  67
  68     def OnItemExpanded(self, evt):
  69         print "OnItemExpanded: ", self.GetItemText(evt.GetItem())
  70
  71     def OnItemExpanding(self, evt):
  72         # When the item is about to be expanded add the first level of child nodes
  73         print "OnItemExpanding:", self.GetItemText(evt.GetItem())
  74         self.AddTreeNodes(evt.GetItem())
  75
  76     def OnItemCollapsed(self, evt):
  77         print "OnItemCollapsed:", self.GetItemText(evt.GetItem())
  78         # And remove them when collapsed as we don't need them any longer
  79         self.tree.DeleteChildren(evt.GetItem())
  80
  81     def OnSelChanged(self, evt):
  82         print "OnSelChanged:   ", self.GetItemText(evt.GetItem())
  83
  84     def OnActivated(self, evt):
  85         print "OnActivated:    ", self.GetItemText(evt.GetItem())
  86
  87
  88 app = wx.PySimpleApp(redirect=True)
  89 frame = TestFrame()
  90 frame.Show()
  91 app.MainLoop()
  92