#python批量读取excel csv文件插入mysql数据库

import os
import csv
import argparse
import pymysql
import sys



class ConnectionDatabase(object):
    # 连接mysql数据库
    def __init__(self, ip, user_name, passwd, db, char='utf8'):
        self.ip = ip
        # self.port = port
        self.username = user_name
        self.passwd = passwd
        self.mysqldb = db
        self.char = char

        self.MySQL_db = pymysql.connect(
            host=self.ip,
            user=self.username,
            password=self.passwd,
            database=self.mysqldb,
            charset=self.char)

    def mysql_findList(self, sql):
        cursor = self.MySQL_db.cursor()
        MySQL_sql = sql
        results = None
        if not cursor:
            raise (NameError, "数据库连接失败")
        try:
            # 执行SQL语句
            cursor.execute(MySQL_sql)
            # 获取所有记录列表
            results = cursor.fetchall()
        except Exception as e:
            print(e)
            self.MySQL_db.close()
        if results:
            return results
        else:
            return None

    # 数据增删改查（sqlserver）
    def mysql_exe_sql(self, sql, params):
        cursor = self.MySQL_db.cursor()
        MySQL_sql = sql
        result = 0
        if not cursor:
            raise (NameError, "数据库连接失败")
        try:
            # 执行SQL语句
            self.MySQL_db.ping(True)
            cursor.execute(MySQL_sql, params)
            result = cursor.rowcount
        except Exception as e:
            print(e)
            self.MySQL_db.rollback()
            self.MySQL_db.close()

        return result > 0

    '''
        提交数据
    '''
    def commitData(self):
        try:
            self.MySQL_db.commit()
        except Exception as e:
            print(e)

    '''
        关闭数据库连接
    '''
    def closeConn(self):
        if self.MySQL_db:
            self.MySQL_db.close()


'''
    读取文件夹下的csv文件
'''
def readAllFiles(filePath):
    fileList = os.listdir(filePath)
    i = 0
    for file in fileList:
        path = os.path.join(filePath, file)
        if os.path.isfile(path):
            file = open(path, 'r', encoding='utf-8')
            print(path)
            i += 1
            print("插入第>>>>", i, ">>>>数据表")
            analysisWorkflowCsv(file)
            pass
        else:
            readAllFiles(path)


def analysisWorkflowCsv(file):
    csvFile = csv.reader(file)
    # 读取一行，下面的reader中已经没有该行了
    # head_row = next(csvFile)
    # print(head_row)
    __conn = ConnectionDatabase(ip="localhost", user_name="root", passwd="", db="mydb", char="utf8")
    counter = 0
    for row in csvFile:
        data = {}
        # 获取excel内需要的数据,从0开始,根据导入的csv文件的列数来决定data的容量
        data['a'] = row[0]
        data['b'] = row[1]
        data['c'] = row[2]
        # data['d'] = row[3]
        # data['e'] = row[4]
        # data['f'] = row[5]
        # data['g'] = row[6]
        # data['h'] = row[7]
        # data['i'] = row[8]
        # data['j'] = row[9]
        if insert_data(__conn, data):
            counter += 1
        if counter % 1000 == 0:
            __conn.commitData()
    print("已经插入工作流数据： %d 条。" % counter)
    __conn.commitData()
    __conn.closeConn()


'''
    插入工作流程数据
'''
def insert_data(__conn, data):
    # 在mysql建立表，字段名可以根据需要设置，也可以按a,b,c这样的简单记录也可以。跟data容量的一一对应。
    # __sql = '''
    #     INSERT INTO `mydb`.`tbl_test` (
    #   `a`,
    #   `b`,
    #   `c`,
    #   `d`,
    #   `e`,
    #   `f`,
    #   `g`,
    #   `h`,
    #   `i`,
    #   `j`
    # )
    # VALUES
    #   (
    #     %s,
    #     %s,
    #     %s,
    #     %s,
    #     %s,
    #     %s,
    #     %s,
    #     %s,
    #     %s,
    #     %s
    #   )
    # '''

    __params = (
        data['a'], data['b'],
        data['c'],
        data['d'],
        data['e'],data['f'],data['g'],data['h'],data['i'],data['j']
    )
    # print(__sql % __params)
    return __conn.mysql_exe_sql(__sql, __params)


if __name__ == "__main__":
    # 文件所在的文件夹父路径,按文件夹下面的文件批量导入
    testFilePath = "F:/excel2Mysql/"
    readAllFiles(testFilePath)