如何通过Mongoengine查询MongoDB中含增量后缀键的字典字段，匹配键前缀及子文档内容

我理解你现在遇到的问题了——原来的media字段里是固定键名，现在为了支持多个同类型的媒体，改成了带增量后缀的键（比如image_1、image_2），之前直接按固定键查询的方法就完全失效了。别担心，我们可以利用MongoDB的数组转换和条件匹配能力，结合Mongoengine来实现需求。

问题根源 #

原来的查询Q(**{f"media.{file_type}.title": search_term})是精确匹配单个键名，但现在media下的键是动态生成的带增量后缀的名称，没法提前指定具体键，所以需要遍历所有键来判断是否符合前缀规则，同时子文档的title匹配搜索词。

解决方案一：使用聚合管道（返回字典结果） #

我们可以先把media这个字典转换成键值对数组，再筛选出符合条件的文档：

from mongoengine import Document, StringField, DictField

# 假设你的GameDocument定义如下
class GameDocument(Document):
    _id = StringField(primary_key=True)
    game = StringField()
    media = DictField()

search_term = "red mage"
file_type = "image"

# 构建聚合管道
pipeline = [
    # 将media字典转为[{"k": "键名", "v": "子文档"}]的数组
    {"$addFields": {"media_kv": {"$objectToArray": "$media"}}},
    # 匹配数组中存在满足条件的元素：键名以file_type开头（如image_1），且子文档title匹配
    {"$match": {
        "media_kv": {
            "$elemMatch": {
                "k": {"$regex": f"^{file_type}_\\d+$"},  # 精确匹配前缀+下划线+数字的格式
                "v.title": search_term
            }
        }
    }},
    # 移除临时生成的media_kv字段，保持结果结构和原文档一致
    {"$project": {"media_kv": 0}}
]

# 执行聚合查询，返回字典格式的结果
found_documents = list(GameDocument.objects.aggregate(*pipeline))

解决方案二：使用Q表达式结合原生MongoDB查询（返回Document实例） #

如果你希望直接返回GameDocument的实例，而不是字典，可以用__raw__参数结合MongoDB的表达式查询：

search_term = "red mage"
file_type = "image"

# 构建原生MongoDB查询表达式
query = Q(
    __raw__={
        "$expr": {
            "$anyElementTrue": {
                "$map": {
                    "input": {"$objectToArray": "$media"},
                    "as": "item",
                    "in": {
                        "$and": [
                            # 匹配键名是否符合前缀+增量的格式
                            {"$regexMatch": {"input": "$$item.k", "regex": f"^{file_type}_\\d+$"}},
                            # 匹配子文档的title字段
                            {"$eq": ["$$item.v.title", search_term]}
                        ]
                    }
                }
            }
        }
    }
)

# 执行查询，返回GameDocument实例
found_documents = GameDocument.objects(query)

优化建议（长远方案） #

虽然上面的方法能解决当前问题，但带增量后缀的键名结构在查询和维护上都不太高效。如果后续还有更多类似的查询需求，建议把media字段改成数组结构，比如：

{
  "_id": "ffgame",
  "game": "Final Fantasy",
  "media": [
    {"type": "image", "title": "red mage", "file_size": 2045432, "index": 1},
    {"type": "movie", "title": "characters dancing", "file_size": 2045432, "index": 1},
    {"type": "image", "title": "blue mage", "file_size": 2045432, "index": 2}
  ]
}

这样查询就会简单很多，直接用Mongoengine的嵌套查询：

query = Q(media__type=file_type, media__title=search_term)
found_documents = GameDocument.objects(query)

这种结构不仅查询更高效，也更容易扩展和维护，避免了动态键名带来的各种麻烦。

备注：内容来源于stack exchange，提问作者linkey apiacess