pymongo query by datetime

Question

У меня была одна проблема, из-за которой у меня была головная боль почти на один день.

скрипты Python являются:

# coding:utf-8 
import pymongo 
import datetime, time 

def query_data(IP,datefrom,dateto): 
    conn = pymongo.MongoClient(IP) 
    db = conn.mymongo 

    startdate = time.mktime(time.strptime(datefrom,'%Y-%m-%d')) 
    enddate = time.mktime(time.strptime(dateto,'%Y-%m-%d')) 

    # all above are correct definitely. 
    # 
    # but I got no result from this line below. 
    cursor = db.find({'username':'test','created':{'$gte':startdate,'$lt':enddate}}) 
    print cursor.count() 
    # with above code, I got 0 returned though I can see it is nonzero in database. 
    # 
    # If I use this below with single datetime, I can see result. But it is not my intention. 
    # cursor = db.find({'username':'test','created':datefrom}) 
    # print cursor.count() 

    tempData = [] 
    for doc in cursor: 
     print doc['twitter'] 
     tempData.append(doc['twitter']) 

    print len(tempData) 
    return tempData 

result = query_data('192.168.100.20','2014-4-1','2014-5-1') 

Вопрос заключается в том, что случилось с моим кодом выше? Или как я могу запрашивать данные из mongoDB между двумя датами, с скриптом pymongo?

Answer 1

Есть 2 проблемы здесь, вы пытаетесь вызвать find на Database объекта, также необходимо запросить с datetime.datetime объекта для $gte и $lt для правильной работы в MongoDB.

def query_data(IP,datefrom,dateto): 
    conn = pymongo.MongoClient(IP) 
    db = conn.mymongo 

    # time.mktime will only return a float point number 
    # what you really need are the datetime.datetime objects 
    startdate = datetime.datetime.strptime(datefrom,'%Y-%m-%d') 
    enddate = datetime.datetime.strptime(dateto,'%Y-%m-%d') 

    # you need to call find method on collection, not database object 
    cursor = db['your_collection'].find({'username':'test','created':{'$gte':startdate,'$lt':enddate}}) 
    print cursor.count() 
    ... 

Я не тестировал его, но он должен работать так, как ожидалось, и надеюсь, что это поможет.