selenium+python 数据驱动-csv篇，可封装

#循环读取csv文件中的数据，可以作为用户名，密码等使用
from selenium import webdriver
import csv
#获取csv文件中password列
with open(r'C:\bbs\data\logindata.csv','r',encoding='utf-8') as csvfile:
    text=csv.DictReader(csvfile)
    pwd=[row['password'] for row in text]
#获取csv文件中username列
with open(r'C:\bbs\data\logindata.csv','r',encoding='utf-8') as csvfile:
    text=csv.DictReader(csvfile)
    user=[row['username'] for row in text]
#打印出来以列表形式，也可以获取单条数据
print(user,pwd)

#打印结果

['a1', 'a2dad', 'daf2_sf'] ['2', '54513', '1655455']

------------------------------------------------------------------------------------

有的系统需要的测试数据比较多，会经常用到这个地方，我们可以封装成一个函数，用到哪列数据调用哪列，达到数据和代码分离的目的

#定义get_data 函数

def get_data(file_path,lie,i):

　　with open(file_path,'r',encoding='utf-8') as csvfile:

　　　　text=csv.DictReader(csvfile)

　　　　pwd=[row[lie] for row in text]

return pwd[i]

#调用的时候传入文件路径，列名，第几行数据i 三个参数

get_data()