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