python---读取/写入excel用例数据

2022-06-26 18:50:24

使用ｅｘｃｅｌ管理用例

①、读取excel的用例数据，每一类参数保存在list中返回；
②、那么接下来使用unitest编写用例时，可以去调用这个函数。使用里面的数据；

个人认为好处是，大部分人还是习惯excel的编写用例习惯，将实际用例与代码分离，有助于分工和维护

一、在excel中写好用例后，在python(unitest)中调用

 #写文件

 import xlwt;

 #读文件

 import xlrd;

 #若要对excel文件进行修改必须！要引入这个工具模块，其中含有复制和修改功能---------注意引入方式

 import xlutils;

 from xltuils.copy import copy;

 import unittest;

 import requests;

 #使用写文件模式打开文件

 open_workbook=xlrd.open_workbook(r'D:\Users\4399-3046\Desktop\test.xls');

 #最终返回的是文件类型

 print(open_workbook);

 #打印该文件下，对应数据表的行数

 #print(open_workbook.sheet_by_name('mytest').nrows);

 #打印该文件下，对应数据表的列数

 #print(open_workbook.sheet_by_name('mytest').ncols);

 #for item_rows in range(1,open_workbook.sheet_by_name('mytest').nrows):

     #for item_cols in range(1,open_workbook.sheet_by_name('mytest').ncols):

         #print(open_workbook.sheet_by_name('mytest').cell(item_rows,item_cols));

 #循环打印出所有数据,并分别保存在list中

 #编号

 listkey=[];

 #测试地址

 listurl=[];

 #提交方式

 listtype=[];

 #参数1

 listcanshu1=[];

 #参数2

 listcanshu2=[];

 #预期结果

 listyuqi=[];

 #实际结果

 listresult=[];

 #在unitest框架中，通过调用参数，执行用例，将实际结果比对预期结果是否一致，若一致，则通过，否则用例失败，并将实际结果写入excel中（写入代码后文将写到）

 for item_row in range(1,open_workbook.sheet_by_name('mytest').nrows):

     #获取用例编号

     listkey.append(open_workbook.sheet_by_name('mytest').cell(item_row,0).value);

     # 获取参数1

     listcanshu1.append(open_workbook.sheet_by_name('mytest').cell(item_row, 1).value);

     # 获取参数2

     listcanshu2.append(open_workbook.sheet_by_name('mytest').cell(item_row, 2).calue);

     # 获取预期结果

     listyuqi.append(open_workbook.sheet_by_name('mytest').cell(item_row, 3).value);

     # 获取实际结果

     listresult.append(open_workbook.sheet_by_name('mytest').cell(item_row, 4).value);------------->注意：必须要有【.value】，因为返回的是dict字典，而实际我们只需要其中的具体值

 print(listkey);

 print(listcanshu1);

 print(listcanshu2);

 print(listyuqi);

 print(listresult);

运行结果：

2、新建文件，并在新文件中写入数据

  open_workbook=xlwt.Workbook();---定义一个工作簿

 new_sheet=open_workbook.add_sheet('lianxi',cell_overwrite_ok = True);----在工作簿中新增一个工作表，自定义一个工作表名称

注意1：若当前单元格已经有值，无法重复写入覆盖原有，会提示异常，需要在新建表单sheet时，就要指定，可以重复写入【cell_overwrite_ok = True】，默认是false

 new_sheet.write(0,5,'lianxi_test');-----在工作表中写入数据（注意：写入数据只在工作表sheet中生效）---

注意2：如果当前目录下已存在同名的工作簿名称，则会报错

 open_workbook.save(r'D:\Users\Desktop\test01.xls');----保存工作簿（需要指定地址和名称）

3、在已有文件下excel下，写入数据

1#思路：打开文件----复制文件----在文件中写入数据----删除原文件----保存新文件 

#打开当前已经存在的工作簿

 open_workbook2=xlrd.open_workbook(r'D:\Users\Desktop\test.xls');

 #复制当前工作簿

 open_sheet2=copy(open_workbook2);
