ログインなしの Python Selenium 自動化 (Cookie とトークン)

2024-07-12

ログインを利用すると、毎回のログインの繰り返し操作を軽減したり、ログイン後のメニューページを直接操作したり、画像認証ログインなどのセキュリティ認証ログインの手間を軽減したりできます。注: Cookie とトークンには有効期限があります。

開発者ツールから直接 Cookie を取得して追加します。下の図は、Web ページ上の複数のサイトの Cookie を示しています。必要なものを選択して追加します。


from selenium import webdriver
from selenium.webdriver.edge.options import Options
 
# 一般只需要name和value
cookie = {'name': 'ZY44', 'value': 'tLonhTkz50iHzxjhIsaaaafferr:C'}
 
options = Options()
# options.add_argument('--headless')
wd = webdriver.Edge(options=options)
 
wd.add_cookie(cookie_dict=cookie)
 
#for c in cookies:   # 如果是多个cookie要添加，cookies存储为列表是，使用循环添加
#   wd.add_cookie(c)
 
wd.refresh()  # 刷新页面
 
wd.get(URL)
 
wd.quit()

exception_class(メッセージ、画面、スタックトレース) を発生させます
selenium.common.exceptions.InvalidCookieDomainException: メッセージ: 無効な Cookie ドメイン
(セッション情報: MicrosoftEdge=126.0.2592.87)

上記のエラーが発生した場合は、次のように、wd.add_cookie(cookie_dict=cookie) の前に wd.get(URL) の行を追加できます。


wd.get(URL)
wd.add_cookie(cookie_dict=cookie)
wd.get(URL)

インターフェイス経由で Cookie を取得する

インターフェイスを通じて Cookie データを取得した後、Selenium に Cookie の使用を追加します


def get_cookies():
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36',
    }
    host = 'https://baike.baidu.com'
    req = requests.get(host, headers=headers)
    cookie_data = req.cookies.get_dict()
    cookies = []
    for key, value in cookie_data.items():
        cookies.append(dict(name=key, value=value))
    return cookies

ブラウザのバイパスログインを有効にする

Google またはエッジブラウザに次のように入力します。chrome://version/ 或 edge://version/ 查看配置文件夹路径，最後に Default を削除し、パスの前に --user-data-dir= を追加して、必要なパスを結合します。

profile_directory = r'--user-data-dir=C:UsersxxxAppDataLocalMicrosoftEdgeUser Data'


# 这里使用模糊匹配，把edge开头的进程都杀掉
if platform.system() == "Windows":
    os.system("taskkill -im msedge* -f")
else:
    os.system("killall -9 msedge*")
 
user_data = r'C:UsersxxxAppDataLocalMicrosoftEdgeUser Data'
profile_directory = rf'--user-data-dir={user_data}'
 
options = Options()
# options.add_argument('--headless')
options.add_argument(profile_directory)
wd = webdriver.Edge(options=options)
 
wd.maximize_window()
wd.get(URL)
 
wd.quit()

注: この方法を使用する場合は、対応するブラウザプログラムを閉じる必要があります。閉じるとエラーが報告されるため、実行前に対応するブラウザプロセスを強制終了する必要があります。上記のコードはあいまい一致クエリを使用してプロセスを強制終了します。次のコードは完全一致です。


returnCode=os.system('taskkill /F /iM chrome.exe')    # 谷歌
returnCode=os.system('taskkill /F /iM iexplore.exe')  # IE
returnCode=os.system('taskkill /F /iM firefox.exe')  # 火狐
returnCode=os.system('taskkill /F /iM msedge.exe')  # edge
assert returnCode==0 #判断浏览器进程是否杀完

トークンを追加


    token = "my_token"
 
    options = Options()
    options.add_argument('--headless')
    wd = webdriver.Edge(options=options)
 
    wd.execute_script("window.localStorage.setItem('token', '%s');" % token)  # 使用selenium执行js的操作添加token
 
    wd.maximize_window()
    wd.get(url)
    wd.quit()

技術共有