pymongo：名称“ISODate”未定义

Question

当我尝试使用 pymongo 选择 mongodb 中的数据时遇到问题，这是我的代码：

import pymongo
from pymongo import MongoClient
import sys
from datetime import datetime

try:
    conn=pymongo.MongoClient('10.33.109.228',27017)
    db=conn.mnemosyne
    data_ip=db.session.aggregate({'$match':{'timestamp':{'$gte': ISODate('2016-11-11T00:00:00.000Z'),'$lte': ISODate('2016-11-11T23:59:59.000Z')}}},{'$group':{'_id':'$source_ip'}})
    for f in data_ip:
        print f['_id']

except pymongo.errors.ConnectionFailure, e:
    print "Could not connect to MongoDB: %s" % e

当我执行它时，我遇到了这样的错误：

Traceback (most recent call last):
  File "test.py", line 9, in <module>
    data_ip=db.session.aggregate({'$match':{'timestamp':{'$gte': ISODate('2016-11-11T00:00:00.000Z'),'$lte': ISODate('2016-11-11T23:59:59.000Z')}}},{'$group':{'_id':'$source_ip'}})
NameError: name 'ISODate' is not defined

我想要这样的结果：

{ "_id" : "60.18.133.207" }
{ "_id" : "178.254.52.96" }
{ "_id" : "42.229.218.192" }
{ "_id" : "92.82.171.117" }
{ "_id" : "103.208.120.205" }
{ "_id" : "185.153.208.142" }

这是我的数据库的示例结构：

> db.session.findOne()

    {
            "_id" : ObjectId("5786398d1f50070f31f27f7c"),
            "protocol" : "epmapper",
            "hpfeed_id" : ObjectId("5786398d1f50070f31f27f7b"),
            "timestamp" : ISODate("2016-07-13T12:52:29.112Z"),
            "source_ip" : "23.251.55.182",
            "source_port" : 2713,
            "destination_port" : 135,
            "identifier" : "d3374f14-48f7-11e6-9e19-0050569163b4",
            "honeypot" : "dionaea"
    }

请帮我修正错误

Answer 1

ISODate

是Mongo shell中的一个函数，Mongo shell是一个javascript环境，它在Python中不可用。

您可以使用 dateutil 在 Python 中将字符串转换为日期时间对象，

import dateutil.parser
dateStr = "2016-11-11T00:00:00.000Z"
dateutil.parser.parse(dateStr)  # returns a datetime.datetime(2016, 11, 11, 00, 0, tzinfo=tzutc())

使用

PyMongo

，如果您想在

MongoDB

中插入日期时间，您可以简单地执行以下操作：

import pymongo
import dateutil
dateStr = '2016-11-11T00:00:00.000Z'
myDatetime = dateutil.parser.parse(dateStr)
client = pymongo.MongoClient()
client.db.collection.insert({'date': myDatetime})

Answer 2

ISODate 是一个 JavaScript 日期对象。要使用 PyMongo 查询日期范围，您需要使用

datetime.datetime

实例，mongod 会将其转换为适当的 BSON 类型。您不需要任何第三方库。

此外，您不应该使用聚合框架来执行此操作，因为

_id

字段在集合中是唯一的，这使得它非常适合

distinct()

方法。

import datetime


start = datetime.datetime(2016, 11, 11)
end = datetime.datetime(2016, 11, 11, 23, 59, 59)

db.session.distinct('_id', {'timestamp': {'$gte': start, '$lte': end}})

如果你确实需要使用

aggregate()

方法，你的

$match

阶段必须如下所示：

{'$match': {'timestamp': {'$gte': start, '$lte': end}}}