#打开置顶索引的工作表

 open_ws=open_sheet2.get_sheet(3);

 #在指定的单元格写入数据

 open_ws.write(1,4,'TEST');
#移除原有的文件

 os.remove(r'D:\Users\4399-3046\Desktop\test.xls')
#保存新文件

 open_sheet2.save(r'D:\Users\4399-3046\Desktop\test.xls');

运行结果：

4、扩展--结合python-unitest,通常我们需要将执行结果自动写入表格中
5、拓展--将写文件和读文件，分别封装为公共方法调用（已完成）

 #by zhonghuihong

 import xlwt;

 import xlrd;

 from xlutils.copy import copy;

 import os;

 #此接口主要用于excel数据的读取和写入

 def get_excel_data(paths,sheetName):

     open_workbook=xlrd.open_workbook(paths);

     open_sheet=open_workbook.sheet_by_name(sheetName);

     rows=open_sheet.nrows;

     listkey = [];

     listcanshu1 = [];

     listcanshu2 = [];

     listyuqi = [];

     listresult = [];

     for item_row in range(1, rows):

         listkey.append(open_sheet.cell(item_row, 0).value);

         listcanshu1.append(open_sheet.cell(item_row, 1).value);

         listcanshu2.append(open_sheet.cell(item_row, 2).value);

         listyuqi.append(open_sheet.cell(item_row, 3).value);

         listresult.append(open_sheet.cell(item_row, 4).value);

     return listkey,listcanshu1,listcanshu2,listyuqi,listresult;

 def write_excel_data(paths,sheet_id,rows_id,cols_id,values):

     open_workbook2 = xlrd.open_workbook(paths);

     open_sheet2 = copy(open_workbook2);

     open_ws = open_sheet2.get_sheet(sheet_id);

     open_ws.write(rows_id,cols_id,values);

     os.remove(paths);

     open_sheet2.save(paths);

     return 'success';


遇到的问题
1、读取的整数变成浮点类型，解决办法

2、脚本中引入copy方法，但提示TypeError: 'module' object is not callable的解决办法：将引入语句写成from xlutils.copy import copy;

3、指定sheet时（ow.get_sheet[0];），提示TypeError: 'method' object is not subscriptable

产生原因：subscriptable的意思是可有下标，错误的原因就是：对象不具有下标，即把不具有下标操作的对象用成了对象[i]，比如int对象变量[i]就会报错。仔细检查错误行。

代码最终优化：

*增加try--exception异常捕获

*增加写入输入不能为空

 #写数据（文件薄,表格id，行id，列id，要写的内容）

 def write_excel_data(paths,sheet_id,rows_id,cols_id,values):

     if(len(values)==0 ):

         return '写入数据不能为空';

     else:

       try:

           #如果写入位置已经有值， 会直接使用新内容覆盖

          open_workbook2 = xlrd.open_workbook(paths);

          open_sheet2 = copy(open_workbook2);

          open_ws = open_sheet2.get_sheet(sheet_id);

          open_ws.write(rows_id, cols_id, values);

          os.remove(paths);

          open_sheet2.save(paths);

          return 'success';

       except Exception as e:

          return e;

 #读取excel_data（文件薄地址，表格名）
   #通过循环读出指定表格中的所有数据，以列表格式输出

 def get_excel_data(paths,sheetName):

         open_workbook = xlrd.open_workbook(paths);

         open_sheet = open_workbook.sheet_by_name(sheetName);

         if open_sheet.nrows <= 1:

             return '文件总行数小于1，请确认';

         else:

             res = []

             j = 1

             for i in list(range(open_sheet.nrows-1)):

                 s = {};

                 #行编号

                 s['rowNum'] = i+1;

                 values = open_sheet.row_values(j)

                 for x in list(range(open_sheet.ncols)):

                     keys=open_sheet.row_values(0);

                     s[keys[x]] = values[x];

                 res.append(s);

                 j += 1;

             return res;

码农公寓

相关文章