Wxpython:TreeCtrl:在树上迭代

时间:2012-04-18 07:55:31

标签: python wxpython

我使用以下方法迭代wxpython treectrl的所有节点。

 def get_desired_parent(self, name, selectednode = None):
    if selectednode == None:
        selectednode = self.treeCtrl.RootItem
    # First perform the action on the first object separately
    childcount = self.treeCtrl.GetChildrenCount(selectednode, False)
    if childcount == 0:
        return None

    (item,cookie) = self.treeCtrl.GetFirstChild(selectednode)
    if self.treeCtrl.GetItemText(item) == name:
        return item

    while childcount > 1:
        childcount = childcount - 1
     # Then iterate over the rest of objects
        (item,cookie) = self.treeCtrl.GetNextChild(item,cookie)
        if self.treeCtrl.GetItemText(item) == name:
            return item
    return None

当我在递归内部迭代迭代时,这个代码过多的问题变得更加明显。 是否有另一种方式以更紧凑的方式执行相同的操作,以使我的代码更简洁/ pythonic。

2 个答案:

答案 0 :(得分:2)

您可以使用此函数(仅在其命名空间中)检查它是否与conditiin匹配。如果它确实返回该项目,则继续。

否则,您可以在while行之后检查您的情况。这样,item变量将由循环前的第一个子节点定义,并像其他任何一样进行评估。

还有另一种方式:(或两者混合)

(child, cookie) = self.GetFirstChild(item)
while child.IsOk():
    do_something(child)
     (child, cookie) = self.GetNextChild(item, cookie)

答案 1 :(得分:0)

这里有一个完整的示例,该示例首先遍历树的行进深度。该功能已绑定到右键。

def OnRightDown(self, event):
    def showChildren(item,cookie):
        # functions goes recursively down the tree
        if item.IsOk():
            child, cookie = self.tree.GetFirstChild(item)
            while child.IsOk():
                child, cookie = self.tree.GetNextChild(child, cookie)
                if child:
                  print(self.tree.GetItemText(child)) #show child label name
                  showChildren(child,cookie)
    pt = event.GetPosition()
    item, flags = self.tree.HitTest(pt)
    if item:
        print(self.tree.GetItemText(item)) #show parent label name
        showChildren(item,0)  #iterate depth first into the tree