1、数据不规则，使用xml
2、使用固定的用户名与密码，使用数据字典
3、搜索关键字，使用数组
4、大量的数据，使用数据库
一、类、函数驱动
do_class.py
#encoding=utf-8
from selenium import webdriver
#调用public文件的Login类里的函数
#from public_class_data import Login
driver = webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")
class Account(object):
'''docstring for Account'''
def __init__(self,username='',password=''):
self.username = username
self.password = password
def do_login_as(user_info):
self.driver.find_element_by_id("idInput").clear()
self.driver.find_element_by_id("idInput").send_keys('username')
self.driver.find_element_by_id("pwdInput").clear()
self.driver.find_element_by_id("pwdInput").send_keys('password')
self.driver.find_element_by_id("loginBtn").click()
#实例化登录信息
admin = Account(username='admin',password='123')
guest = Account(username='guest',password='123')
#调用登录函数
do_login_as(admin)
do_login_as(guest)
#退出
def logout(self):
self.driver.find_element_by_link_text("退出").click()
self.driver.quit()
二、数组和字典方式的调用
public_class_data.py
#coding=utf-8
class Login():
def __init__(self):
self.driver=driver
def login(self,username,password):
self.driver.find_element_by_id("idInput").clear()
self.driver.find_element_by_id("idInput").send_keys(username)
self.driver.find_element_by_id("pwdInput").clear()
self.driver.find_element_by_id("pwdInput").send_keys(password)
self.driver.find_element_by_id("loginBtn").click()
#退出
def logout(self):
self.driver.find_element_by_link_text("退出").click()
self.driver.quit()
#coding=utf-8
#使用数组的方式读取
from selenium import webdriver
'''
file_info = open("baidu.txt",'r')
values = file_info.readlines()
file_info.close()
'''
shuzu = ['python','selenium','day day up']
for search in shuzu:
driver = webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("https://www.baidu.com")
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("idInput").send_keys('search')
driver.find_element_by_id("su").click()
driver.quit()
#使用数据字典的方式读数据，如用户名和密码
#encoding=utf-8
from selenium import webdriver
from public_class_data import Login
from time import sleep
'''
file_info = open("baidu.txt",'r')
values = file_info.readlines()
file_info.close()
'''
zidian = {'user01':'pass01','user02':'pass02'}
for k,v in zidian.items():
driver = webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("https://www.126.com")
Login(driver).login(k,v)
sleep(1)
Login(driver).logout()
prink(k,v)
三、csv文件驱动
#encoding=utf-8
import csv
my_file = 'user_info1.csv'
data = csv.reader(open(my_file,'r'))
#data2 = data.encode('gbk','ignore').decode("gbk")
for user in data:
print (user)
#显示其中的某一列
print (user[0])
print (user[1])
print (user[2])
print (user[3])
注意：Python3，读文件只能用r，不能用rb，否则会报错：
Traceback (most recent call last):
File "D:\Python3\study\do_data_file_csv.py", line 11, in <module>
for user in data:
_csv.Error: iterator should return strings, not bytes (did you open the file in text mode?)
[Finished in 0.5s with exit code 1]
四、txt文件驱动
#coding=utf-8
from selenium import webdriver
from public_class_data import Login
from time import sleep
user_file = open("user_info.txt",'r')
values = user_file.readlines()
user_file.close()
#ctrl +d ,批量修改，按几次就选中几个,#分隔行为;
for i in values:
username= i.split(';')[0]
print (username)
password = i.split(';')[1] #1代表读行里的第二列数据
print (password)
五、xml驱动
