python实现excel的覆盖写入和追加

背景：需要生成类似自动化case格式的那种excel表格

覆盖式写入：

献上代码：

import xlrd
import openpyxl
from xlutils.copy import copy


#覆盖式写入，定义覆盖写入excel函数
def WriteExcel(path, sheet_name):
    '''
   #     :param sheet_name: 需要改写的sheet_name
   #     :param path: 工作薄的路径
   #     :return:
   #     '''
    workbook = openpyxl.Workbook()
    sheet = workbook.active
    sheet.title = sheet_name
    k_list = []
    v_list = []
    for k,v in value_dict.items():
        k_list.append(k)
        v_list.append(v)
        for i in range(0, len(k_list)):
            sheet.cell(row=1, column=i + 1, value=k_list[i])
            for j in range(0, len(v_list)):
                sheet.cell(row=i + 2, column=j + 1, value=v_list[j])

    workbook.save(path)
    print("xlsx格式表格【覆盖】写入数据成功!")


if __name__ == '__main__':

    #定义工作薄的路径
    path = '/Users/dongyue/Desktop/code/leecode/xlsx格式demo.xlsx'
    #定义要写入的行和列的值
    value_dict = {
        "请求方法":"post",
        "请求参数":"测试参数",
        "返回参数":"测试参数",
        "预期结果":"successful",
        "实际结果":"successful",
    }
    #定义excel的sheet_name  "xlsx格式测试表 "
    sheet_name = "xlsx格式测试表"


    WriteExcel(path,sheet_name)

执行结果是：

excel中的显示是：

这个在第二次执行的时候会覆盖原来的内容，所以这种是全部覆盖式的写入。

 

追加式写入：

献上代码：

import xlrd
import openpyxl
from xlutils.copy import copy


#append写入，定义追加写入excel函数
def WriteExcelAppend(path):
    '''
    :param line_number: 行数
    :param path: 工作薄的路径
    :return:
    '''
    workbook = xlrd.open_workbook(path)    #打开工作薄
    sheets_name = workbook.sheet_names()   #获取工作薄的所有的sheet名称
    worksheet  = workbook.sheet_by_name(sheets_name[0]) #获取工作薄中的第一个sheet
    rows_exists = worksheet.nrows          #获取已经存在的数据行数
    new_workbook = copy(workbook)          #将xlrd对象拷贝转化为xlwt对象
    new_worksheet = new_workbook.get_sheet(0)  #获取转化后的第一个表格
    v_list = []
    for k, v in value_dict.items():
        v_list.append(v)
        for j in range(0, len(v_list)):
            new_worksheet.write(rows_exists, j, v_list[j])
    new_workbook.save(path)  # 保存工作簿
    print("xls格式表格【追加】写入数据成功！")



if __name__ == '__main__':

    #定义工作薄的路径
    path = '/Users/dongyue/Desktop/code/leecode/xlsx格式demo.xlsx'
    #定义要写入的行和列的值
    value_dict = {
        "请求方法":"post",
        "请求参数":"测试参数",
        "返回参数":"测试参数",
        "预期结果":"successful",
        "实际结果":"successful",
    }
    #定义excel的sheet_name  "xlsx格式测试表 "
    sheet_name = "xlsx格式测试表"
    
    WriteExcelAppend(path)

在执行之前我在excel中备注下：

希望达到的预期是在这行之后进行追加

执行结果为：

excel中的展示

我在追加的时候只追加了一条数据，如果小伙伴们想追加多条，可以进行循环。这种的数据是我们自行写死的，在测试过程中的话，是需要拉取各种参数，然后进行写入的，我们可以获取各种参数后，给他处理成dict形式。

还有就是如果想自行的让代码看这个操作是执行追加还是覆盖的话。可以先获取下，已经存在的行数（上面的代码中有），如果值等于0，就说明该表中没有数据，可以执行覆盖式的写入，如果值大于0，就需要进行追加式写入。

现在的excel表格是这样的：

将下方代码进行更改

if __name__ == '__main__':

    #定义工作薄的路径
    path = '/Users/dongyue/Desktop/code/leecode/xlsx格式demo.xlsx'
    #定义要写入的行和列的值
    value_dict = {
        "请求方法":"post",
        "请求参数":"测试参数",
        "返回参数":"测试参数",
        "预期结果":"successful",
        "实际结果":"successful",
    }
    #定义excel的sheet_name  "xlsx格式测试表 "
    sheet_name = "xlsx格式测试表"


    workbook = xlrd.open_workbook(path)
    sheets_name = workbook.sheet_names()
    worksheet = workbook.sheet_by_name(sheets_name[0])
    rows_exists = worksheet.nrows
    if rows_exists == 0:
       WriteExcel(path,sheet_name)
    else:
        WriteExcelAppend(path)

其中取出的是第一个sheet，此时是有数据，这种情况下执行应该是进行追加。执行结果如下：

excel中展示：

如果将worksheet = workbook.sheet_by_name(sheets_name[0]) 后面的sheets_name[0]改为sheets_name[1]，此时获取的就是sheet_name为ceshi的那个表格，这个表格是没有数据的。所以我们此时执行就是覆盖写入，并且之前的sheet是没有了的。

执行结果如下：

excel展示：