python連接集群mongodb，封裝增刪改查

1.下載pymongo

pip install pymongo

2.直接上代碼

[ini配置文件]

 

 封裝讀ini省略~~

[db.py]

class Database():

    def __init__(self):
        self.cf = ConfigParser.ConfigParser()
        self.mq = StringManipulation
    

     # 操作mongodb
    def db_mongo(self,Positioning_mode,Location_element,Input_content):

        self._mongo_p= self.mq._mongo_path()
        self.cf.read(self._mongo_p)

        db_host = self.cf.get(Positioning_mode, "host")

        operation_mr=['delete','select','update','insert']

        get_mer=Location_element[0:6]

        count_str=Location_element.count('.')

        if Location_element and count_str == 2:

            get_me_list = Location_element.split('.')

            get_method=get_me_list[0]

            get_base=get_me_list[1]

            get_table=get_me_list[2]
            if get_method in operation_mr and get_method and get_base and  get_table:

                print(get_method)
                print(get_base)
                print(get_table)

                # 連接mongodb 目前只寫了集群的連接方法
                client = MongoClient(db_host)

                # 目標數據庫
                db=eval("client."+get_base)

                # 指定表
                collection = eval("db."+get_table)
                # 需要select
                # Input_content=eval(Input_content)
                if Input_content:
                    if get_method == 'select':
                        Input_content = eval(Input_content)
                        for get_select in collection.find(Input_content):
                            print("查詢的數據成功:")
                            print(get_select)

                    elif get_method == 'update':
                        Input_content=eval(Input_content)
                        # print(Input_content)

                        get_find_dict=Input_content[0]
                        get_set_dict=Input_content[1]
                        collection.update(get_find_dict, {"$set":get_set_dict})

                        print("數據修改成功")

                        # collection.update({"userName":"[e-mail]"},{"$set":{"code":"1783558010"}})

                    elif get_method =='insert':

                        Input_content=eval(Input_content)
                        if isinstance(Input_content,tuple):
                            print(type(Input_content))
                            insert_list=[]
                            for i in Input_content:
                                insert_list.append(i)
                            collection.insert(insert_list)
                            # print(insert_list)
                            print("插入數據成功")
                        else:
                            collection.insert(Input_content)
                            print("插入數據成功")

                    elif  get_method =='delete':
                        print('delete')
                        Input_content=eval(Input_content)
                        if isinstance(Input_content,tuple):
                            for get_del_data in Input_content:
                                collection.remove(get_del_data)
                                print("數據已刪除")

                        else:
                            print("非t")
                            collection.remove(Input_content)
            else:
                print("操作關鍵字或數據庫或表填寫格式有誤")
        else:
            print("操作關鍵字或數據庫或表填寫格式有誤")

　　

調用上面的方法增刪改查

"""
#參數解釋
Positioning_mode 對應配置文件[base_test],連接指定的mongodb
Location_element  參數格式: 操作方法.數據庫.表    
Input_content   增刪該查的條件，mongo都是以字典的形式
"""


# 刪除數據
# Database().db_mongo('base_test','delete.mydb.email','{"userName" : "[e-mail]"},{"userName" : "[e-mail]"}')
#
# 插數據，多個或單個
# Database().db_mongo('base_test','insert.mydb.email','{"userName":"[e-mail]","code" : "9999999999"},{"userName":"[e-mail]","code" : "99999999999"}')
#
# 查詢
# Database().db_mongo('base_test','select.mydb.email','{"userName":"[e-mail]"}')
#
#
# 修改
# Database().db_mongo('base_test','update.mydb.email','{"userName":"[e-mail]"},{"code":"1783558010","success":True}')
#
# {"code":"9999999999"}
#