使用Python脚本写入MongoDB数据

_{使用Python脚本对MongoDB数据库内插入数据}

1、使用python创建项目MongoDB(名字随意)，如下图：

2、在项目下创建DB.conf文件，把可配置的项写入到文件内，如下图：

数据库IP、数据库端口、数据库名字、集合名字（相当于MySQL的表名）、插入的数据、插入数据的条数及脚本共运行时间

 3、创建mongoDB.py文件，如下图：

首先需要一个配置文件操作的类class Config()类，类下面有一个用来获取配置文件的绝对路径的方法def get_config_path()，并且有一个获取配置文件内参数值的def read_config()

 其次需要一个操作MongoDB的类class MongoDB()，类下面有个初始化方法def __init__()，还有个用来连接数据库的方法def connecting()，还有一个以插入总条数来运行的方法def start_insert_num()与 以配置运行时间来运行的方法def start_insert_time()

# @File    : mongoDB_Insert.py
# @Time    : 2018/12/13 10:28
import os,time,configparser
from pymongo import MongoClient

class MongoDB():

    def __init__(self):
        '''
        初始化操作
        :param host:MongoDB Ip
        :param port: MongoDB Port
        :param db_name: MongoDB Name
        :param collection: MongoDB  Collection Name
        '''  
        self.conf_path = Config().get_config_path()
        self.host = Config().read_config(self.conf_path,"MONGODB_CONF","host")
        self.port = int(Config().read_config(self.conf_path,"MONGODB_CONF","port"))
        self.db_name = Config().read_config(self.conf_path,"MONGODB_CONF","db_name")
        self.collection = Config().read_config(self.conf_path,"MONGODB_CONF","collection_name")
        self.insert_data = Config().read_config(self.conf_path,"INSERT_DATA","insert_data")
        self.insert_num =int((Config().read_config(self.conf_path,"INSERT_NUM","insert_num")))
        self.insert_time =int((Config().read_config(self.conf_path,"INSERT_TIME","insert_time")))

    def connecting(self):
        '''
        DB连接
        :return:    my_collection
        '''
        try:
            conn = MongoClient(host=self.host,port=self.port)
            my_db = conn[self.db_name]
            my_collection = my_db[self.collection]
            return my_collection
        except Exception as e:
            print('连接失败！错误原因：{}'.format(e))
            raise e

    def start_insert_num(self):
        '''
        插入数据，配置共插入多少数据
        :return:None
        '''
        my_set = self.connecting()
        try:
            i = 0
            start_time = int(time.time() * 1000)
            while True:
                my_set.insert((eval((self.insert_data))))
                i += 1
                print('插入{}条数据。'.format(i))
                if i == self.insert_num:
                    break
            end_time = int(time.time() * 1000)
            print('插入数据成功。共插入{}条数据，耗时{}ms。'.format(self.insert_num,end_time - start_time))
            input('按回车键结束程序......')
        except Exception as e:
            print('插入数据失败！失败原因：{}'.format(e))
            raise e

    def start_insert_time(self):
        '''
        插入数据,配置共运行时间
        :return:None
        '''
        my_set = self.connecting()
        try:
            i = 0
            start_time = int(time.time() * 1000)
            while True:
                my_set.insert((eval((self.insert_data))))
                i += 1
                print('插入{}条数据。'.format(i))
                if int(time.time()*1000) >= (start_time+self.insert_time):
                    break
            print('插入数据成功。共插入{}条数据，耗时{}ms。'.format(i,self.insert_time))
            input('按回车键结束程序......')
        except Exception as e:
            print('插入数据失败！失败原因：{}'.format(e))
            raise e

class Config():
    '''
    读取配置文件类
    '''

    def get_config_path(self):
        '''
        获取配置文件的绝对路径
        :return: 配置文件的绝对路径
        '''
        try:
            conf_path = os.path.split(os.path.realpath(__file__))[0] + r"DB.conf"
            return conf_path
        except Exception as e:
            print('出错啦！！！错误原因：{}'.format(e))
            raise e

    def read_config(self,conf_path,section,option):
        '''
        读取配置文件
        :param conf_path:配置文件绝对路径
        :param section:
        :param option:
        :return: 配置文件内的value
        '''
        try:
            cf = configparser.ConfigParser()
            cf.read(conf_path)
            value = cf.get(section=section, option=option)
            return value
        except Exception as e:
            print('读取配置文件失败！失败原因：{}'.format(e))
            raise e

if __name__ == '__main__':
    #两种不同的运行方式
    #MongoDB().start_insert_num()
    MongoDB().start_insert_time()

 每个人的代码写法不一样，有的喜欢写到一个类里面，有的喜欢分为多个模块写，有的喜欢分为多个类写，看个人喜好吧，只要能实现功能就可以了；

暂时不会多线程，所以导致插入数据库数据很慢，我的解决办法是把代码编译为.exe可执行程序，然后多启动几个进程，虽然还是很慢，但是比较之前强很多；

这个只是单纯写着玩儿，如果大家要是测试数据库读、写性能的话建议使用YCSB工具。