wxPython樹控件

1、樹控件

樹（tree）是一種通過層次結構展示信息的控件，如下圖所示是樹控件示例，左窗口中是樹控件，在wxPython中樹控件類是wx.TreeCtrl。

 

wx.TreeCtrl中一個常用的方法有：

• AddRoot(text, image=-1, selImage=-1, data=None)。添加根節點，text參數根節點顯示的文本；image參數是該節點未被選中時的圖片索引，wx.TreeCtrl中使用的圖片被放到wx.ImageList圖像列表中；selImage參數是該節點被選中時的圖片索引。data參數是給節點傳遞的數據。方法返回節點，節點類型是wx.TreeItemId。
• AppendItem(parent, text, image=-1, selImage=-1, data=None)。添加子節點，parent 參數是父節點，其他參數同AddRoot()方法。方法返回值wx.TreeItemId。
• SelectItem(item, select=True)。選中item節點。
• Expand(item)。展開item節點。
• ExpandAll()。展開根節點下的所有子節點。
• ExpandAllChildren(item)。展開item節點下的所有子節點。
• AssignImageList(imageList)。將wx.ImageList圖像列表保存到樹中，這樣就可以在AddRoot()和AppendItem()方法中使用圖像列表索引了。

2、代碼實現

下面用代碼實現上圖所示的效果：

import wx
 
 
# 自定義窗口類MyFrame
class MyFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title="Tree", size=(500, 400))
        self.Center()
        swindow = wx.SplitterWindow(parent=self, id=-1)
        left = wx.Panel(parent=swindow)
        right = wx.Panel(parent=swindow)
        # 設置左右布局的分割窗口left和right
        swindow.SplitVertically(left, right, 200)
        # 設置最小窗格大小，左右布局指左邊窗口大小
        swindow.SetMinimumPaneSize(80)
        # 創建一棵樹
 
        self.tree = self.CreateTreeCtrl(left)
        self.Bind(wx.EVT_TREE_SEL_CHANGING, self.on_click, self.tree)
        # 為left面板設置一個布局管理器
        vbox1 = wx.BoxSizer(wx.VERTICAL)
        left.SetSizer(vbox1)
        vbox1.Add(self.tree, 1, flag=wx.EXPAND | wx.ALL, border=5)
        # 為right面板設置一個布局管理器
        vbox2 = wx.BoxSizer(wx.VERTICAL)
        right.SetSizer((vbox2))
        self.st = wx.StaticText(right, 2, label='右側面板')
        vbox2.Add(self.st, 1, flag=wx.EXPAND | wx.ALL, border=5)
 
    def on_click(self, event):
        item = event.GetItem()
        self.st.SetLabel(self.tree.GetItemText(item))
 
    def CreateTreeCtrl(self, parent):
        tree = wx.TreeCtrl(parent)
        # 通過wx.ImageList()創建一個圖像列表imglist並保存在樹中
        imglist = wx.ImageList(16, 16, True, 2)
        imglist.Add(wx.ArtProvider.GetBitmap(wx.ART_FOLDER, size=wx.Size(16, 16)))
        imglist.Add(wx.ArtProvider.GetBitmap(wx.ART_NORMAL_FILE, size=(16, 16)))
        tree.AssignImageList(imglist)
        # 創建根節點和5個子節點並展開
        root = tree.AddRoot('TreeRoot', image=0)
        item1 = tree.AppendItem(root, 'Item1', 0)
        item2 = tree.AppendItem(root, 'Item2', 0)
        item3 = tree.AppendItem(root, 'Item3', 0)
        item4 = tree.AppendItem(root, 'Item4', 0)
        item5 = tree.AppendItem(root, 'Item5', 0)
        tree.Expand(root)
        tree.SelectItem(root)
 
        # 給item1節點添加5個子節點並展開
        tree.AppendItem(item1, 'file 1', 1)
        tree.AppendItem(item1, 'file 2', 1)
        tree.AppendItem(item1, 'file 3', 1)
        tree.AppendItem(item1, 'file 4', 1)
        tree.AppendItem(item1, 'file 5', 1)
        tree.Expand(item1)
 
        # 給item2節點添加5個子節點並展開
        tree.AppendItem(item2, 'file 1', 1)
        tree.AppendItem(item2, 'file 2', 1)
        tree.AppendItem(item2, 'file 3', 1)
        tree.AppendItem(item2, 'file 4', 1)
        tree.AppendItem(item2, 'file 5', 1)
        tree.Expand(item2)
 
        # 返回樹對象
        return tree
 
 
class App(wx.App):
    def OnInit(self):
        # 創建窗口對象
        frame = MyFrame()
        frame.Show()
        return True
 
    def OnExit(self):
        print("應用程序退出")
        return 0
 
 
if __name__ == '__main__':
    app = App()
    app.MainLoop()