pymongo 简单操作

我就是我 2022-06-17 08:44 184阅读 0赞

# -*- coding: utf-8 -*-
    #__author__="ZJL"
    
    
    from pymongo import MongoClient
    import traceback
    
    
    # mongdb数据库链接
    class MongoDBManager(object):
        def __init__(self, mongodb_uri):
            try:
                self.host = mongodb_uri['host']
                self.port = mongodb_uri['port']
                self.database = mongodb_uri['database']
                self.user_name = mongodb_uri['user_name']
                self.user_password = mongodb_uri['password']
                self.conn = self.createConn(mongodb_uri)
            except Exception as e:
                print(traceback.format_exc())
                print(str(e))
    
    
        # 插入数据
        def insert(self, col, data):
            if isinstance(data, dict):
                try:
                    self.conn[self.database][col].insert_one(data)
                except Exception as e:
                    print(traceback.format_exc())
                    print('insert_one error ' + str(e))
                    self.conn.close()
                    return traceback.format_exc()
            elif isinstance(data, list):
                try:
                    self.conn[self.database][col].insert_many([var for var in data])
                except Exception as e:
                    print(traceback.format_exc())
                    print('insert_many error ' + str(e))
                    self.conn.close()
                    return traceback.format_exc()
            else:
                print('data is not a dict or a list of dicts')
                return 'data is not a dict or a list of dicts'
    
    
        # 更新
        def update(self, col, data, filter, upsert=True):
            if isinstance(data, dict):
                modify = '$set'
                modify_data = data
                try:
                    update_result = self.conn[self.database][col].update_one(
                        filter=filter,
                        update={modify: modify_data},
                        upsert=upsert
                    )
                    return update_result
                except Exception as e:
                    print(traceback.format_exc())
                    print('update data error:' + str(e))
                    self.conn.close()
                    return traceback.format_exc()
            else:
                print ('data is not a dict or a list of dicts')
                return 'data is not a dict or a list of dicts'
    
    
        # 碰到一个字段加减法问题新增这个函数
        # collection = self.conn[self.database][col].update({'_id': 1111}, {'$inc': {'aa': 1}})
        #'$inc'用来加减，-1就是减法
        def update_new(self, col, filter,data):
            if isinstance(data, dict) and isinstance(filter,dict):
                try:
                    update_result = self.conn[self.database][col].update(filter,data)
                except Exception as e:
                    print(traceback.format_exc())
                    print('update_new data error:' + str(e))
                    self.conn.close()
                    return traceback.format_exc()
                else:
                    return update_result
            else:
                print ('data is not a dict or a list of dicts')
                return 'data is not a dict or a list of dicts'
    
    
        # 查询数据，返回游标,skip,limit用于分页
        # skip为跳过多少条数据，limit为显示多少条数据，skip,limit都是0，显示所有
        # def find_one(self, col, filter={}, kwargs={}, skip=0, limit=0):
            # kwargs用于返回指定字段
            # if kwargs:
                # return self.find(col, filter, kwargs)
            # else:
                # return self.find(col, filter)
    
    
        # 查询一条数据
        def find(self, col, filter={}, kwargs={}):
            if isinstance(filter, dict):
                try:
                    # kwargs用于返回指定字段
                    if kwargs:
                        return self.conn[self.database][col].find_one(filter, kwargs)
                    else:
                        return self.conn[self.database][col].find_one(filter)
                except Exception as e:
                    print(traceback.format_exc())
                    print(str(e))
                    self.conn.close()
                    return traceback.format_exc()
            return
    
    
        # 查询多条数据，返回一个可迭代类型
        # skip, limit用于分页，skip为跳过多少条数据，limit为显示多少条数据，skip,limit都是0，显示所有
        # sortkey是排序关键字，sort是排序，-1是降序，1是升序
        def find_cursor(self, col, filter={}, kwargs={}, skip=0, limit=0, sortkey="",sort=-1):
            if isinstance(filter, dict):
                try:
                    # kwargs用于返回指定字段
                    if kwargs and sortkey:
                        return self.client[self.database][col].find(filter, kwargs).skip(skip).limit(limit).sort(sortkey,sort)
                    elif kwargs and not sortkey:
                        return self.client[self.database][col].find(filter, kwargs).skip(skip).limit(limit)
                    elif not kwargs and sortkey:
                        return self.client[self.database][col].find(filter).skip(skip).limit(limit).sort(sortkey,sort)
                    else:
                        return self.client[self.database][col].find(filter).skip(skip).limit(limit)
                except Exception as e:
                    print(traceback.format_exc())
                    print(str(e))
                    self.conn.close()
                    return traceback.format_exc()
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # find_cursor_data = list(find_cursor()) 是一个list
        # find_cursor()的结果赋值给一个空list，结果就是一个List,所以这个方法舍弃
        # # 查询多条数据，返回一个list
        # skip, limit用于分页，skip为跳过多少条数据，limit为显示多少条数据，skip,limit都是0，显示所有
        # def find_cursor_list(self, col, filter={}, kwargs={}, skip=0, limit=0):
            # if isinstance(filter, dict):
                # try:
                    # data_list = []
                    # # kwargs用于返回指定字段
                    # if kwargs:
                        # data_dict = self.conn[self.database][col].find(filter, kwargs).skip(skip).limit(limit)
                        # if data_dict:
                            # for d in data_dict:
                                # data_list.append(d)
                        # return data_list
                    # else:
                        # data_dict = self.conn[self.database][col].find(filter).skip(skip).limit(limit)
                        # if data_dict:
                            # for d in data_dict:
                                # data_list.append(d)
                        # return data_list
                # except Exception as e:
                    # print(traceback.format_exc())
                    # print(str(e))
                    # self.conn.close()
                    # return traceback.format_exc()
            # else:
                # print("type not dict!")
                # return "type not dict!"
    
    
        # 删除数据
        def remove(self, col, filter={}):
            if isinstance(filter, dict):
                try:
                    return self.conn[self.database][col].remove(filter)
                except Exception as e:
                    print(traceback.format_exc())
                    print(str(e))
                    self.conn.close()
                    return traceback.format_exc()
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # 统计
        def count(self, col, filter={}):
            if isinstance(filter, dict):
                try:
                    return self.conn[self.database][col].find(filter).count()
                except Exception as e:
                    print(traceback.format_exc())
                    print(str(e))
                    self.conn.close()
                    return traceback.format_exc()
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # 连接数据库
        @classmethod
        def createConn(cls, mongodb_uri):
            try:
                conn = MongoClient(
                    host=mongodb_uri['host'],
                    port=mongodb_uri['port'],
                )
                db = conn[mongodb_uri['database']]
                # 身份认证,有用户名密码放开这段注释
                # db.authenticate(name=mongodb_uri['user_name'], password=mongodb_uri['password'])
                return conn
            except Exception as e:
                print(traceback.format_exc())
                print('Mongo connected Fail!', str(e))
                return traceback.format_exc()
    
    
        # 关闭数据库
        def close(self):
            self.conn.close()
    
    
    ''''' 
    #以下例子  
    MONGODB_URI={  
        "host":"127.0.0.1",  
        "port": 27017,  
        "database":"py_demo",  
        "user_name":"",  
        "password":"",  
    }  
    
    
    def main():  
        mb = MongoDBManager(MONGODB_URI)  
        #删除操作  
        # print(mb.remove("zjl_demo",{"a":"1"}))  
        #单条插入：  
        # mb.insert('zjl_demo', {'a': ' 哈哈哈', 'b': [1, 2, 3, 4]},)  
        #批量插入操作  
        # mb.insert('zjl_demo',[{'a':'1','b':[1,2,3,4,"哈哈哈"]},{'a':'2','b':[1,2,3,4,"哈哈哈"]}])  
        #更新操作  
        # mb.update('zjl_demo', {"aa":["aaaaa","bbb哈哈哈bb"]},{'a': "2"})  
        #查询操作  
        # print(mb.find("zjl_demo",{"_id":"4"}))  
        #字段后面1是显示，id默认显示，0为不显示 
        # print(mb.find("zjl_demo",{"_id":"4"},{"aaa":1,"_id":0})) 
        # print(mb.find_one("zjl_demo", {"_id":"4"}))  
        # for i in mb.find_cursor("zjl_demo", {}):  
        #     print(i)  
        #统计  
        # print(mb.count("zjl_demo",{}))  
        mb.close()  
    
    
    if __name__ == "__main__":  
        main()  
    '''

