yield的运用——QTreeWidgetItem的迭代

yield的应用——QTreeWidgetItem的迭代
最近学python 用 pyside，所以做个笔记

在Qt里，要是要遍历QTreeWidget所有节点，其中一个方法就是用QTreeWidgetItemIterator，但有个烦人的地方，这个东西在C++里还行，但在python里就太不python了。

it = QTreeWidgetItemIterator(tree)while it.value():    v = it.value()    do_something()    it += 1

天呀，这么多行代码，还必须用while和it += 1。
传统简化方法，在js里，最典型的简化方法就是用模板。

tree_item_loop = function(tree, fn) {    it = QTreeWidgetItemIterator(tree)    while (it.value()) {        fn(it.value());        it += 1    }}tree_item_loop(tree, function(item){    // })

在python里是这样

def tree_item_loop(tree, fn):    it = QTreeWidgetItemIterator(tree)    while it.value():        s_item = it.value()        fn(s_item)        it += 1def fn(item):    passtree_item_loop(tree, fn)

但是在又有一个问题，如果我需要知道当前节点在遍历时的index咋办？在循环里计算index也就算了，但更要命的是，python和js不一样，js里，fn这个参数，其本身的参数可以任意，甚至可以不声明，也就是说，定义成

function some(item, index) {} 

或者

function(item) {} 

都是可以的，而python不行，必须要声明具体参数，即使是可变参数，也必须要声明，否则多传参数或少接收参数之类会报错。所以如果我要传index，所有传进来的fn就都必须是声明index或者 *args，这让调用者就郁闷了。

不过python既然这样无法解决，那么就用python的方法来解决。用yield。

def tree_items(tree):    it = QTreeWidgetItemIterator(tree)    while it.value():        yield it.value()        it += 1

要遍历

for item in tree_items(tree):    print item

好爽。再来要带index

for index, item in enumerate(tree_items(tree)):    print index, item

查看更多 下一篇