数据库学习之将MongoDB的操作封装成类

import pymongo

class MyMonge:
def __init__(self,db,collection):
self.client = pymongo.MongoClient()
self.db = self.client[db]
self.col = self.db[collection]

def insert(self,date,ins_var = True):
if ins_var:
self.col.insert_one(date)
else:
self.col.insert_many(date)

def find(self,date,find_var = True):
if find_var:
result = self.col.find_one(date)
print(result)
else:
result = self.col.find(date)
for i in result:
print(i)

def update(self,data,new_data,update_var = True):
if update_var:
self.col.update_one(data,{'$set':new_data})
else:
self.col.update_many(data,{'$set':new_data})

def delete(self,data,del_var = True):
if del_var:
self.col.delete_one(data)
else:
self.col.delete_many(data)

if __name__ == '__main__':
m = MyMonge('python','stu')
m.update({'age':20},{'age':18},update_var = False)
m.find({'name': 'hansha'},find_var=False)

• 点赞 1
• 收藏
• 分享
• 文章举报