这是一个简单的 MongoDB 客户端工具类,可以用来创建集合、判断集合是否存在、以及进行文档的增删改查操作:
import pymongoclass MongoClient:
def __init__(self, host, port, database):
self.client = pymongo.MongoClient(host, port)
self.db = self.client[database]
def create_collection(self, collection_name):
self.db.create_collection(collection_name)
def collection_exists(self, collection_name):
return collection_name in self.db.list_collection_names()
def insert_one(self, collection_name, document):
self.db[collection_name].insert_one(document)
def insert_many(self, collection_name, documents):
self.db[collection_name].insert_many(documents)
def find_one(self, collection_name, query):
return self.db[collection_name].find_one(query)
def find(self, collection_name, query):
return self.db[collection_name].find(query)
def update_one(self, collection_name, query, update):
self.db[collection_name].update_one(query, update)
def update_many(self, collection_name, query, update):
self.db[collection_name].update_many(query, update)
def delete_one(self, collection_name, query):
self.db[collection_name].delete_one(query)
def delete_many(self, collection_name, query):
self.db[collection_name].delete_many(query)