python - 如何从 mongodb 中的集合中获取完整的键列表

Question

如何获取 mongodb 集合中使用的所有键的列表？

我正在尝试使用以下命令在 python 中创建一个 csv 文件:

userAgentProp_list = []
# userAgentProp is a set of user ids
# userColl is a mongodb collection
for user in userAgentProp:
    userAgentProp_list.append(userColl.find_one({"_id":user}))

with open('userWtihAgentProp.csv','w+') as f:
    w = csv.DictWriter(f,userAgentProp_list[0].keys())
    w.writeheader()
    for user in userAgentProp_list:
        # del row['_id']
        w.writerow(user)

但我不断收到错误:

ValueError: dict contains fields not in fieldnames

我认为这是因为 userAgentProp_list[0] 并不具有与集合中其他文档相同的字段。

如何获取集合中使用的所有键的列表？

Answer 1

我想我已经找到解决方案了。我希望有更多经验的人告诉我这是否正确和/或可以做得更好/更有效。

def getCollectionKeys(collection):
    """Get a set of keys from a collection"""
    keys_list = []
    collection_list = collection.find()
    for document in collection_list:
        for field in document.keys():
            keys_list.append(field)
    keys_set =  set(keys_list)
    return keys_set