一些网站需要用户登录后才能看内容,就要用到cookie把登录信息记下来。
操作过程就是:先用selenium打开一个网页,手动登录之后,再把cookie导出到一个文件,下次自动化访问网页就可以导入cookie文件, 不需要再手动登录了。
#导出cookie
import pickle
import selenium.webdriver
driver = selenium.webdriver.Firefox()
driver.get("http://www.google.com")
pickle.dump(driver.get_cookies(), open("cookies.txt", "wb"))
#导入cookie
driver = selenium.webdriver.Firefox()
driver.get("http://www.google.com")
cookies = pickle.load(open("cookies.txt", "rb"))
for cookie in cookies:
driver.add_cookie(cookie)