如何以xxx[dbname?]的形式使用mongoengine

Question

通过使用 pymongo，我可以像这样连接到一个集合：

conn = pymongo.MongoClient()
db = conn.example
collection = db.col

然后，我可以连接到名为“example”的数据库和名为“col”的集合。

而且我还可以编写像这样的代码：

dbname = 'example'
collectionName = 'col'
conn = pymongo.MongoClient()
db = conn[dbname]
connection = db[collectionName]

这两个代码块会做同样的事情。

现在，我使用 mongoengine 连接到 MongoDB，并且我有一个带有 collectionA 的数据库，我还有一个名为 'model.py' 的文件中的类：

class collectionA(Document):
    A=StringField()
    B= StringField()

所以我只能使用：

import model
x = model.collectionA.object()

查询数据。

因为我有这么多收藏，所以我想用

model[collectionName].object() 

查询数据，但是抛出异常，mongoengine好像不支持model[xxx]的形式连接集合。

那么，有什么办法可以解决这个问题吗？

Answer 1

您正在将 model.py 作为模块导入。方括号语法是对象上getitem 方法的“快捷方式”。 Because everything in python is an object，包括模块，你会得到一个TypeError，因为模块没有getitem 方法。

您想要实现的目标可以使用inspect 模块完成：

import inspect

import model

collections = dict(inspect.getmembers(model, inspect.isclass))

collections[collectionName].object() 现在可以工作了。

FWIW 这似乎不是很pythonic，而 mongoengine 的强大之处在于您将文档映射到明确定义的类，这似乎有点混淆了这一点。它可以很容易地引入难以发现的错误，并且还需要额外的击键，因为现在每个类都被引用为字符串。