python selenium - 利用excel实现参数化

前言

在进行软件测试或设计自动化测试框架时，一个比可避免的过程就是: 参数化，在利用python进行自动化测试开发时，通常会使用excel来做数据管理，利用xlrd、xlwt开源包来读写excel。

环境安装

首先在命令行下安装xlrd、xlwt

pip install xlrd

pip install xlwt

一个简单的读写示例

让我们先看一个简单的excel读写示例，示例代码功能，从表1中读取数据。

excel数据表如图所示

demo.xlsx

1、读取代码示例

#_*_ coding:utf-8 _*_

__author__ = '苦叶子'

import xlrd

if __name__ == '__main__':

    # excel文件全路径

    xlPath = "C:\\Users\\lyy\\Desktop\\demo.xlsx"

    # 用于读取excel

    xlBook = xlrd.open_workbook(xlPath)

    # 获取excel工作簿数

    count = len(xlBook.sheets())

    print u"工作簿数为:  ", count

    # 获取 表 数据的行列数

    table = xlBook.sheets()[0]

    nrows = table.nrows

    ncols = table.ncols

    print u"表数据行列为(%d, %d)" % (nrows, ncols)

    # 循环读取数据

    for i in xrange(0, nrows):

        rowValues = table.row_values(i) # 按行读取数据

        # 输出读取的数据

        for data in rowValues:

             print data, "    ",

        print ""

运行效果如图

运行结果

2、写excel示例代码

#_*_ coding:utf-8 _*_

__author__ = '苦叶子'

import xlwt

import random

if __name__ == '__main__':

    # 注意这里的excel文件的后缀是xls 如果是xlsx打开是会提示无效

    newPath =  unicode("C:\\Users\\lyy\\Desktop\\demo_new.xls", "utf8")

    wtBook = xlwt.Workbook()

    # 新增一个sheet

    sheet = wtBook.add_sheet("sheet1", cell_overwrite_ok=True)

    # 写入数据头

    headList = [u'序号', u'数据1', u'数据2', u'数据3']

    rowIndex = 0

    col = 0

    # 循环写

    for head in headList:

         sheet.write(rowIndex, col, head)

         col =  col + 1

     # 写入10行 0-99的随机数据

     for index in xrange(1, 11):

           for col in xrange(1, 4):

               data = random.randint(0,99)

               sheet.write(index, 0, index)  # 写序号

               sheet.write(index, col, data) # 写数据

          print u"写第[%d]行数据" % index

     # 保存

    wtBook.save(newPath)

运行结果如图

写excel结果

结束语

这里只是简单的对xlrd、xlwt模块的应用演示，对于实际做自动化测试过程中，需要封装一个通用的excel解析类，以便提高复用性和降低维护成本。

在实际应用中，我们通常需要对表格进行合并、样式设置等等系列动作，请参考官方文档，进行更深入的学习研究

python excel官网： http://www.python-excel.org/