python - web.py sql查询，为什么只能遍历第一遍的结果？

Question

这是我的代码:

import web
user_db = web.database(dbn='mysql', ....)

info_list = user_db.query("select * from tablename where t_id= ")
for info in info_list:
# work ok at first time, print the correct id
    print info.id

for info in info_list: 
# Code can't reach here 
    print info.id

第二次好像不行。为什么？

Answer 1

根据source code ，底层的 query() 调用返回一个 iterator，它在第一个循环后就耗尽了。

如果你需要多次迭代它，把它转换成一个列表:

info_list = list(user_db.query("select * from tablename where t_id= "))

或者，您可以使用 itertools.tee()创建新的迭代器:

info_list1, info_list2 = itertools.tee(info_list)