Python操作MongoDB数据库

from pymongo import MongoClient

cf = configparser.ConfigParser()
cf.read("conf.conf")

class Mongo_Operation:
    def __init__(self):
        self.host = cf.get("mongDB_address", "host")
        self.port = cf.get("mongDB_address", "port")
        self.username = cf.get("mongDB_address", "username")
        self.password = cf.get("mongDB_address", "password")
        url = 'mongodb://' + self.username + ':' + self.password + \
        '@' + self.host + ':' + self.port + '/AI'
        self.m = MongoClient(url)

    def insert_data(self, model_message):
        db = self.m.AI  # 连接AI数据库，没有则自动创建
        my_set = db.AI_set  # 使用AI_set集合，没有则自动创建
        my_set.insert(model_message)
        return 1

    def del_data(self, model_message):
        db = self.m.AI
        my_set = db.AI_set
        my_set.remove(model_message)
        return 1

    def select_data(self, model_name):
        db = self.m.AI
        my_set = db.AI_set
        model_message = my_set.find_one(model_name)
        if model_message is not None:
            return model_message
        else:
            return {"Model_File": "1"}

    def select_all_data(self):
        db = self.m.AI
        my_set = db.AI_set
        file_list = my_set.find()
        return file_list

cf.get()是从配置文件从读取信息，各个函数的作用如同函数名一样，导入后直接调用。这里需要强调的是，我用的是MongDB2.6，而3.0及以上版本与此有差异，需要读者注意。