Selenium使用教程

Selenium是一款功能強大的自動化測試框架，它可以模擬用戶在瀏覽器中的各種操作，比如點擊、輸入等，通過自動化測試幫助開發人員減少手動測試的時間和成本，提高測試效率和代碼質量。

本篇文章將從選取元素、常見操作、等待元素、多窗口處理和自定義瀏覽器驅動等多個方面詳細闡述Selenium的使用方法。

一、選取元素

1、選取ID元素

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_id('su')
element.click()

driver.quit()

2、選取class元素

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_class_name('s_btn')
element.click()

driver.quit()

3、選取name元素

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_name('wd')
element.send_keys('selenium')

driver.quit()

4、選取link_text元素

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_link_text('新聞')
element.click()

driver.quit()

二、常見操作

1、輸入文字

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_name('wd')
element.send_keys('selenium')
element.submit()

driver.quit()

2、點擊按鈕

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_id('su')
element.click()

driver.quit()

3、切換窗口

from selenium import webdriver

driver = webdriver.Chrome()
url1 = 'https://www.baidu.com/'
url2 = 'http://news.baidu.com/'

driver.get(url1)
driver.find_element_by_link_text('新聞').click()

handles = driver.window_handles
for handle in handles:
    if handle != driver.current_window_handle:
        driver.switch_to.window(handle)
        print(driver.title)

driver.quit()

三、等待元素

1、強制等待

from selenium import webdriver
import time

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

time.sleep(5)  # 等待5秒

driver.quit()

2、隱式等待

from selenium import webdriver

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.implicitly_wait(10)  # 最長等待10秒
driver.get(url)

element = driver.find_element_by_name('wd')
element.send_keys('selenium')
element.submit()

driver.quit()

3、顯式等待

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
url = 'https://www.baidu.com/'
driver.get(url)

element = WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.NAME, 'wd')))
element.send_keys('selenium')
element.submit()

driver.quit()

四、多窗口處理

from selenium import webdriver

driver = webdriver.Chrome()
url1 = 'https://www.baidu.com/'
url2 = 'http://news.baidu.com/'

driver.get(url1)
driver.find_element_by_link_text('新聞').click()

handles = driver.window_handles
for handle in handles:
    if handle != driver.current_window_handle:
        driver.switch_to.window(handle)
        driver.close()

driver.switch_to.window(handles[0])
driver.find_element_by_name('wd').send_keys('selenium')
driver.find_element_by_id('su').click()

driver.quit()

五、自定義瀏覽器驅動

1、火狐瀏覽器

from selenium import webdriver

firefox_options = webdriver.FirefoxOptions()
firefox_options.set_headless()
firefox_options.add_argument('--disable-gpu')
firefox_options.add_argument('--no-sandbox')

driver = webdriver.Firefox(firefox_options=firefox_options)
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_name('wd')
element.send_keys('selenium')
element.submit()

driver.quit()

2、Edge瀏覽器

from selenium import webdriver
from msedge.selenium_tools import EdgeOptions, Edge

edge_options = EdgeOptions()
edge_options.use_chromium = True
edge_options.headless = True

driver = Edge(options=edge_options)
url = 'https://www.baidu.com/'
driver.get(url)

element = driver.find_element_by_name('wd')
element.send_keys('selenium')
element.submit()

driver.quit()

通過以上的示例，您可以快速掌握Selenium的基本使用方法，希望能對您的自動化測試工作有所幫助。