使用Python的Selenium(網頁測試工具模組),有時候需要測試登入功能,但是Selenium登入後再到下一個頁面無法保留登入狀態,瀏覽器的登入資訊就都會被刷掉,是因為每次使用的Cookie暫存資訊是都沒有儲存著,要如何解決這個問題呢?
Python 使用 Selenium並保留登入狀態 教學
STEP1
先把cookie儲存下來。
import pickle
from selenium import webdriver
browser = webdriver.Chrome()
browser.get("https://www.google.com")
pickle.dump(browser.get_cookies(), open("cookies.pkl","wb"))
STEP2
再來將存下的cookie,加載回去。
import pickle
from selenium import webdriver
browser = webdriver.Chrome()
browser.get("https://www.google.com")
cookies = pickle.load(open("cookies.pkl", "rb"))
for cookie in cookies:
browser.add_cookie(cookie)