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函数的快捷方式。

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

在管道的情况下，您正在使用$project阶段返回一种只包含name和brating字段的新型文档。

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

• 将brating字段存储在Post文档上。在创建post和更新$total_score或$total_votes时，将评等初始化为0，同时更新评等。
• 接受您要返回的非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]