最近想尝试着做一个接口自动化的脚本,首先得需要从文件中读取数据,很早之前更新过一篇,但是局限性较大,一次性只能读取一个Excel的一张sheet表,且读取出的格式也无法满足要求,找了好多方法,终于研究出来啦。
也遇到了很多坑,因为自学python,没有系统的学习过,基础知识也不扎实,且平时工作也没有用到,过段时间就需要从头学起 =_=,所以遇到的坑都是基础知识没掌握到的原因啦。记录一下吧,方便以后复习或知识回顾。
现在是坑:
def getTables():
for excel in os.listdir(data_path):
print(excel)
if excel.startswith('~'):
break
excelPath = os.path.join(data_path, excel)
excels = xlrd.open_workbook(excelPath)
sheetNames = excels.sheet_names()
print('sheetNames:',sheetNames)
return sheetNames #这儿我就直接把获取到的sheet名return了,这是一个list类型
def getData(sheets):
data = []
for i in range(2, sheets.nrows):#写这儿的时候,sheets无法自动带出nrows,强行写上去也会报错
data.append(dict(zip(sheets.row_values(1), sheets.row_values(i))))
print('读取出的data:',data)
getData(getTables())
既然list没有这个方法,len()也可以获取所有的行数啊,所以我就换成了len(),这样:
def getData(sheets):
data = []
for i in range(2, len(sheets)):
data.append(dict(zip(sheets.row_values(1), sheets.row_values(i))))
print('读取出的data:',data)
然而还是不行:
于是再回头看了下之前写的test,恍然大悟:
附上正确的代码
import os,xlrd
data_path = os.path.join('F:\pythonProjrect\itf_simple','data')#os.path.join()函数:连接两个或更多的路径名
print(data_path)
#获取目录下所有Excel中sheet
def getTables():
for excel in os.listdir(data_path):#os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表
print(excel)
if excel.startswith('~'):
break
excelPath = os.path.join(data_path, excel)#拼接路径
excels = xlrd.open_workbook(excelPath)
sheetNames = excels.sheet_names()#获取目录下所有sheet的名字,sheetNames为list类型
print('sheetNames:',sheetNames)
# 循环从sheetNames中读取sheet表,返回的是sheet的内存地址
for sheet in sheetNames:
table = excels.sheet_by_name(sheet)
print('table:',table)
return table
#从getTables()中返回的table中提取数据,sheet中的数据组成一个列表,Excel中的表头和数据分别组成key:value的格式
def getData(sheets):
data = []
for i in range(2, sheets.nrows):
data.append(dict(zip(sheets.row_values(1), sheets.row_values(i))))
print('读取出的data:',data)
getData(getTables())
这样就运行出来啦~