Автоматизация браузера с Selenium: примеры и инструкции

Работа с iframe

Для взаимодействия с элементами внутри iframe необходимо переключить контекст на него.

# Переключение по имени или id
driver.switch_to.frame('iframe-name')
# Или по WebElement
iframe = driver.find_element(By.CSS_SELECTOR, 'iframe')
driver.switch_to.frame(iframe)

# Теперь можно искать элементы внутри iframe
inner_element = driver.find_element(By.ID, 'inside-iframe')
print(inner_element.text)

# Вернуться к основному содержимому
driver.switch_to.default_content()

Вывод: текст элемента внутри iframe, например "Это содержимое iframe".

Настройка опций браузера (headless, прокси, user-agent)

Через класс Options можно конфигурировать запуск браузера.

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

options = Options()
options.add_argument('--headless=new')  # безголовый режим
options.add_argument('--proxy-server=http://proxy:8080')  # прокси
options.add_argument('--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36')
options.add_experimental_option('excludeSwitches', ['enable-logging'])  # скрыть логи

driver = webdriver.Chrome(options=options)
driver.get('https://httpbin.org/headers')
print(driver.page_source)
driver.quit()

В исходном коде страницы виден переданный User-Agent и IP прокси.

Обработка всплывающих окон и алертов

Алерты (диалоговые окна) можно перехватить и взаимодействовать с ними.

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# Нажимаем кнопку, вызывающую alert
driver.find_element(By.ID, 'alert-btn').click()

# Ждём появления алерта
alert = WebDriverWait(driver, 5).until(EC.alert_is_present())
print(alert.text)  # 'Вы уверены?'
alert.accept()  # принять
alert.dismiss() # отклонить (если нужно)
# Для ввода текста в prompt:
alert.send_keys('Ответ')
alert.accept()

Вывод: 'Вы уверены?'. После accept окно закрывается.

Загрузка файлов с помощью Selenium

Для загрузки файла обычно используется элемент input[type=file], в который отправляется путь к файлу через send_keys.

from pathlib import Path

upload_input = driver.find_element(By.CSS_SELECTOR, 'input[type="file"]')
file_path = str(Path.cwd() / 'example.txt')
upload_input.send_keys(file_path)

# Дождаться завершения загрузки
WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.CLASS_NAME, 'upload-success')))
print('Файл загружен')

В консоль выводится 'Файл загружен' после появления индикатора успеха.

Параллельный запуск браузеров с ThreadPoolExecutor

Для ускорения тестов можно запускать несколько экземпляров браузера параллельно.

from concurrent.futures import ThreadPoolExecutor
from selenium import webdriver
from selenium.webdriver.chrome.service import Service

def open_and_get_title(url):
    service = Service()  # предполагается, что chromedriver в PATH
    driver = webdriver.Chrome(service=service)
    driver.get(url)
    title = driver.title
    print(f'{url}: {title}')
    driver.quit()
    return title

urls = ['https://www.example.com', 'https://www.google.com', 'https://www.python.org']
with ThreadPoolExecutor(max_workers=3) as executor:
    results = executor.map(open_and_get_title, urls)
print(list(results))

Вывод в консоль:
https://www.example.com: Example Domain
https://www.google.com: Google
https://www.python.org: Welcome to Python.org
['Example Domain', 'Google', 'Welcome to Python.org']

Использование Selenium Grid для распределенного тестирования

Selenium Grid позволяет запускать тесты на удалённых машинах и в разных браузерах одновременно.

# Настройка удалённого драйвера
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

capabilities = DesiredCapabilities.CHROME.copy()
driver = webdriver.Remote(
    command_executor='http://localhost:4444/wd/hub',
    options=webdriver.ChromeOptions()
)
driver.get('https://www.selenium.dev')
print(driver.title)
driver.quit()

Вывод: 'Selenium'. Тест выполнен на удалённом узле Grid, что видно в логах хаба.