Automação Python Selenium sem login (cookie e token)

2024-07-12

Índice

Habilitar login para ignorar o navegador

O uso do login pode reduzir as operações repetidas de cada login, operar diretamente a página do menu após o login e também pode reduzir o login de verificação de segurança, como operações de login de verificação de imagem. Nota: Cookies e tokens têm datas de validade.

Obtenha cookies diretamente das ferramentas do desenvolvedor e adicione-os. A imagem abaixo mostra os cookies de vários sites na página web.


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()

levantar exception_class(mensagem, tela, rastreamento de pilha)
selenium.common.exceptions.InvalidCookieDomainException: Mensagem: domínio de cookie inválido
(Informações da sessão: MicrosoftEdge=126.0.2592.87)

Se você receber o erro acima, poderá adicionar uma linha de wd.get(URL) antes de wd.add_cookie(cookie_dict=cookie), como segue:


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

Obtenha cookies através da interface

Após obter os dados do cookie por meio da interface, adicione o uso do cookie no Selenium


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

Habilitar login para ignorar o navegador

Entre no Google ou no navegador Edge:chrome://version/ 或 edge://version/ 查看配置文件夹路径，Remova o padrão no final e adicione --user-data-dir= antes do caminho para unir o caminho que desejamos.

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()

Nota: Ao usar este método, você precisa fechar o programa do navegador correspondente, caso contrário, um erro será relatado, então você precisa encerrar o processo do navegador correspondente antes da execução. O código acima usa consulta de correspondência difusa para encerrar o processo, e o código a seguir é a correspondência completa.


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 #判断浏览器进程是否杀完

adicionar token


    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()

Compartilhamento de tecnologia