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

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

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

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

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

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

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

# Список актуальных опций для настройки браузера
available_options = [
    ("--headless=new", "Запуск без окна браузера (headless-режим)"),
    ("--start-maximized", "Запуск браузера в развернутом окне"),
    ("--window-size=1000,800", "Установка размера окна 1000x800 пикселей"),
    ("--incognito", "Запуск в режиме инкогнито"),
    ("--disable-extensions", "Отключение расширений браузера"),
    ("--no-sandbox", "Запуск без песочницы (для Linux/CI)"),
    ("--disable-dev-shm-usage", "Использование /tmp вместо /dev/shm (Linux)")
]

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)  # Ждём загрузки страницы

        # Получаем заголовок страницы
        print(f"🏷️ Заголовок страницы: {driver.title}")

        # Сохраняем скриншот
        screenshot_dir = ensure_screenshot_directory()
        screenshot_path = os.path.join(screenshot_dir, f"screenshot_{index}_{arg.replace('--','')}.png")
        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' для результатов.")