加一个修饰器

# -*- coding: utf-8 -*-
    #__author__="ZJL"
    
    
    from pymongo import MongoClient
    import traceback
    
    
    def decorator(func):
        def wrapper(self,*args,**kwargs):
            try:
                return func(self,*args, **kwargs)
            except Exception as e:
                print(traceback.format_exc())
                print(str(e))
                self.conn.close()
                return traceback.format_exc()
        return wrapper
    
    
    def decorator_conn(func):
        def wrapper(self,*args,**kwargs):
            try:
                return func(self,*args, **kwargs)
            except Exception as e:
                print(traceback.format_exc())
                print(str(e))
                return traceback.format_exc()
        return wrapper
    
    
    # mongdb数据库链接
    class MongoDBManager(object):
        @decorator_conn
        def __init__(self, mongodb_uri):
            self.host = mongodb_uri['host']
            self.port = mongodb_uri['port']
            self.database = mongodb_uri['database']
            self.user_name = mongodb_uri['user_name']
            self.user_password = mongodb_uri['password']
            self.conn = self.createConn(mongodb_uri)
    
    
        # 插入数据
        @decorator
        def insert(self, col, data):
            if isinstance(data, dict):
                self.conn[self.database][col].insert_one(data)
            elif isinstance(data, list):
                self.conn[self.database][col].insert_many([var for var in data])
            else:
                print('data is not a dict or a list of dicts')
                return 'data is not a dict or a list of dicts'
    
    
        # 更新
        @decorator
        def update(self, col, data, filter, upsert=True):
            if isinstance(data, dict):
                modify = '$set'
                modify_data = data
                update_result = self.conn[self.database][col].update_one(
                    filter=filter,
                    update={modify: modify_data},
                    upsert=upsert
                )
                return update_result
            else:
                print ('data is not a dict or a list of dicts')
                return 'data is not a dict or a list of dicts'
    
    
        # 碰到一个字段加减法问题新增这个函数
        # collection = self.conn[self.database][col].update({'_id': 1111}, {'$inc': {'aa': 1}})
        #'$inc'用来加减，-1就是减法
        @decorator
        def update_new(self, col, filter,data):
            if isinstance(data, dict) and isinstance(filter,dict):
                update_result = self.conn[self.database][col].update(filter,data)
                return update_result
            else:
                print ('data is not a dict or a list of dicts')
                return 'data is not a dict or a list of dicts'
    
    
        # 查询数据，返回游标,skip,limit用于分页
        # skip为跳过多少条数据，limit为显示多少条数据，skip,limit都是0，显示所有
        # @decorator
        # def find_one(self, col, filter={}, kwargs={}, skip=0, limit=0):
            # kwargs用于返回指定字段
            # if kwargs:
                # return self.find(col, filter, kwargs)
            # else:
                # return self.find(col, filter)
    
    
        # 查询一条数据
        @decorator
        def find(self, col, filter={}, kwargs={}):
            if isinstance(filter, dict):
                # kwargs用于返回指定字段
                if kwargs:
                    return self.conn[self.database][col].find_one(filter, kwargs)
                else:
                    return self.conn[self.database][col].find_one(filter)
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # 查询多条数据，返回一个可迭代类型
        # skip, limit用于分页，skip为跳过多少条数据，limit为显示多少条数据，skip,limit都是0，显示所有
        # sortkey是排序关键字，sort是排序，-1是降序，1是升序
        @decorator
        def find_cursor(self, col, filter={}, kwargs={}, skip=0, limit=0, sortkey="",sort=-1):
            if isinstance(filter, dict):
                # kwargs用于返回指定字段
                    if kwargs and sortkey:
                        return self.client[self.database][col].find(filter, kwargs).skip(skip).limit(limit).sort(sortkey,sort)
                    elif kwargs and not sortkey:
                        return self.client[self.database][col].find(filter, kwargs).skip(skip).limit(limit)
                    elif not kwargs and sortkey:
                        return self.client[self.database][col].find(filter).skip(skip).limit(limit).sort(sortkey,sort)
                    else:
                        return self.client[self.database][col].find(filter).skip(skip).limit(limit)
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # find_cursor_data = list(find_cursor()) 是一个list
        # find_cursor()的结果赋值给一个空list，结果就是一个List,所以这个方法舍弃
        # 查询多条数据，返回一个list
        # skip, limit用于分页，skip为跳过多少条数据，limit为显示多少条数据，skip,limit都是0，显示所有
        # @decorator
        # def find_cursor_list(self, col, filter={}, kwargs={}, skip=0, limit=0):
            # if isinstance(filter, dict):
                # data_list = []
                # # kwargs用于返回指定字段
                # if kwargs:
                    # data_dict = self.conn[self.database][col].find(filter, kwargs).skip(skip).limit(limit)
                    # if data_dict:
                        # for d in data_dict:
                            # data_list.append(d)
                    # return data_list
                # else:
                    # data_dict = self.conn[self.database][col].find(filter).skip(skip).limit(limit)
                    # if data_dict:
                        # for d in data_dict:
                            # data_list.append(d)
                    # return data_list
            # else:
                # print("type not dict!")
                # return "type not dict!"
    
    
        # 删除数据
        @decorator
        def remove(self, col, filter={}):
            if isinstance(filter, dict):
                return self.conn[self.database][col].remove(filter)
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # 统计
        @decorator
        def count(self, col, filter={}):
            if isinstance(filter, dict):
                return self.conn[self.database][col].find(filter).count()
            else:
                print("type not dict!")
                return "type not dict!"
    
    
        # 连接数据库
        @classmethod
        @decorator_conn
        def createConn(cls, mongodb_uri):
            conn = MongoClient(
                host=mongodb_uri['host'],
                port=mongodb_uri['port'],
            )
            db = conn[mongodb_uri['database']]
            # 身份认证,有用户名密码放开这段注释
            # db.authenticate(name=mongodb_uri['user_name'], password=mongodb_uri['password'])
            return conn
    
    
        # 关闭数据库
        def close(self):
            self.conn.close()

发现一个问题find\_cursor()函数得到的对象迭代一次以后居然就不能迭代了，什么鬼？如果需要多次操作这个对象可以将他赋值给一个空的List，这样他就成了一个list