学习python下使用selenium2自动测试第6天，参数化编程这节课花了两天时间。
本次编程主要时间是花在熟悉python上
知识点or坑点：
1、读取txt、xml、csv等文件存储的账号、密码
txt文件格式,逗号分割（也可使用其他符号）：
www.126.com,user1,pwd1
www.qq.com,user2,pwd2
www.163.com,user3,pwd3
user_file = open('user_info.txt','r')
lines = user_file.readlines()
user_file.close()
for line in lines:
mail = line.split(',')[0]
username = line.split(',')[1]
pwd = line.split(',')[2]
print(mail,username,pwd)
2、打开多窗口,定位新窗口
获取所有窗口句柄：cur_windows = dr.window_handles
定位窗口：
for cur_window in cur_windows:
dr.switch_to.window(cur_window)
3、python编程中self的作用
在我的理解中，self是全局的this对象，定义一个class LoginSetup():
self就是指LoginSetup这个对象本身
在本class中定义多个对象时，可使用self.function( [param1,param2,...] )来调用，
被调用方法的参数self为默认参数，真实接收参数从第二个开始
被调用函数：
def open_url(self,url):
js = 'window.open("'+url+'")'
print(js)
self.driver.execute_script(js)
调用函数：
def login(self):
json_lines = []
……
self.open_login(json_lines)
4、python的init初始化，是前后两个下划线横杠（坑点）
#初始化，两个下划横杠
def __init__(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(10)
url = 'http://www.baidu.com'
self.init_url = url
self.driver.get(url)
在类运行时，初始化一些参数
5、python的in
for 循环、比较部分字符串都可使用in
foreach：
for line in lines:
print(line['username'])
print(line['pwd'])
比较字符串：
if 'QQ' in url:
print('true')
登陆QQ邮箱代码：
1 from selenium import webdriver 2 from time import sleep 3 4 class QqLogin(): 5 6 def user_login(dr,username,pwd): 7 8 print(username,pwd) 9 10 dr.switch_to.frame("login_frame") 11 idInput = dr.find_element_by_id('u') 12 pwdInput = dr.find_element_by_id('p') 13 idInput.clear() 14 idInput.send_keys(username) 15 pwdInput.clear() 16 pwdInput.send_keys(pwd) 17 18 #登录 19 dr.find_element_by_id('login_button').click() 20 21 #返回上级frame 22 #dr.switch_to.parent_frame() 23 24 #返回主frame,此处两个方法都可以 25 dr.switch_to.default_content() 26 27 #调用返回主frame需要等一下 28 sleep(1) 29 30 switchs = dr.find_elements_by_css_selector('div') 31 print( len(switchs) ) 32 33 #获取登录用户名、邮箱 34 name = dr.find_element_by_id('useralias') 35 email = dr.find_element_by_id('useraddr') 36 print('qq登录成功|',name.text,'---',email.text) 37 38 #dr.quit()
登陆网易邮箱代码
1 from time import sleep 2 3 class WyLogin(): 4 5 #登录 6 def user_login(driver,username,pwd): 7 sleep(1) 8 print( driver.current_url ) 9 driver.switch_to.frame('x-URS-iframe') 10 emailInput = driver.find_element_by_name("email") 11 emailInput.clear() 12 #emailInput.send_keys(username)#火狐执行无效 13 email_id = emailInput.get_attribute("id") 14 js = 'document.getElementById("'+email_id+'").value="'+username+'"' 15 print(js) 16 driver.execute_script(js)#执行js 17 pwdInput = driver.find_element_by_name("password") 18 pwdInput.clear() 19 pwdInput.send_keys(pwd) 20 dologin = driver.find_element_by_id("dologin") 21 dologin.click() 22 23 print('网易邮箱登陆成功') 24 25 driver.switch_to.default_content() 26
登陆方法：
1 # coding=utf-8 2 from selenium import webdriver 3 from time import sleep 4 from loginQq import QqLogin 5 from loginWy import WyLogin 6 7 class LoginSetup(): 8 9 #初始化，两个下划横杠 10 def __init__(self): 11 self.driver = webdriver.Chrome() 12 self.driver.implicitly_wait(10) 13 url = 'http://www.baidu.com' 14 self.init_url = url 15 self.driver.get(url) 16 17 #登录 18 def login(self): 19 user_file = open('user_info.txt','r') 20 lines = user_file.readlines() 21 user_file.close() 22 23 try: 24 json_lines = [] 25 26 for line in lines: 27 lineArr = line.split(',') 28 mail_type = lineArr[0] 29 mail = lineArr[1] 30 username = lineArr[2] 31 pwd = lineArr[3] 32 33 # 打开浏览器窗口，定位当前窗口 34 url = 'http://'+mail 35 self.open_url(url) 36 37 json_line = {} 38 json_line['username'] = username 39 json_line['pwd'] = pwd 40 json_line['mail'] = mail 41 json_line['mail_type'] = mail_type 42 json_lines.append(json_line) 43 #for end 44 45 print(json_lines) 46 self.open_login(json_lines) 47 48 #关闭浏览器 49 #self.driver.quit() 50 51 except BaseException as error: 52 print('error:',error) 53 self.driver.quit() 54 #end login 55 56 #打开新窗口 57 def open_url(self,url): 58 js = 'window.open("'+url+'")' 59 print(js) 60 self.driver.execute_script(js) 61 ''' 62 win_handles = self.driver.window_handles 63 print( len(win_handles) ) 64 for hand in win_handles: 65 print( hand ) 66 67 cur_window = self.driver.current_window_handle 68 self.driver.switch_to.window(cur_window) 69 print('now url is ',self.driver.current_url) 70 ''' 71 # win_handles = self.driver.window_handles 72 #end open_url 73 74 #定位新打开窗口,登录 75 def open_login(self,json_lines): 76 dr = self.driver 77 cur_windows = dr.window_handles 78 print( len(cur_windows) ) 79 username = '' 80 pwd = '' 81 mail_type = '' 82 83 for cur_window in cur_windows: 84 dr.switch_to.window(cur_window) 85 cur_url = dr.current_url 86 print('cur_url 1 = ',cur_url) 87 88 for line in json_lines: 89 mail = line['mail'] 90 mail_in = mail.replace('www.','') 91 print(mail_in,cur_url) 92 if mail_in in cur_url: 93 print('username') 94 mail_type = line['mail_type'] 95 username = line['username'] 96 pwd = line['pwd'] 97 98 99 print(mail_type,username) 100 101 if username == '': 102 continue 103 104 #调用登录方法 105 print('username is ',username) 106 if 'QQ' in mail_type: 107 QqLogin.user_login(dr,username,pwd) 108 if 'WY' in mail_type: 109 WyLogin.user_login(dr,username,pwd) 110 111 # end open_login 112 113 114 115 #调用登录方法 116 LoginSetup().login() 117 118
txt文件格式：
WY,www.126.com,user1,pwd1
WY,mail.163.com,user2,pwd2
QQ,mail.qq.com,user3,pwd3