Python批量写入数据到MySQL

'''批量将Excel数据写入MySQL'''
import pymysql
import xlrd

def mysql_link(db_name):
  '''
    连接数据库
    guest:数据库名称
    return:db
	'''
    try:
        db = pymysql.connect(host='192.168.0.*',
                             user='username',
                             passwd='password',
                             db=db_name,
                             charset='utf8mb4')
        return db
    except ConnectionError as e:
        print("数据库连接异常!" + str(e))

def open_excel(excel_file):
	'''
    读取Excel函数
    excel_file:表格文件路径
    return：book
	'''
  try:
        book = xlrd.open_workbook(excel_file)  # 同一目录下
        return book
    except Exception as e:
        print(str(e))

def store_to(db_name, table_name, file_url):
  	'''
    执行插入操作
    db_name（数据库名称）
    table_name(表名称）
    excel_file（excel文件名，把文件与py文件放在同一目录下）
		'''
    db = mysql_link(db_name)  # 打开数据库连接
    cursor = db.cursor()  # 使用 cursor() 方法创建一个游标对象 cursor

    book = open_excel(file_url)  # 打开excel文件
    sheets = book.sheet_names()  # 获取所有表名
    for sheet in sheets:
        sh = book.sheet_by_name(sheet)  # 打开每一张表
        row_nuw = sh.nrows
        print(row_nuw - 1)
        listdata = []  # 定义列表原来存放数据
        for i in range(1, row_nuw):  # 表格首行title对应插入对应表的字段名
            row_data = sh.row_values(i)  # 按行获取Excel的值
            listdata.append(row_data)  # 将数据暂存在list
        print(listdata)
        sql = "INSERT INTO " + table_name + "(tracking_name, tracking_no, create_date, sort_num, del_flag) " \
                                            "VALUES (%s, %s, %s, %s, %s)"
        cursor.executemany(sql, listdata)  # 执行SQL语句
        db.commit()
        listdata.clear()
        print('插入成功！' + '共计' + str(row_nuw - 1) + '条数据!')
    cursor.close()
    db.close()


if __name__ == '__main__':
    store_to('cccivaoperate', 't_tracking', '快递.xlsx')