selenium+python实现自动化登录

 1 #encoding=utf-8
 2 from selenium import webdriver  
 3 import time  
 4 import os
 5 
 6 #模拟登陆weibo  
 7 
 8 def getCookies(weibo):
 9     """ 获取Cookies """
10     cookies = []
11     driver = webdriver.Firefox()
12     time.sleep(3)   #sleep一下，否则有可能报错
13     driver.get("https://weibo.com/login/")
14     #cur_path=os.getcwd()
15     #fileSuc = open(cur_path+"/login.html", ‘w‘)
16     #fileSuc.write(driver.page_source)
17     #用户名 密码  
18     elem_user = driver.find_element_by_xpath(‘//input[@id="loginname"]‘) 
19     elem_user.send_keys(‘*****@163.com‘) #浏览器版本不匹配的时候这里可能报错
20     elem_pwd = driver.find_element_by_xpath(‘//input[@type="password"]‘)
21     elem_pwd.send_keys(‘*****‘)
22     
23     commit = driver.find_element_by_xpath(‘//a[@node-type="submitBtn"]‘)
24     commit.click()
25     time.sleep(3)
26     #fileSuc1 = open(cur_path+"/weibo2.html", ‘w‘)
27     #fileSuc1.write(driver.page_source)
28     #print driver.title  
29     #登录成功后获取cookie
30     cookie = {}
31     if "微博-随时随地发现新鲜事" in driver.title:
32         for elem in driver.get_cookies():
33             cookie[elem["name"]] = elem["value"]
34         if len(cookie) > 0:
35             logger.warning("Get Cookie Successful: %s" % account)
36             cookies.append(cookie)
37             continue
38     else:
39         logger.warning("Get Cookie Failed: %s!" % account)
40     
41     driver.close()  
42     driver.quit()  
43     return cookies
44 
45 cookies = getCookies(myWeiBo)
46 print cookies
47 logger.warning("Get Cookies Finish!( Num:%d)" % len(cookies))