#encoding:utf-8

import wx

import wx.lib.agw.customtreectrl as CT

class MyFrame(wx.Frame):

    def __init__(self, parent):

        self.checked_items = []

        wx.Frame.__init__(self, parent, -1, "customtreectrl")

        self.custom_tree = CT.CustomTreeCtrl(self, agwStyle=wx.TR_DEFAULT_STYLE)

        self.root = self.custom_tree.AddRoot("root",  ct_type=1)

        for y in range(5):

            item = self.custom_tree.AppendItem(self.root, "wangjian", ct_type=1)

        self.custom_tree.ExpandAll()

        self.Bind(CT.EVT_TREE_ITEM_CHECKED, self.checked_item)

    def checked_item(self, event):

        # 只要树控件中的任意一个复选框状态有变化就会响应这个函数

        if (event.GetItem() == self.root):

            if self.custom_tree.IsItemChecked(event.GetItem()):

                self.custom_tree.CheckChilds(self.root)

                for item in self.get_childs(self.root):

                    self.checked_items.append(self.custom_tree.GetItemText(item))

            else:

                for item in self.get_childs(self.root):

                    self.custom_tree.CheckItem(item, False)

                    self.checked_items.remove(self.custom_tree.GetItemText(item))

        # else:

        #     if self.custom_tree.IsItemChecked(event.GetItem()):

        #         self.checked_items.append(self.custom_tree.GetItemText(event.GetItem()))

        #         print "add"

        #     else:

        #         self.checked_items.remove(self.custom_tree.GetItemText(event.GetItem()))

        #         print "remove"

        print self.checked_items

    def get_childs(self, item_obj):

        item_list = []

        (item, cookie) = self.custom_tree.GetFirstChild(item_obj)

        while item:

            item_list.append(item)

            print "ok"

            (item, cookie) = self.custom_tree.GetNextChild(item_obj, cookie)

        return item_list

app = wx.App()

frame = MyFrame(None)

frame.Show()

app.MainLoop()