如何使用 pymongo 编写连接到 mongodb 的函数

Question

我正在尝试编写一个函数来连接到 mongodb 和 return 一个集合对象。我有以下内容：

def getCollection(dbname,collection):
    client = MongoClient()
    data_base = client.dbname
    collObject = data_base.collection
    return collObject

当我运行:

collection = getCollection(client, "hkpr_restore", "agents")
print collection

我得到：

Collection(Database(MongoClient('localhost', 27017), u'dbname'), u'collection')

我做错了什么？

Answer 1

当使用 client.dbname 时，会调用属性 dbname，这意味着您正在检索名为 dbname 的数据库。

同样适用于 data_base.collection。

解决方案：

def getCollection(dbname, collection):
    client = MongoClient()
    data_base = getattr(client, dbname)
    collObject = getattr(data_base, collection)
    return collObject

备选方案：您可以使用字典式访问：

def getCollection(dbname, collection):
    client = MongoClient()
    data_base = client[dbname]
    collObject = data_base[collection]
    return collObject

如何使用 pymongo 编写连接到 mongodb 的函数

how do I write a function for connecting to mongodb using pymongo

python

mongodb

pymongo