实现一个MONGO工具类
时间: 2023-02-06 11:28:24 浏览: 55
好的,下面是一个基本的 MongoDB 工具类的例子,希望能帮到你:
```python
import pymongo
class MongoDB:
def __init__(self, host, port, database, collection):
self.client = pymongo.MongoClient(host, port)
self.db = self.client[database]
self.collection = self.db[collection]
def insert_one(self, doc):
self.collection.insert_one(doc)
def insert_many(self, docs):
self.collection.insert_many(docs)
def find_one(self, query):
return self.collection.find_one(query)
def find(self, query):
return self.collection.find(query)
def update_one(self, query, update):
self.collection.update_one(query, update)
def update_many(self, query, update):
self.collection.update_many(query, update)
def delete_one(self, query):
self.collection.delete_one(query)
def delete_many(self, query):
self.collection.delete_many(query)
def close(self):
self.client.close()
```
使用方法:
```python
# 创建工具类实例
mongodb = MongoDB('localhost', 27017, 'test', 'test_collection')
# 插入文档
doc = {'name': 'Tom', 'age': 20}
mongodb.insert_one(doc)
# 查询文档
query = {'name': 'Tom'}
result = mongodb.find_one(query)
print(result)
# 关闭连接
mongodb.close()
```
希望这个例子能帮到你。