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

使用ｅｘｃｅｌ管理用例

①、读取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;