]> git.saurik.com Git - wxWidgets.git/blame - wxPython/samples/wxPIA_book/Chapter-15/tree_simple.py
added missing button state
[wxWidgets.git] / wxPython / samples / wxPIA_book / Chapter-15 / tree_simple.py
CommitLineData
be05b434
RD
1import wx
2import data
3
4class TestFrame(wx.Frame):
5 def __init__(self):
6 wx.Frame.__init__(self, None, title="simple tree", size=(400,500))
7
8 # Create the tree
9 self.tree = wx.TreeCtrl(self)
10
11 # Add a root node
12 root = self.tree.AddRoot("wx.Object")
13
14 # Add nodes from our data set
15 self.AddTreeNodes(root, data.tree)
16
17 # Bind some interesting events
18 self.Bind(wx.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded, self.tree)
19 self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed, self.tree)
20 self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged, self.tree)
21 self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivated, self.tree)
22
23 # Expand the first level
24 self.tree.Expand(root)
25
26
27 def AddTreeNodes(self, parentItem, items):
28 """
29 Recursively traverses the data structure, adding tree nodes to
30 match it.
31 """
32 for item in items:
33 if type(item) == str:
34 self.tree.AppendItem(parentItem, item)
35 else:
36 newItem = self.tree.AppendItem(parentItem, item[0])
37 self.AddTreeNodes(newItem, item[1])
38
39 def GetItemText(self, item):
40 if item:
41 return self.tree.GetItemText(item)
42 else:
43 return ""
44
45 def OnItemExpanded(self, evt):
46 print "OnItemExpanded: ", self.GetItemText(evt.GetItem())
47
48 def OnItemCollapsed(self, evt):
49 print "OnItemCollapsed:", self.GetItemText(evt.GetItem())
50
51 def OnSelChanged(self, evt):
52 print "OnSelChanged: ", self.GetItemText(evt.GetItem())
53
54 def OnActivated(self, evt):
55 print "OnActivated: ", self.GetItemText(evt.GetItem())
56
57
58app = wx.PySimpleApp(redirect=True)
59frame = TestFrame()
60frame.Show()
61app.MainLoop()
62