PyMongo Cursor Iteration

Question

I'm looking to create and handle a cursor in python the way cursors natively work in mongo. I know the intended way is to do 'result = collection.find()' and do a 'for record in result' but I'm looking to wrap iteration functionality in a class. I'd like to be able to create a new class object and call a function e.g. init_cursor() to make a db connection and do a find returning a cursor. I would then like the have a get_next() function that would move to the next result and set class data members based on the result. Here's the pesudo-code: ``` class dataIter(): def __init__(self): self.collection = pymongo.Connection().db.collection self.cursor = self.collection.find({}) #return all self.age = None self.gender = None def get_next(self): if self.cursor.hasNext(): data = self.cursor.next() self.set_data(data) def set_data(self, data): self.age = data['age'] self.gender = data['gender'] ``` This way I would be able to simply call: ``` obj.get_next() age = obj.age gender = obj.gender ``…

Accepted Answer

I don't understand how what you are showing is any more convenient that just doing:

```
col = pymongo.Connection().db.collection
cur = col.find({})

obj = next(cur, None)
if obj:
    age = obj['age']
    gender = obj['gender']
```

Its not clear how this wrapper is helpful. Also, if what you are really after is an ORM, then don't reinvent the wheel when this exists: http://mongoengine.org/

PyMongo Cursor Iteration

Solution

Problem