MongoDB数据库操作的封装

以你之姓@ 2022-08-29 00:23 188阅读 0赞

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})