python - penggunaan kursor dalam pymongo

Question

Masalah:
Saya perlu menggunakan kursor yang sama beberapa kali. Kemudian saya mendapati bahawa kursor dalam gelung for tidak berguna.
Kemudian saya gunakan

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

Tetapi selepas gelung a for, b dan c tidak boleh digunakan.
Kemudian saya terfikir untuk menggunakan salinan dalam:

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

Dengan cara ini anda boleh menggunakannya.
Tetapi adakah ini akan meningkatkan penggunaan memori~!

Bagaimana cara menggunakannya dengan cara yang paling cantik? Terima kasih

Answer 1

Anda boleh menggunakan tee itertools

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'}]