MongoDB数据库操作的封装

import pymongo
class MongoDB:
    # 初始化
    def __init__(self,ip='192.168.111.109',user='admin',pwd='sq',port=27017):
        client=pymongo.MongoClient(f'mongodb://{user}:{pwd}@{ip}:{port}')
        self.db=client['test']
    # 插入操作
    def insert(self,collection,query,many=False):
        res_set=self.db[collection]
        if many==False:
            res_set.insert_one(query)
        elif many==True:
            res_set.insert_many(query)
    # 查询操作
    def find(self,collection,query,many=True):
        res_set = self.db[collection]
        result=[]
        if many==True:
            res=res_set.find(query)
            for one in res:
                result.append(one)
            return result
        elif many==False:
            result=res_set.find_one(query)
            return result
    # 修改操作
    def update(self,collection,myquery,newvalues,many=False):
        res_set = self.db[collection]
        if many==False:
            res_set.update_one(myquery,{'$set':newvalues})
        elif many==True:
            res_set.update_many(myquery,{'$set':newvalues})
    # 删除操作
    def delete(self,collection,query,many=False):
        res_set = self.db[collection]
        if many==False:
            res_set.delete_one(query)
        elif many==True:
            res_set.delete_many(query)

if __name__ == '__main__':
    db=MongoDB()
    res=db.find('categories',{'count':10})
    print(res)
    db.update('categories',{'count':10},{'level':2})
    res1=db.find('categories', {'count': 10})
    print(res1)
    db.delete('categories', {'count': 10})