Демонстрация опций профиля Selenium для настройки браузера

Этот пример демонстрирует, как использовать библиотеку Selenium для запуска браузера Chrome с опциями профиля, такими как использование профиля по умолчанию. Код выполняет следующие действия:

  1. Создание директории для скриншотов: Функция ensure_screenshot_directory() создаёт папку screenshots, если она не существует.
  2. Отображение сообщений: Функция show_message() показывает всплывающие окна с информацией о текущей опции.
  3. Запуск браузера: Функция run_browser_with_option() запускает Chrome с указанной опцией профиля, загружает страницу https://parsertools.ru/, делает скриншот и сохраняет его в папку screenshots.
  4. Проверка User-Agent: Используется библиотека requests для получения User-Agent с сайта и вывода его в консоль.
  5. Перебор опций: Основной цикл проходит по списку available_options, применяя каждую опцию с задержкой в 3 секунды между запусками и задержкой перед закрытием окна в 15 секунд.
  6. Обработка ошибок: Код включает обработку исключений для случаев, когда браузер не запускается или возникают проблемы при загрузке страницы.

Каждая опция сопровождается описанием, а результат работы — скриншот страницы с именем, соответствующим опции.

Урок по этой теме с описанием можно посмотреть здесь.

import os
import time
import tkinter as tk
from tkinter import messagebox
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.common.exceptions import WebDriverException
from webdriver_manager.chrome import ChromeDriverManager
import requests  # Для получения User-Agent с сайта

# Задержка между запусками в секундах
DELAY = 3

# Список опций профиля для настройки браузера
available_options = [
    ("--profile-directory=Default", "Использование профиля по умолчанию"),
    # ("--user-data-dir=/path/to/profile", "Использование пользовательской директории профиля (замените /path/to/profile на реальный путь)")
]

def ensure_screenshot_directory():
    """Создаёт директорию screenshots, если она не существует."""
    screenshot_dir = os.path.abspath("screenshots")
    if not os.path.exists(screenshot_dir):
        os.makedirs(screenshot_dir)
    return screenshot_dir

def show_message(title, message):
    """Отображает всплывающее окно с указанным заголовком и сообщением."""
    root = tk.Tk()
    root.withdraw()  # Скрываем основное окно
    messagebox.showinfo(title, message)
    root.destroy()

def run_browser_with_option(index, arg, desc):
    """Запускает браузер с выбранной опцией и делает скриншот."""
    # Показываем всплывающее окно с текущей задачей
    show_message("Текущая задача", f"Опция: {arg}\nОписание: {desc}")

    print(f"\n=== Запуск браузера с опцией: {arg} ({desc}) ===")

    # Настройка Chrome с выбранной опцией
    opts = webdriver.ChromeOptions()
    opts.add_argument(arg)

    try:
        driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=opts)
    except WebDriverException as e:
        print(f"🚫 Ошибка при запуске браузера с опцией '{arg}': {e}")
        return

    try:
        url = "https://parsertools.ru/"
        print(f"📄 Загружаем страницу: {url}")
        driver.get(url)
        time.sleep(15)  # Ждём загрузки страницы

        # Получаем User-Agent с сайта
        try:
            # Извлекаем User-Agent из аргументов для проверки
            user_agent = None
            for opt in opts.arguments:
                if opt.startswith("--user-agent="):
                    user_agent = opt.replace("--user-agent=", "")
                    break
            if not user_agent:
                user_agent = "Default Chrome User-Agent"
            response = requests.get(url, headers={"User-Agent": user_agent})
            if response.status_code == 200:
                print(f"🏷️ User-Agent, полученный с сайта: {response.json()['user-agent']}")
            else:
                print(f"⚠️ Не удалось получить User-Agent с сайта: {response.status_code}")
        except Exception as e:
            print(f"⚠️ Ошибка при запросе User-Agent: {e}")

        # Сохраняем скриншот
        screenshot_dir = ensure_screenshot_directory()
        screenshot_name = f"screenshot_{index}_{arg.replace('--','').replace('=','_').replace('/', '_')}.png"
        screenshot_path = os.path.join(screenshot_dir, screenshot_name)
        driver.save_screenshot(screenshot_path)
        print(f"📸 Скриншот сохранён: {screenshot_path}")

    except Exception as e:
        print(f"🚫 Ошибка при работе с браузером: {e}")
    finally:
        driver.quit()
        print("🔌 Браузер закрыт.")

if __name__ == "__main__":
    # Приветственное сообщение
    print("🎉 Добро пожаловать в демонстрацию опций профиля Selenium!")
    print(f"Каждая опция будет запущена с задержкой в {DELAY} секунд.\n")

    # Перебираем все опции профиля
    for i, (arg, desc) in enumerate(available_options, start=1):
        run_browser_with_option(i, arg, desc)
        print(f"\n⏳ Ожидание {DELAY} секунд перед следующим запуском...")
        time.sleep(DELAY)

    print("\n🎉 Все опции профиля протестированы! Проверьте папку 'screenshots' для результатов.")

Похожие темы:

Selenium Alert Confirm Prompt

Автоматическое заполнение формы обратной связи

Парсинг ссылок с классом <button> с BeautifulSoup