ExceltoDatabase.py

import pymysql
import xlrd
import time
 
'''
    连接数据库
    args：db_name（数据库名称）
    returns:db
     
'''
def mysql_link(de_name):
    try:
        db = pymysql.connect(host="127.0.0.1", user="user",
                             passwd="123456",
                             db=de_name,
                             charset='utf8')
        return db
    except:
        print("could not connect to mysql server")
 
 
'''
    读取excel函数
    args：excel_file（excel文件，目录在py文件同目录）
    returns：book
'''
def open_excel(excel_file):
    try:
        book = xlrd.open_workbook(excel_file)  #文件名，把文件与py文件放在同一目录下
        return book
    except:
        print("open excel file failed!")
 
 
'''
    执行插入操作
    args:db_name（数据库名称）
         table_name(表名称）
         excel_file（excel文件名，记得把文件与py文件放在同一目录下）
          
'''
def store_to(db_name,table_name,excel_file):
    db = mysql_link(db_name)  # 打开数据库连接
    cursor = db.cursor()  # 使用 cursor() 方法创建一个游标对象 cursor
 
    book = open_excel(excel_file)   # 打开excel文件
    sheet = book.sheet_names()[0]   # 获取第一张sheet表名（如果有多张表请用循环）
    sh = book.sheet_by_name(sheet)   # 打开表
    row_num = sh.nrows
    print(row_num)
    list = []   # 定义列表用来存放数据
    for i in range(1, row_num):  # 第一行是标题名，表中的数据应该从第二行开始，所以值是1（0是第一行）
        row_data = sh.row_values(i)  # 按行获取excel的值
        #将Excel表的日期格式转换为数据表的格式
        row_data[1] = xlrd.xldate_as_datetime(row_data[1],0).strftime('%Y-%m-%d %H:%M:%S')
        value = (row_data[0], row_data[1],row_data[2],row_data[3],row_data[4],row_data[5],\
                    row_data[6],row_data[7],row_data[8],row_data[9],row_data[10],row_data[11],row_data[12])
        list.append(value)  # 将数据暂存在列表
        # print(i)
    #下面要根据自己的表格进行更改,对应数量后面的%s数量也要一样,以及上面的SQL中的变量数也要改变
    sql = "INSERT INTO "+ table_name + " (assets,data,saison,cash,\
        mitsui,suica,uber,paypay,linepay,yuchyo,makku,majika,conpini)VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
    cursor.executemany(sql, list)  # 执行sql语句
    db.commit()  # 提交
    list.clear()  # 清空list
    print("worksheets: " + sheet + " has been inserted " + str(row_num) + " datas!")#打印成功信息
    cursor.close()  # 关闭连接
    db.close()
 
 
if __name__ == '__main__':#下面最左边是库名，中间是表名，最后是Excel文件名
    store_to('ebdb','assets','ee.xlsx')