如何删除所有topLevel QTreeWidgetItems 和secondLevel QTreeWidgetItems 没有孩子？

如何解决如何删除所有topLevel QTreeWidgetItems 和secondLevel QTreeWidgetItems 没有孩子？

我在 Python 3.9.6 上使用 PyQt6。

我有一个三级QTreeWidget，意思是所有顶级项都有子项，顶级项的子项也有子项，顶级项的孙子项没有子项。

它的结构是这样的：

Tree                level 0
    Artist          level 1
        Album       level 2
            Song    level 3

我有一些删除歌曲节点的代码，也就是三级节点，只有三级节点会被直接删除。

很明显，当删除专辑中的所有歌曲时，专辑将是空的，我想删除所有空的专辑，这与链中的另一个操作相关联：当删除某个艺术家的所有专辑时，艺术家也应该被删除。

所以这就变成了：删除所有没有子级的二级项，然后删除所有没有子级的一级项。

这是我尝试过的，显然它不起作用也不应该起作用：

count = tree.topLevelItemCount()
for i in range(count):
    node = tree.topLevelItem(i)
    num = node.childCount()
    for j in range(num):
        child = node.child(j)
        if child.childCount() == 0:
            sip.delete(child)
    if node.childCount() == 0:
        sip.delete(node)

因为当项目被删除时，索引会发生变化，所以简单的 for 循环不应该工作。

而且我不能使用 while 循环，因为我不想清空所有内容，所以我不知道如何在这里使用 while 循环而不让它永远运行。

那么如何做到这一点？

解决方法

我已经想通了，我只需要在删除之前使用列表理解获取所有顶级项，然后使用第一个循环的结果使用另一个列表理解来获取第二级项。

然后直接遍历第二个推导的结果，不用index，因为for循环是单向的，保证当前项之后的项都存在于树中，直接删除而不是index就可以了。

然后循环第一次推导的结果，直接去掉项。

代码：

count = tree.topLevelItemCount()
nodes = [tree.topLevelItem(i) for i in range(count)]
children = [node.child(j) for node in nodes for j in range(node.childCount())]
for child in children:
    if child.childCount() == 0:
        sip.delete(child)
for node in nodes:
    if node.childCount() == 0:
        sip.delete(node)