Selenium + Python 自动化测试19(补充-读取各种文件数据操作)

我们的目标是:按照这一套资料学习下来,大家可以独立完成自动化测试的任务。

上一篇我们讨论了数据驱动测试中如何完成重复的测试实例,今天我们补充一些读取各种文件的方法。

本篇文章我们讨论一下如何使用读取txt、CSV、Excel文件,Excel文件之前文章已经大概说明了,这里是做一个总结。

1、txt 文件的读取

txt是我们经常需要操作的文本类型,可以由如下方法操作:

1)read():读取整个文件

2)readline():读取一行数据

3)readline()分行读取全部的数据

举例说明,如下图所示的user_password.txt:

复制代码
#coding = utf-8

user_password =open("user_password.txt","r")  #打开文件

file_lines =user_password.readlines()  #读取文件

user_password.close()#关闭


for line infile_lines:#打印读取的文件

    print(line)

查看结果:

小练习,想将用户名和密码分开的话,用split进行分割:

复制代码
#用户名和密码分开

user=line.split(",")[0]

password=line.split(",")[1]

print("用户名为%s,密码是:%s"%(user,password))

2、CSV文件读取

若是还是之前的数据,在CSV文件中存储的。

内容如下:

读取csv方法:导入csv模块>reader()读取>输出信息查看

复制代码
#coding = utf-8

import csv  #导入csv 包

user_file =csv.reader(open('user_password1.csv','r'))  #打开文档

#输出信息

for user_info inuser_file:print(user_info)

执行后输出结果:

'xiaoming', '123456'

'xiaotong', '135790'

'xiaoli', '112233'

3、Excel文件读取

1)引入xlrd 库

python 读取Excel文件,需要先安装xlrd 库,可以直接在命令行输入:pip install xlrd

上面的提示说明我们之前已经安装了。

2)编写读取Excel 文件函数
复制代码
需求:如下图Excel表。通过Excel来获取数据,希望返回字典组成的列表:[{"username":" standard_user","password":"secret_sauce"},

# {"username":" visual_user","password":"secret_sauce"},{"username":" error_user","password":"password_NG"}]

编写函数实现:

复制代码
def get_excel_data(filename,sheetnum):path ='login_data.xlsx'

    book_data =xlrd.open_workbook(path) #打开文档

    book_sheet=book_data.sheet_by_index(0) #打开Excel中第一个表

    rows_num =book_sheet.nrows  #sheet1 行数,row:行

    rows0 =book_sheet.row_values(0) #第一行的各个名称作为字典的键,列表

    rows0_num =len(rows0)   #第一行列表的长度即为数据的列数



    data_list =[ ]  #存放读取的数据

    for i inrange(1,rows_num):rows_data =book_sheet.row_values(i) #取出每一行的值作为列表,列表的每一项为一个字典

        rows_dir ={ }

        fory inrange(0,rows0_num):#将每一列的值与每一行对应起来

            rows_dir[rows0[y]] =rows_data[y] #键值对应

        data_list.append(rows_dir)   #每个字典插入到列表中

    return data_list

输出结果如下,方便后面使用

今天就先学习到这里吧。

每天进步一点点,加油!

相关推荐
databook4 小时前
Manim实现脉冲闪烁特效
后端·python·动效
程序设计实验室5 小时前
2025年了,在 Django 之外,Python Web 框架还能怎么选?
python
倔强青铜三6 小时前
苦练Python第46天:文件写入与上下文管理器
人工智能·python·面试
用户25191624271110 小时前
Python之语言特点
python
刘立军10 小时前
使用pyHugeGraph查询HugeGraph图数据
python·graphql
数据智能老司机13 小时前
精通 Python 设计模式——创建型设计模式
python·设计模式·架构
数据智能老司机15 小时前
精通 Python 设计模式——SOLID 原则
python·设计模式·架构
c8i16 小时前
django中的FBV 和 CBV
python·django
c8i16 小时前
python中的闭包和装饰器
python
这里有鱼汤19 小时前
小白必看:QMT里的miniQMT入门教程
后端·python