### Python MongoDB 操作工具类 对于Python操作MongoDB的需求，`pymongo`是一个广泛使用的官方驱动程序[^3]。下面展示了一个基于`pymongo`构建的操作MongoDB的简单工具类实例。 #### 工具类实现 此工具类支持基本的数据帧保存功能，并考虑到了多线程环境下的安全性[^4]： ```python from pymongo import MongoClient, errors import pandas as pd class MongoTool: def __init__(self, host='localhost', port=27017, username=None, password=None, authSource='admin'): self.client = None try: # 创建客户端并尝试连接到MongoDB服务器 self.client = MongoClient(host, port, username=username, password=password, authSource=authSource) info = self.client.server_info() # 测试连接是否成功 print("Connected to MongoDB.") except errors.ServerSelectionTimeoutError as err: print(f"Could not connect to server: {err}") def save_dataframe(self, df: pd.DataFrame, db_name: str, collection_name: str, mode='append'): """ 将Pandas DataFrame保存至指定集合中 :param df: 要存储的数据框对象 :param db_name: 数据库名称 :param collection_name: 集合名称 :param mode: 插入模式 ('append' 或 'overwrite') """ database = self.client[db_name] collection = database[collection_name] records = df.to_dict(orient='records') if mode == 'overwrite': collection.drop() result = collection.insert_many(records) print(f"{len(result.inserted_ids)} documents inserted.") ``` 该工具类允许用户轻松地将数据帧中的记录批量写入MongoDB集合内。如果指定了覆盖模式，则会先删除现有集合再执行插入；如果是追加模式则直接向已有集合添加新文档。