python - pymongo中游标的使用

Question

问题：
我需要多次使用同一个游标。然后发现一个for循环这个游标就没有用了。
然后我使用了

a = db.base.find()
c = b = a 

但是a for循环之后b,c都不能用了。
之后又想到用深度拷贝：

import copy
a = db.base.find()
b = copy.deepcopy(a)
c = copy.deepcopy(a)

这样就可以使用。
但是这样会不会增加内存的使用～！

请问怎么使用才是最优美的呢？谢谢

Answer 1

可以使用itertools的tee

In [20]: from itertools import tee

In [21]: x1,x2 = tee(db.x.find())

In [22]: list(x1)
Out[22]: 
[{u'_id': ObjectId('590026b521d7dd4a1beb3c1a'), u'name': u'bar'},
 {u'_id': ObjectId('590026b921d7dd4a1beb3c1b'), u'name': u'foo'}]

In [23]: list(x2)
Out[23]: 
[{u'_id': ObjectId('590026b521d7dd4a1beb3c1a'), u'name': u'bar'},
 {u'_id': ObjectId('590026b921d7dd4a1beb3c1b'), u'name': u'foo'}]