Search code examples
pythonmongodbdictionarymongodb-querypymongo

How to convert a pymongo.cursor.Cursor into a dict?


I am using pymongo to query for all items in a region (actually it is to query for all venues in a region on a map). I used db.command(SON()) before to search in a spherical region, which can return me a dictionary and in the dictionary there is a key called results which contains the venues. Now I need to search in a square area and I am suggested to use db.places.find, however, this returns me a pymongo.cursor.Cursor class and I have no idea how to extract the venue results from it.

Does anyone know whether I should convert the cursor into a dict and extract the results out, or use another method to query for items in a square region? BTW, db is pymongo.database.Database class

The code is:

>>> import pymongo
>>> db = pymongo.MongoClient(host).PSRC 
>>> resp = db.places.find({"loc": {"$within": {"$box": [[ll_lng,ll_lat], [ur_lng,ur_lat]]}}})
>>> for doc in resp:
>>>     print(doc)

I have values of ll_lng, ll_lat, ur_lng and ur_lat, use these values but it prints nothing from this codes


Solution

  • The find method returns a Cursor instance, which allows you to iterate over all matching documents.

    To get the first document that matches the given criteria, you need to use find_one. The result of find_one is a dictionary.

    You can always use the list constructor to return a list of all the documents in the collection but bear in mind that this will load all the data in memory and may not be what you want.

    You should do that if you need to reuse the cursor and have a good reason not to use rewind()


    Demo using find:

    >>> import pymongo
    >>> conn = pymongo.MongoClient()
    >>> db = conn.test #test is my database
    >>> col = db.spam #Here spam is my collection
    >>> cur = col.find()  
    >>> cur
    <pymongo.cursor.Cursor object at 0xb6d447ec>
    >>> for doc in cur:
    ...     print(doc)  # or do something with the document
    ... 
    {'a': 1, '_id': ObjectId('54ff30faadd8f30feb90268f'), 'b': 2}
    {'a': 1, 'c': 3, '_id': ObjectId('54ff32a2add8f30feb902690'), 'b': 2}
    

    Demo using find_one:

    >>> col.find_one()
    {'a': 1, '_id': ObjectId('54ff30faadd8f30feb90268f'), 'b': 2}