如何在Python函数中动态设置MongoDB集合名以写入CSV数据

问题原因 #

使用点号访问pymongo的数据库集合时，点后面的内容会被当作固定的属性名处理，不会解析为对应变量的实际值，因此你当前的写法会固定生成名为file_name的集合，无法读取入参的实际值。

修改方法 #

改用方括号索引的方式访问集合，将存储集合名的变量放入方括号内即可：

def write_toDB(file_name, my_key):
    ...
    # 动态读取file_name变量的值作为集合名
    db[file_name].insert_one(data)

补充说明 #

• 如果传入的file_name可能包含特殊字符、空格或者为空，建议提前做合法性校验，避免生成不符合MongoDB命名规范的集合
• 该写法同样适用于动态指定数据库名，用法为client[数据库名][集合名]

内容的提问来源于stack exchange，提问作者Peter Lawrence