Flask_Mongoengine：使用聚合后不是 mongoengine 对象

Question

当aggregate与mongoengine一起使用时，它返回一个CommandCursor而不是mongoengine对象列表，这意味着mongonengine并没有真正被使用，

例如：如果某些文档没有标题字段，则会引发错误。如何将结果转换为 mongoengine 对象？

class Post(Document):
    title = StringField(max_length=120, required=True)
    author = ReferenceField(User)

Host.objects()
# [<Post: Post object>, <Post: Post object>, ...]

pipeline = [
    {
        "$match": {
            'types': type,
        }
    },
    {
        "$project": {
            "name": 1,
            'brating': {
                "$divide": [
                    {"$add": ["$total_score", 60]},
                    {"$add": ["$total_votes", 20]}
                ]
            }
        }
    },
    {"$sort": {"brating": -1}},
    {"$limit": 100}

]

Host.objects.aggregate(*pipeline)
# <class 'pymongo.command_cursor.CommandCursor'>

list(Host.objects.aggregate(*pipeline))
# <class 'list'>

Answer 1

aggregate 函数只是底层pymongo function 的快捷方式。

从aggregate 返回的文档可能涉及一些$group 或其他阶段，这意味着它们与您的对象模型无关，因此 mongoengine 无法将它们转换为 mongoengine 对象。

就您的管道而言，您正在使用$project 阶段来返回只有name 和brating 字段的新型文档。

Mongoengine 不能在这里做你想做的事，所以你有几个选择：

• 将brating 字段存储在Post 文档中。创建帖子时将评分初始化为 0，当更新 $total_score 或 $total_votes 时，也更新评分。

• 接受您正在取回非 mongoengine 对象并相应地处理它们。光标将生成普通的 Python 字典，然后您可以访问客户端代码中的字段 post['name'] 或 post['brating']。

• 使用普通的.objects 查询并在客户端进行排序。

如果您有很多文档，但对于少数文档，请尝试以下步骤：

posts = Post.objects(types=type).only("name", "total_score", "total_votes")
top_posts = sorted(list(posts),key=lambda p: (p.total_score+60)/(p.total_votes+20))[:100]