Автоматизация Python Selenium без входа в систему (cookie и токен)

2024-07-12

Оглавление

Включить обход браузера при входе в систему

Использование входа в систему может уменьшить количество повторяющихся операций при каждом входе в систему, напрямую управлять страницей меню после входа в систему, а также может уменьшить количество операций входа в систему с проверкой безопасности, таких как операции входа в систему с проверкой изображения. Примечание. Файлы cookie и токены имеют срок действия.

Получите файлы cookie непосредственно из инструментов разработчика и добавьте их. На рисунке ниже показаны файлы 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.get(URL) перед wd.add_cookie(cookie_dict=cookie) следующим образом:


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

Получить куки через интерфейс

После получения данных cookie через интерфейс добавьте использование файлов cookie в 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

Включить обход браузера при входе в систему

Введите в Google или браузере Edge: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()

Обмен технологиями