#循环读取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()
