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" }

这是mydatabase的示例结构：

> 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中的一个函数，它是一个javascript环境，在Python中不可用。

您可以使用dateutil将字符串转换为Python中的datetime对象，

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 Date对象。要使用PyMongo查询日期范围，您需要使用datetime.datetime实例，mongod将转换为适当的BSON类型。你不需要任何第三方图书馆。

此外，您不应该使用聚合框架来执行此操作，因为_id字段在集合中是唯一的，这使得distinct()方法成为完美的工作。

import datetime


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

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

如果您确实需要使用aggregate()方法，则$match阶段必须如下所示：

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

pymongo：名称'ISODate'未定义

2 个答案: