如何在 Python 使用 Selenium 並保留登入狀態呢?設定瀏覽器Cookie暫存資訊!

Python&Selenium
使用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)