使用Excel初始化数据库

# 读取excel表的内容然后写入数据库
import xlrd,pymysql

# 数据库配置
host = 'localhost' # localhost:本地数据库
user = 'root' # 用户名
password = '123456' # 密码
database = 'test_database' # 数据库名称
charset = 'utf8' # 编码方式

# 创建数据库连接
conn = pymysql.connect(host=host, user=user, password=password
                           , database=database, charset=charset)
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

excel_file = "#I1FGQ0数据库.xlsx" # 设置excel文件名
file = xlrd.open_workbook(excel_file) # 读取excel文件

sheets = file.sheet_names() #获取所有的sheet页
for sheet in sheets:
    sheet_values = file.sheet_by_name(sheet) # 获取当前sheet页内地址空间
    sheet_rows = sheet_values.nrows # 获取当前sheet行数
    sheet_cols = sheet_values.ncols # 获取当前sheet列数

    # 因为sheet长度限制在31个字符，所以当sheet过长时，需要手动设置丢失部分

    # sql字符串拼接
    sql = "insert into " + sheet + " ("
    for i in range(0, sheet_cols): # 拼接字段
        field_name = str(sheet_values.cell(0,i).value)
        if(field_name == 'desc'): # 解决自定义字段与sql关键字冲突
            field_name = '`desc`'
        if(i == (sheet_cols-1)): # 校验是否到最后一列
            sql +=  field_name + ") values ("
        else:
            sql +=  field_name + ","
    for j in range(0, sheet_cols): # 拼接占位符
        if(j == (sheet_cols-1)):
            sql +=  "%s" + ")"
        else:
            sql +=  "%s" + ","

    # 获取每行value
    for sheet_row in range(1, sheet_rows):
        values = [] # 使用列表存储
        for sheet_col in range(0, sheet_cols):
            sheet_value = str(sheet_values.cell(sheet_row,sheet_col).value)
            # 当有些日期字段必填时，可以在这里自定义
            if (str(sheet_values.cell(0,sheet_col).value) == "add_time"):
                sheet_value = '2020-05-06'
            if not sheet_value:
                sheet_value = None
            values.append(sheet_value)
        values = tuple(values) # 转换成元组
        
        cursor.execute(sql,values) # 执行sql语句插入数据
        conn.commit()
cursor.close()
conn.close()