Python实现批量导入xlsx数据1000条

遇到的问题：

用户批量导入数据1000条，导入不成功的问题，提示查询不到商品资料。这个场景需要依靠批量的数据，每次测试的时候需要手动生成批量的数据，然后再导入操作，费时费劲。所以写了个脚本来实现。

在前面有说过用Python实现导入接口请求，现在来说一下xlsx中1000条自动生成：

xlsx数据离不开openpyxl库，openpyxl库有个Workbook对象，其代表一个Excel文档，Workbook中有一张或多张sheet，一个sheet就是一个Worksheet对象。

首先需要导包，from openpyxl import load_workbook，from openpyxl import Workbook。

xlsx文件的新增：

from openpyxl import load_workbook
import xlsxwriter

file_name = "max_1000.xlsx"
# 新建xlsx文件
workbook = xlsxwriter.Workbook(file_name)
worksheet = workbook.add_worksheet('Sheet1')
worksheet.write_row("A1",
                    ['公司', '编码', '渠道'])
#新增1000条数据
while m < 1000:
    m += 1
    # ["18"]、["21"]、["8"]分别代表第m行第1，2，3列的值，这里是写死的数据，也可以是参数化的数据
    worksheet.write_row(m, 0, ["18"])
    worksheet.write_row(m, 1, ["21"])
    worksheet.write_row(m, 2, ["8"])
# 关闭文件
workbook.close()
# 执行批量导入
data = load_yaml('./max_1000.yaml')['upload_file']
execute(data)

xlsx文件的读取：

通过Cell对象读取每一格内容

wb = load_workbook('max_1000.xlsx')
sheets = wb.worksheets
# 获取第一个sheet
sheet1 = sheets[0]
# 获取sheet1表第2行第1列的数据
cell_21 = sheet1.cell(2,1).value
# 获取sheet1表第3行第2列的数据
cell_32 = sheet1.cell(3,2).value

xlsx文件新增后，可用于批量导入的数据，xlsx文件的读取可用于数据的清理，比如已导入数据的清除。

也可用for循环，读取xlsx文件的所有数据。

xlsx文件读取所有数据：

wb = load_workbook("max_1000.xlsx")
sheets = wb.worksheets
Sheet1 = sheets[0]
# 循环读取数据，Sheet1.max_row代表表格的最大行数
for i in range(Sheet1.max_row - 1):
    # 获取Sheet1表中所有行第4列的数据
    data = Sheet1.cell(i + 2, 4).value

xlsx文件的编辑：

wb = load_workbook(file_path)

ws = wb[sheet]

# 修改.xlsx文件中的数据，cell[0]对应行数, cell[1]对应列数，cell[-1]对应需要修改后的值

ws.cell(cell[0], cell[1]).value = cell[-1]

wb.save(file_path)

wb.close()