如何从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)

但我一直收到一个错误：

^{pr2}$

我认为这是因为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