18.4 操作 excel 文件
Python 中一般使用 xlrd 库来读取 Excel 文件, xlrd 库是 Python 的第三方库。
18.4.1 xlrd 库安装
Xlrd 库跟其他第三方库一样,都是通过 pip install xlrd 命令来安装。
安装成功之后,在 C:Python34Libsite-packages 下可以看到相应的Xlrd 库目录。
以下是xlrd 库下的相应模块方法
备注:喜欢研究的同学,可以去研究 xlrd 库下的相应模块的实现原理。
18.4.2 在 Python3 下打开excel文件,获取一个Book()对象
例如:读取 result.xls 的数据。
程序实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
result_file =xlrd.open_workbook("./result.xls")
18.4.3 获取 sheets 数目
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("./result.xls")
print(file.nsheets)
运行结果:
18.4.4 获取 sheets 列表
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("./result.xls")
print(file.sheets())
运行结果:
18.4.5 获取 sheets name 列表
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("./result.xls")
print(file.sheet_names())
运行结果:
18.4.6 获取文件中的 Sheet
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("./result.xls")
print(file.sheets())
sheet1 = file.sheets()[0] #sheets返回一个sheet列表
sheet2 = file.sheet_by_index(0) #通过索引顺序获取
sheet3 = file.sheet_by_name('保单查询结果清单') #通过名称获取
18.4.7 获取行数,列数
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("result.xls")
# 获取当前文件的表
shxrange = range(file.nsheets)
try:
sh = file.sheet_by_name("保单查询结果清单")
except:
print("no sheet in %s named '保单查询结果清单'",format("result.xls"))
# 获取行数
nrows =sh.nrows
# 获取列数
ncols = sh.ncols
#打印表格的行数和列数
print("nrows{0},ncols{1}".format(nrows,ncols))
运行结果:
18.4.8 获取某行,某行值列表,某列,某列值列表
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("result.xls")
# 获取当前文件的表
shxrange = range(file.nsheets)
try:
sh = file.sheet_by_name("保单查询结果清单")
print(sh)
except:
print("no sheet in %s named '保单查询结果清单'",format("result.xls"))
#获取第一行
print(sh.row(1))
#获取第一行值列表
print(sh.row_values(1))
#获取第一列
print(sh.col(1))
#获取第一列值列表
print(sh.col_values(1))
运行结果:
18.4.9 获取单元格的值
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("result.xls")
# 获取当前文件的表
shxrange = range(file.nsheets)
try:
sh = file.sheet_by_name("保单查询结果清单")
print(sh)
except:
print("no sheet in %s named '保单查询结果清单'",format("result.xls"))
#获取单元格的值
cell = sh.cell(1,1)
print(cell)
cell_value1 = sh.cell_value(1,1)
print(cell_value1)
cell_value2 = sh.cell(1,1).value
print(cell_value2)
运行结果:
注意:用xlrd读取excel是不能对其进行操作的:xlrd.open_workbook()方法返回xlrd.Book类型,是只读的,不能对其进行操作。
18.4.10 读取 Excel 应用案例
案例1:获取result.xls文件下的保单查询结果清单表数据
代码实现:
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("result.xls")
# 获取当前文件的表
shxrange = range(file.nsheets)
try:
sh = file.sheet_by_name("保单查询结果清单")
except:
print("no sheet in %s named '保单查询结果清单'",format("result.xls"))
# 获取行数
nrows =sh.nrows
# 获取列数
ncols = sh.ncols
#打印表格的行数和列数
print("nrows{0},ncols{1}".format(nrows,ncols))
#获取第二行第一列数据(第一行是标题)
cell_value = sh.cell_value(1,0)
print(cell_value)
# 新建一个空的列表
row_list =[]
for i in range(0,nrows):
row_data =sh.row_values(i)
row_list.append(row_data)
# 输出数据
print(row_list)
运行结果:
案例2:xlrd读取Excel中的日期。
#导入 xlrd 库
import xlrd
# 打开 Excel 读取文件,open_workbook()为打开 Excel文件的方法,参数为:文件名
file =xlrd.open_workbook("./result.xls")
# 获取当前文件的表
shxrange = range(file.nsheets)
try:
sh = file.sheet_by_name("保单查询结果清单")
except:
print("no sheet in %s named '保单查询结果清单'",format("result.xls"))
#获取单元格的值
cell = sh.cell(2,5)
print(cell)
cell_value1 = sh.cell_value(2,5)
print(cell_value1)
运行结果:
18.4.11 xlwt 库安装
Xlrd 库只读,如果想新建一个test.xls文件,就需要安装 xlwt 库,跟其他第三方库一样,都是通过 pip install xlwt 命令来安装。
安装成功之后,在 C:Python34Libsite-packages 下可以看到相应的 xlwt 库目录。
18.4.12 创建一个Excel文件并创建一个Sheet
程序实现:
from xlwt import *
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
book.save('myExcel.xls')
运行结果:
Workbook 类可以有 encoding 和 style_compression 参数。
【encoding】:设置字符编码,
【style_compression】:表示是否压缩。
这样设置:w = Workbook(encoding='utf-8'),就可以在excel中输出中文了。默认是ascii。
18.4.13 向sheet写入内容
格式:sheet.write(r, c, label="", style=Style.default_style)
(1)简单写入。
程序实现:
from xlwt import *
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
# 在sheet1 中简单插入内容
sheet.write(0, 0, label = 'Row 0, Column 0 Value')
# 保存内容
book.save('test1.xls')
运行结果:
(2)设置格式写入。
代码实现:
from xlwt import *
import xlwt
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
# 设置格式
font = xlwt.Font() # 字体
font.name = 'Times New Roman'
font.bold = True
font.underline = True
font.italic = True
style = xlwt.XFStyle() # 创建一个格式
style.font = font # 设置格式字体
# 在sheet1 中插入格式化的内容
sheet.write(1, 0, label = 'Formatted value', style) # Apply the Style to the Cell
# 保存内容
book.save('test2.xls')
备注:这个例子,需要运行的机器有对应的字体。
(3)写入日期。
程序实现:
from xlwt import *
import xlwt
import time
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
style = xlwt.XFStyle()
#写入当前时间
time = time.strftime("%Y%m%d%H%M%S", time.localtime())
style.num_format_str = 'M/D/YY' # Other options: D-MMM-YY, D-MMM, MMM-YY, h:mm, h:mm:ss, h:mm, h:mm:ss, M/D/YY h:mm, mm:ss, [h]:mm:ss, mm:ss.0
sheet.write(0, 0, time, style)
# 保存内容
book.save('test3.xls')
运行结果:
(4)写入公式。
程序实现:
from xlwt import *
import xlwt
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
#写入公式
sheet.write(0, 0, 5) # Outputs 5
sheet.write(0, 1, 2) # Outputs 2
sheet.write(1, 0, xlwt.Formula('A1*B1')) # 输出 "10" (A1[5] * A2[2])
sheet.write(1, 1, xlwt.Formula('SUM(A1,B1)')) # 输出 "7" (A1[5] + A2[2])
# 保存内容
book.save('test4.xls')
运行结果:
(5)写入链接。
程序实现:
from xlwt import *
import xlwt
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
#写入链接
sheet.write(0, 0, xlwt.Formula('HYPERLINK("http://www.google.com";"Google")')) #输出 "Google"链接到http://www.google.com
# 保存内容
book.save('test5.xls')
运行结果:
18.4.14 xlutils 库安装
如果想修改一个test.xls文件内容,就需要安装 xlutils 库,跟其他第三方库一样,也是通过 pip install xlutils 命令来安装。
安装成功之后,在 C:Python34Libsite-packages 下可以看到相应的 xlutils 库目录。
18.4.15 修改Excel表内容
原理:复制一个 xlrd.Book 对象,生成一个 xlwt.Workbook 对象,可以对 xlwt.Workbook 进行修改。
例子:
第一步:先新建一个test1.xls文件,在里面写入4条数据。
程序实现:
from xlwt import *
from xlrd import open_workbook
from xlutils.copy import copy
#新建一个 test1.xls 的文件,并在新建的文件里写入内容。
book = Workbook(encoding='utf-8')
sheet = book.add_sheet('Sheet1')
sheet.write(0, 0, label = 'name')
sheet.write(0, 1, label = 'age')
sheet.write(1, 0, label = 'xiao')
sheet.write(1, 1, label = '18')
book.save('test1.xls')
运行结果:
第二步:修改test1.xls文件内容。
程序实现:
from xlwt import *
from xlrd import open_workbook
from xlutils.copy import copy
#第二步:打开文件,修改文件内容
book = open_workbook('test1.xls')
wb = copy(book) #wb即为xlwt.WorkBook对象
ws = wb.get_sheet(0) #通过get_sheet()获取的sheet有write()方法
ws.write(1, 0, '张三')
ws.write(1, 1, '28')
wb.save('test1.xls')
运行结果:
18.4.16 Python3下接口数据驱动之Excel案例
教育局招生管理系统登录接口用例参数化实现步骤:
(一)编写完整的接口测试用例
1、打开抓包工具:fiddler。
2、登录教育局招生管理系统。
3、抓取登录http请求。
4、分析登录http请求(请求地址、是否重定向、get请求还是post请求、请求的头信息、请求的response)。
5、数据的处理(处理抓取到的头信息)
6、编写接口代码。
7、接口测试结果断言验证。
第一步:下面对fiddler抓取到的数据进行分析。
【请求方法】:带参数的get
【请求地址】:http://127.0.0.1:8090/recruit.students/login/in?account=admin&pwd=660B8D2D5359FF6F94F8D3345698F88C
【请求头信息】:
Host: 127.0.0.1:8090
Connection: keep-alive
Upgrade-Insecure-Requests: 1
User-Agent: Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36
Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8
Referer: http://127.0.0.1:8090/recruit.students/login/view
Accept-Encoding: gzip, deflate, br
Accept-Language: zh-CN,zh;q=0.9
Cookie: JSESSIONID=AED5BBB1B5E4F9BEDCC3A0FC88668871; userInfoCookie=""
【请求的response】:空
请求的 response 为空是因为登录的时候,做了跳转,状态码为:302,跳转到了
http://127.0.0.1:8090/recruit.students/school/manage/index 这个地址,这个状态码为200。
查看http://127.0.0.1:8090/recruit.students/school/manage/index 这个地址请求的response,返回的是登陆后的信息。
通过分析,我们是清楚的了解到这个接口的情况。
第二步:接着还需要请求头信息的处理,去掉一些没用的请求头信息,保留如下:
"Connection": "keep-alive",
"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36",
"Referer": "http://127.0.0.1:8090/recruit.students/login/view",
【Connection】:如果只是测试登录接口,这个参数可以去掉,如果需要测试登录之后新建学校,那这个头信息就需要保留。
【User-Agent】:模拟用户利用浏览器访问Web网站的真实行为,每个接口都需要。
【Referer】:登录重定向的时候用到。
第三步:最后编写代码实现。
程序实现:
import requests
url="http://127.0.0.1:8090/recruit.students/login/in?"
#把请求头信息进行处理,去掉一些没用的,保留一些有用头信息·
headers = {
"Connection": "keep-alive",
"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36",
"Referer": "http://127.0.0.1:8090/recruit.students/login/view",
}
# URL参数
payload = {'account': 'admin','pwd':'660B8D2D5359FF6F94F8D3345698F88C'}
# 发送Post请求
response = requests.post(url,headers = headers,data=payload)
# 查看响应内容,response.text 返回的是Unicode格式的数据
print(response.text)
# 查看响应码
print(response.status_code)
(二)新建一个data.xls文件,组织测试用例参数。
(三)编写一个readExcel方法。
程序实现:
import requests
import xlrd
def readExcel(rowx, filePath='data.xls'):
'''
读取excel中数据并且返回
:parameter filePath:xlsx文件名称
:parameter rowx:在excel中的行数
'''
book = xlrd.open_workbook(filePath)
sheet = book.sheet_by_index(0)
return sheet.row_values(rowx)
#提取第一个测试用例数据
print("第一行数据内容:",readExcel(2))
#查看数据的类型
print("数据类型:{0}:".format(type(readExcel(2))))
# 从列表中提取url
url = readExcel(2)[3]
print(url)
# 从列表中提取data参数
data = readExcel(2)[4]
print(data)
运行结果:
(四)测试用例数据参数化。
完整程序:
import requests
from bs4 import BeautifulSoup
import xlrd
import json
def readExcel(rowx, filePath='data.xls'):
'''
读取excel中数据并且返回
:parameter filePath:xlsx文件名称
:parameter rowx:在excel中的行数
'''
book = xlrd.open_workbook(filePath)
sheet = book.sheet_by_index(0)
return sheet.row_values(rowx)
#提取第一个测试用例数据
print("第一行数据内容:",readExcel(2))
#查看数据的类型
print("数据类型:{0}:".format(type(readExcel(2))))
# 从列表中提取url
url = readExcel(2)[3]
print(url)
# 从列表中提取data参数
# 由于JSON中,标准语法中,不支持单引号,属性或者属性值,都必须是双引号括起来,用字符串方法replace("'",'"')进行处理
data1 =readExcel(2)[4].replace("'",'"')
#因为请求参数数据类型是字典,所以进行了反序列化的处理。
data = json.loads(data1)
print(data)
print(type(data))
print("---------------------------------------------")
#招生系统接口例子
headers = {
"Connection": "keep-alive",
"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36",
"Referer": "http://127.0.0.1:8090/recruit.students/login/view",
}
# URL参数
payload = data
# 发送get请求
response = requests.get(url,headers = headers,params=payload)
# 打印请求response后的URL
print(response.url)
# 查看响应内容,response.text 返回的是Unicode格式的数据
print(response.text)
# 查看响应码
print(response.status_code)运行结果:
运行结果:
提取Excel表数据注意的几点:
1、需要明确提取excel 表中那个sheet的数据。
2、一个测试用例按行去提取。
3、从Excel表提取的行数据,是一个列表,列表的元素是字符串,如果接口参数类型如果是字典类型,需要反序列化的处理(json.loads())。
4. JSON中,标准语法中,不支持单引号,属性或者属性值,都必须是双引号括起来,用字符串方法replace("'",'"')进行处理。
18.4.17 接口数据驱动之Excel案例代码结构的优化(同个文件)
上面已经实现了通过读取Excel表数据进行参数化,大家可能会说,代码看起来很乱!确实是有点乱,接下来我们对上面的代码进行优化。
第一步:定义一个读取 Excel 表的 readExcel() 方法,参数:rowx 。
def readExcel(rowx):
'''
读取excel中数据并且返回
:parameter filePath:xlsx文件名称
:parameter rowx:在excel中的行数
'''
book = xlrd.open_workbook('data.xls')
sheet = book.sheet_by_index(0)
return sheet.row_values(rowx)
第二步:定义一个获取请求 URL 的方法 getUrl(),参数:rowx 。
def getUrl(rowx):
'''
获取请求URL
:parameter rowx:在excel中的行数
'''
return readExcel(rowx)[3]
第三步:定义一个获取请求参数的方法 getData(),参数:rowx 。
def getData(rowx):
'''
获取请求参数
:parameter rowx:在excel中的行数
'''
# 由于JSON中,标准语法中,不支持单引号,属性或者属性值,都必须是双引号括起来,用字符串方法replace("'",'"')进行处理
data1 = readExcel(rowx)[4].replace("'", '"')
# 因为请求参数数据类型是字典,所以进行了反序列化的处理。
data = json.loads(data1)
return data
第四步:用 unittest 单元测试框架编写测试用例。
完整的程序如下:
import requests
from bs4 import BeautifulSoup
import xlrd
import json
import unittest
def readExcel(rowx):
'''
读取excel中数据并且返回
:parameter filePath:xlsx文件名称
:parameter rowx:在excel中的行数
'''
book = xlrd.open_workbook('data.xls')
sheet = book.sheet_by_index(0)
return sheet.row_values(rowx)
def getUrl(rowx):
'''
获取请求URL
:parameter rowx:在excel中的行数
'''
return readExcel(rowx)[3]
def getData(rowx):
'''
获取请求参数
:parameter rowx:在excel中的行数
'''
# 由于JSON中,标准语法中,不支持单引号,属性或者属性值,都必须是双引号括起来,用字符串方法replace("'",'"')进行处理
data1 = readExcel(rowx)[4].replace("'", '"')
# 因为请求参数数据类型是字典,所以进行了反序列化的处理。
data = json.loads(data1)
return data
class Test(unittest.TestCase):
def setup(self):
print("------开始执行用例--------")
def teardown(self):
print("------用例执行结束--------")
def test_case1(self):
url = getUrl(2)
payload = getData(2)
headers = {
"Connection": "keep-alive",
"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36",
"Referer": "http://127.0.0.1:8090/recruit.students/login/view",
}
# 发送get请求
response = requests.get(url,headers=headers,params=payload)
# 打印请求response后的URL
print(response.url)
# 查看响应内容,response.text 返回的是Unicode格式的数据
print(response.text)
# 查看响应码
print(response.status_code)
if __name__ == '__main__':
unittest.main(verbosity=2)
运行结果:
18.4.18 接口数据驱动之Excel案例代码结构的继续优化(多个文件)
通过上面的优化之后,可能有些人还是觉得一个文件代码太多,层级不明显,不好维护,那好,我们接着继续优化。
第一步,在当前项目目录下新建一个 Data-Driven 文件夹,把数据文件 data.xls 放到这个目录下,同时,还需要新建一个 __init__.py 的空白文件(该文件的主要作用是初始化Python包)。在__init__.py文件中编写内容
from Data_Driven.data_driven import Data_Excel
第二步:在 Data-Driven 文件夹下,新建一个 data_driven.py 文件,在文件中编写代码。
import xlrd
import json
class Data_Excel(object):
def __init__(self):
pass
def readExcel(self,rowx):
'''
读取excel中数据并且返回
:parameter filePath:xlsx文件名称
:parameter rowx:在excel中的行数
'''
book = xlrd.open_workbook('data.xls')
sheet = book.sheet_by_index(0)
return sheet.row_values(rowx)
def getUrl(self,rowx):
'''
获取请求URL
:parameter rowx:在excel中的行数
'''
return self.readExcel(rowx)[3]
def getData(self,rowx):
'''
获取请求参数
:parameter rowx:在excel中的行数
'''
# 由于JSON中,标准语法中,不支持单引号,属性或者属性值,都必须是双引号括起来,用字符串方法replace("'",'"')进行处理
data1 = self.readExcel(rowx)[4].replace("'", '"')
# 因为请求参数数据类型是字典,所以进行了反序列化的处理。
data = json.loads(data1)
return data
if __name__ == "__main__":
#定义一个类对象
t= Data_Excel()
url = t.getUrl(2)
print(url)
payload = t.getData(2)
print(payload)
第三步:编写我们的测试用例。
import os,sys,requests,unittest
sys.path.append("./Data_Driven")
from data_driven import Data_Excel
class Test(unittest.TestCase,Data_Excel):
def setup(self):
print("------开始执行用例--------")
def teardown(self):
print("------用例执行结束--------")
def test_case1(self):
t = Data_Excel()
url = t.getUrl(2)
payload = t.getData(2)
headers = {
"Connection": "keep-alive",
"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36",
"Referer": "http://127.0.0.1:8090/recruit.students/login/view",
}
# 发送get请求
response = requests.get(url,headers=headers,params=payload)
# 打印请求response后的URL
print(response.url)
# 查看响应内容,response.text 返回的是Unicode格式的数据
print(response.text)
# 查看响应码
print(response.status_code)
if __name__ == '__main__':
unittest.main(verbosity=2)
运行结果:
18.4.19 python跨路径调用的三种方法
假设路径结构为:
project
mode
count.py
data.py
test.py
count.py文件里有A类,data.py里有B类,现在想要在test.py中调用count.py和data.py里的方法 。
方法一:
import sys
sys.path.append(“./mode”)
from count import A
from data import B
之后在test.py中直接使用 类名(),方法名() 。
方法二:(推荐)
在 mode 中新建 init.py,内容为空 。
之后在test.py中 。
from model.count import A
from model.new_count import B
之后在test.py中直接使用 类名(),方法名() 。
方法三:
在 model中新建 init.py,内容为 :
from count import A
from new_count import B
之后在test.py中 。
import mode
调用时,mode.类名(),mode.方法名() 。
或
from model import *
调用时,直接使用 类名(),方法名()。
--------------------------
个人今日头条账号: 听海8 (上面上传了很多相关学习的视频以及我书里的文章,大家想看视频,可以关注我的今日头条)