Added the sample code from wxPython In Action to the samples dir

[wxWidgets.git] / wxPython / samples / wxPIA_book / Chapter-15 / tree_misc.py

diff --git a/wxPython/samples/wxPIA_book/Chapter-15/tree_misc.py b/wxPython/samples/wxPIA_book/Chapter-15/tree_misc.py
new file mode 100644 (file)
index 0000000..2df8e00
--- /dev/null
+++ b/wxPython/samples/wxPIA_book/Chapter-15/tree_misc.py
@@ -0,0 +1,155 @@
+import wx
+import data
+
+class TestFrame(wx.Frame):
+    def __init__(self):
+        wx.Frame.__init__(self, None, title="tree: misc tests", size=(400,500))
+
+        # Create an image list
+        il = wx.ImageList(16,16)
+
+        # Get some standard images from the art provider and add them
+        # to the image list
+        self.fldridx = il.Add(
+            wx.ArtProvider.GetBitmap(wx.ART_FOLDER, wx.ART_OTHER, (16,16)))
+        self.fldropenidx = il.Add(
+            wx.ArtProvider.GetBitmap(wx.ART_FILE_OPEN,   wx.ART_OTHER, (16,16)))
+        self.fileidx = il.Add(
+            wx.ArtProvider.GetBitmap(wx.ART_NORMAL_FILE, wx.ART_OTHER, (16,16)))
+        
+
+        # Create the tree
+        self.tree = wx.TreeCtrl(self, style=wx.TR_DEFAULT_STYLE | wx.TR_EDIT_LABELS)
+
+        # Give it the image list
+        self.tree.AssignImageList(il)
+
+        
+        # Add a root node and assign it some images
+        root = self.tree.AddRoot("wx.Object")
+        self.tree.SetItemPyData(root, None)
+        self.tree.SetItemImage(root, self.fldridx,
+                               wx.TreeItemIcon_Normal)
+        self.tree.SetItemImage(root, self.fldropenidx,
+                               wx.TreeItemIcon_Expanded)
+        
+        # Add nodes from our data set
+        self.AddTreeNodes(root, data.tree)
+
+        # Bind some interesting events
+        self.Bind(wx.EVT_TREE_ITEM_EXPANDED, self.OnItemExpanded, self.tree)
+        self.Bind(wx.EVT_TREE_ITEM_COLLAPSED, self.OnItemCollapsed, self.tree)
+        self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged, self.tree)
+        self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivated, self.tree)
+
+        self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.tree)
+        
+
+        # Expand the first level
+        self.tree.Expand(root)
+
+
+        menu = wx.Menu()
+        mi = menu.Append(-1, "Edit Item")
+        self.Bind(wx.EVT_MENU, self.OnEditItem, mi)
+        mi = menu.Append(-1, "Sort Children")
+        self.Bind(wx.EVT_MENU, self.OnSortChildren, mi)
+        mi = menu.Append(-1, "Delete Children")
+        self.Bind(wx.EVT_MENU, self.OnDeleteChildren, mi)
+        mi = menu.Append(-1, "Find all Children")
+        self.Bind(wx.EVT_MENU, self.OnFindChildren, mi)
+        mb = wx.MenuBar()
+        mb.Append(menu, "Tests")
+        self.SetMenuBar(mb)
+
+
+    def AddTreeNodes(self, parentItem, items):
+        """
+        Recursively traverses the data structure, adding tree nodes to
+        match it.
+        """
+        for item in items:
+            if type(item) == str:
+                newItem = self.tree.AppendItem(parentItem, item)
+                self.tree.SetItemPyData(newItem, None)
+                self.tree.SetItemImage(newItem, self.fileidx,
+                                       wx.TreeItemIcon_Normal)
+            else:
+                newItem = self.tree.AppendItem(parentItem, item[0])
+                self.tree.SetItemPyData(newItem, None)
+                self.tree.SetItemImage(newItem, self.fldridx,
+                                       wx.TreeItemIcon_Normal)
+                self.tree.SetItemImage(newItem, self.fldropenidx,
+                                       wx.TreeItemIcon_Expanded)
+   
+                self.AddTreeNodes(newItem, item[1])
+                
+
+    def GetItemText(self, item):
+        if item:
+            return self.tree.GetItemText(item)
+        else:
+            return ""
+        
+    def OnItemExpanded(self, evt):
+        print "OnItemExpanded: ", self.GetItemText(evt.GetItem())
+        
+    def OnItemCollapsed(self, evt):
+        print "OnItemCollapsed:", self.GetItemText(evt.GetItem())
+
+    def OnSelChanged(self, evt):
+        print "OnSelChanged:   ", self.GetItemText(evt.GetItem())
+
+    def OnActivated(self, evt):
+        print "OnActivated:    ", self.GetItemText(evt.GetItem())
+
+
+    def OnBeginEdit(self, evt):
+        print "OnBeginEdit:    ", self.GetItemText(evt.GetItem())
+        # we can prevent nodes from being edited, for example let's
+        # not let the root node be edited...
+        item = evt.GetItem()
+        if item == self.tree.GetRootItem():
+            evt.Veto()
+            print "*** Edit was vetoed!"
+            
+        
+    def OnEditItem(self, evt):
+        item = self.tree.GetSelection()
+        if item:
+            self.tree.EditLabel(item)
+            
+    def OnSortChildren(self, evt):
+        item = self.tree.GetSelection()
+        if item:
+            self.tree.SortChildren(item)
+
+    def OnDeleteChildren(self, evt):
+        item = self.tree.GetSelection()
+        if item:
+            self.tree.DeleteChildren(item)
+
+    def OnFindChildren(self, evt):
+        item = self.tree.GetSelection()
+        if item:
+            self.PrintAllItems(item)
+
+    def PrintAllItems(self, parent, indent=0):
+        text = self.tree.GetItemText(parent)
+        print ' ' * indent, text
+        indent += 4
+        item, cookie = self.tree.GetFirstChild(parent)
+        while item:
+            if self.tree.ItemHasChildren(item):
+                self.PrintAllItems(item, indent)
+            else:
+                text = self.tree.GetItemText(item)
+                print ' ' * indent, text
+            item, cookie = self.tree.GetNextChild(parent, cookie)
+
+
+app = wx.PySimpleApp(redirect=True)
+frame = TestFrame()
+frame.Show()
+app.MainLoop()
+