selenium中常見的表單元素操作方法總結

2022-02-25 13:01:30

前言

selenium是瀏覽器自動化測試框架，是一個用於Web應用程式測試的工具，可以直接執行在瀏覽器當中，並可以驅動瀏覽器執行指定的動作，如點選、下拉、填充資料、刪除cookie等操作，還可以獲取瀏覽器當前頁面的原始碼，就像使用者在瀏覽器中操作一樣。該工具所支援的瀏覽器有IE瀏覽器、Mozilla Firefox以及Google Chrome等。selenium有很多語言的版本，比如：Java、Ruby、Python等。

操作表單元素

常見的表單元素

§ Input
§ button
§ checkbox
§ select

1、操作輸入框：分為兩步，第一步：找到元素第二步：使用send_keys(value),將資料填充進去。範例程式碼如下：

# 以百度為例
#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 3:35 PM
# 檔案      ：selenium操作input.py
# IDE      ：PyCharm

# 匯入webdriver 模組
from selenium import webdriver
# 匯入time模組
import time

# 建立瀏覽器引擎
driver = webdriver.Chrome(executable_path='chromedriver')

# 指定url
url = 'http://www.baidu.com'

# 使用引擎開啟網頁
driver.get(url)

# 通過ID查詢input框
inputTag = driver.find_element_by_id('kw')
# 向input框傳送python值
inputTag.send_keys('python')

# 停頓3秒
time.sleep(3)

使用clear 方法可以清除輸入框中的內容。範例程式碼如下：

# 自動清除
inputTag.clear()
# 停頓3秒
time.sleep(3)

# 關閉當前瀏覽頁
driver.close()

2、操作checkbox:因為要選中checkbox 標籤，在網頁中是通過滑鼠點選的。因此想要選中checkbox標籤，那麼選中這個標籤，然後執行click事件。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 3:35 PM
# 檔案      ：selenium操作checkbox.py
# IDE      ：PyCharm


# 匯入webdriver模組
from selenium import webdriver
import time

# 建立瀏覽器引擎
driver = webdriver.Chrome(executable_path='chromedriver')

# 設定url
url = 'https://passport.mingrisoft.com/login/index.html?tpl=sch'

# 開啟網頁
driver.get(url)

# 通過name查詢checkbox，本例預設是選中的
checkboxTag = driver.find_element_by_name('rempwd')
# 執行click()命令，取消預設選中
checkboxTag.click()
# 停頓3秒
time.sleep(3)
# 關閉當前瀏覽頁
driver.close()

3、選擇select：select元素不能直接點選。因為點選後還需要選中元素，因此selenium專門為select標籤提供了一個類selenium.webdriver.support.ui.Select。將獲取的元素當成引數傳到這個類中，建立這個物件。以後就可以使用這個物件進行選擇了。範例程式碼如下：

from selenium.webdriver.support.ui import Select
# 選中這個標籤，然後使用Select建立物件
selectTag = Select(driver.find_element_by_name('jumpMenu'))
# 根據索引選擇
selectTag.select_by_index(1)
# 根據值選擇
selectTag.select_by_value('http://www.95yueba.com')
# 根據可是的文字選擇
selectTag.select_by_visible_text('95秀使用者端')
# 取消所有選項
selectTag.deselect_all()

4、操作按鈕：操作按鈕有很多種方式。比如單擊、右擊、雙擊等。常用的是點選。直接呼叫click函數就可以。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 4:59 PM
# 檔案      ：selenium操作按鈕標籤.py
# IDE      ：PyCharm

# 匯入webdriver 模組
from selenium import webdriver
# 匯入Select模組
from selenium.webdriver.support.ui import Select
# 匯入time模組
import time

# 建立瀏覽器引擎
driver = webdriver.Chrome(executable_path='chromedriver')

# 指定url
url = 'http://www.baidu.com'

# 使用引擎開啟網頁
driver.get(url)

# 通過ID查詢input框
inputTag = driver.find_element_by_id('kw')
# 向input框傳送python值
inputTag.send_keys('python')

# 查詢百度一下按鈕
submitTag = driver.find_element_by_id('su')
# click()執行百度命令
submitTag.click()
# 關閉當前瀏覽器頁面
driver.close()

行為鏈

所謂“行為鏈”就是一連串的操作動作。有時候在頁面中的操作可能要有很多步，那麼這時候可以使用滑鼠行為鏈類ActionChains來完成。例如將滑鼠移動到某個元素上並執行點選事件。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 5:34 PM
# 檔案      ：selenium操作行為鏈.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

url = 'http://www.baidu.com/'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)
driver.get(url)

# 找到input、百度一下標籤
inputTag = driver.find_element_by_id('kw')
submitTag = driver.find_element_by_id('su')

# 建立行為
actions = ActionChains(driver)
# 選中input框，然後輸入'中國女足'
actions.move_to_element(inputTag)
actions.send_keys_to_element(inputTag, '中國女足')
# 選中百度一下標籤，然後單擊
actions.move_to_element(submitTag)
actions.click(submitTag)
# 統一執行actions
actions.perform()

time.sleep(13)
driver.close()

還有更多的滑鼠相關操作

click_and_hold(element):點選但不鬆開滑鼠
context_click(element):右鍵點選
double_click(element):雙擊

更多方法參考：http://selenium-python.readthedocs.io/api.html

Cookie操作

1、獲取所有的cookie

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 6:57 PM
# 檔案      ：selenium操作cookie.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'https://www.baidu.com/'

# 存取url
driver.get(url)
for cookie in driver.get_cookies():
    print(cookie)

driver.close()

程式執行結果如下：

{'domain': '.baidu.com', 'expiry': 1675940497, 'httpOnly': False, 'name': 'BAIDUID_BFESS', 'path': '/', 'sameSite': 'None', 'secure': True, 'value': '058DB780F413036AE207FA5D5DBDFD0D:FG=1'}
{'domain': '.baidu.com', 'expiry': 1644408097, 'httpOnly': False, 'name': 'BA_HECTOR', 'path': '/', 'secure': False, 'value': 'ac81212k8l8180e5ug1h077oi0q'}
{'domain': '.baidu.com', 'httpOnly': False, 'name': 'H_PS_PSSID', 'path': '/', 'secure': False, 'value': '35414_35104_31254_35488_34584_35491_35796_35316_26350_35765_35745'}
{'domain': '.baidu.com', 'expiry': 1675940496, 'httpOnly': False, 'name': 'BAIDUID', 'path': '/', 'secure': False, 'value': '058DB780F413036AE207FA5D5DBDFD0D:FG=1'}
{'domain': '.baidu.com', 'expiry': 3791888143, 'httpOnly': False, 'name': 'BIDUPSID', 'path': '/', 'secure': False, 'value': '058DB780F413036A0DFEEB9DEAF54433'}
{'domain': '.baidu.com', 'expiry': 3791888143, 'httpOnly': False, 'name': 'PSTM', 'path': '/', 'secure': False, 'value': '1644404497'}
{'domain': 'www.baidu.com', 'expiry': 1645268497, 'httpOnly': False, 'name': 'BD_UPN', 'path': '/', 'secure': False, 'value': '123253'}
{'domain': 'www.baidu.com', 'httpOnly': False, 'name': 'BD_HOME', 'path': '/', 'secure': False, 'value': '1'}
Process finished with exit code 0

2、根據cookie的key獲取cookie的value，範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 6:57 PM
# 檔案      ：selenium操作cookie.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'https://www.baidu.com/'

# 存取url
driver.get(url)
# 通過某個key獲取cookie的value
print(driver.get_cookie('PSTM'))
driver.close()

程式執行結果如下：

{'domain': '.baidu.com', 'expiry': 3791888742, 'httpOnly': False, 'name': 'PSTM', 'path': '/', 'secure': False, 'value': '1644405096'}

Process finished with exit code 0

3、刪除所有的cookies,範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 6:57 PM
# 檔案      ：selenium操作cookie.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'https://www.baidu.com/'

# 存取url
driver.get(url)
# 刪除所有cookies
driver.delete_all_cookies()

程式執行結果如下：

4、刪除某個cookie,範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 6:57 PM
# 檔案      ：selenium操作cookie.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'https://www.baidu.com/'

# 存取url
driver.get(url)

for cookie in driver.get_cookies():
    print(cookie)

# driver.close()

# 根據cookie的key獲取cookie值
# print(driver.get_cookie('PSTM'))

# 刪除指定的cookie
driver.delete_cookie('PSTM')
print(driver.get_cookie('PSTM'))

程式執行結果如下：

{'domain': '.baidu.com', 'expiry': 1644409502, 'httpOnly': False, 'name': 'BA_HECTOR', 'path': '/', 'secure': False, 'value': '2s04ag8l0g252g6k6o1h0794f0q'}
{'domain': '.baidu.com', 'httpOnly': False, 'name': 'H_PS_PSSID', 'path': '/', 'secure': False, 'value': '35105_31253_34584_35490_35246_35796_35315_26350_35724_35765_35746'}
{'domain': '.baidu.com', 'expiry': 1675941902, 'httpOnly': False, 'name': 'BAIDUID', 'path': '/', 'secure': False, 'value': 'ACFC052478761063262775E7A2DE4C5F:FG=1'}
{'domain': '.baidu.com', 'expiry': 3791889549, 'httpOnly': False, 'name': 'BIDUPSID', 'path': '/', 'secure': False, 'value': 'ACFC052478761063DE9BEFA0F014E38F'}
# 沒刪除之前是存在的
{'domain': '.baidu.com', 'expiry': 3791889549, 'httpOnly': False, 'name': 'PSTM', 'path': '/', 'secure': False, 'value': '1644405902'}
{'domain': 'www.baidu.com', 'expiry': 1645269902, 'httpOnly': False, 'name': 'BD_UPN', 'path': '/', 'secure': False, 'value': '123253'}
{'domain': 'www.baidu.com', 'httpOnly': False, 'name': 'BD_HOME', 'path': '/', 'secure': False, 'value': '1'}
# 刪除之後，返回結果None
None

Process finished with exit code 0

頁面等待

現在網頁很多采用了Ajax技術，這樣程式便不能確定何時某個元素完全載入出來了。如果實際頁面等待時間過長導致某個dom元素還沒載入，但是你的程式碼直接使用了這個WebElement，那麼會報錯NullPointer的異常。為了解決這個問題，selenium提供了兩種等待方式：隱式等待，顯示等待。

1.隱式等待：呼叫driver.implicitly_wait，那麼在獲取任何元素之前，會先等待10秒時間。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 8:25 PM
# 檔案      ：隱式等待.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver
# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'https://www.csdn.net/'

driver.get(url)

# 隱式等待10秒
driver.implicitly_wait(10)
# 隨意指定一個id值
driver.find_element_by_id('dfsfasdfasfasfsfdsafds')

2.顯式等待：顯式等待是表明某個條件成立後才執行獲取元素的操作。也可以在等待的時候指定一個最大的時間，如果超過這個時間那麼就報異常錯誤。顯式等待應該使用selenium.webdriver.support.excepted_conditions期望的條件和selenium.webdriver.support.ui.WebDriverWait來配合完成。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 8:30 PM
# 檔案      ：顯式等待.py
# IDE      ：PyCharm

# 匯入模組
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

headers = {
  'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.99 Safari/537.36',
 'cookie':'uuid_tt_dd=10_20562014200-1642230088053-480005; __gads=ID=6d32e6ec1ea72d65-225f830cd5cf00dc:T=1642230106:RT=1642230106:S=ALNI_MZjtH_S6lcshDmkkDY4DQKIA1Ateg; ssxmod_itna=QqAxyD0DRD9G9ixl+D+rFOep8GfDc0AWOYPhx0HAeiODUxn4iaDToPTDGq+mR4YeSQm3BDQeYGYPGCijnRbiVb1YP5QbDneG0DQKGmzxGUFD7w+e2p=qDPDbxYPDGQqDbDY8tGwDY5HDGqDnDY=XD7Ug7AmWFMjgeudDCgmpli+zWi5oDOpxfi5IW8D=0DeI7i5K+A+zWc4qeowjSEFDGSTQ2iD=; ssxmod_itna2=QqAxyD0DRD9G9ixl+D+rFOep8GfDc0AWOYPxnK3xqqDsqDBQDjbPxx=MdidGYnBa2KYHl+iq7KwCb8uGrDgeC7owaD4m0DHDa7dIL3soD+I+02APk8wU977qxT8zGTNw63Xu1Mf=8QGDStRKPzTE+Df0p/+vBjRQszRPoMqf248fp03YujRGGUWjhzlUEIjrfAC2fzOYpG3dF1RfOaGDWD7jgD7=DeqxxD==; UserName=weixin_41905135; UserInfo=5d63720898c541cfa30ae44e4525ef44; UserToken=5d63720898c541cfa30ae44e4525ef44; UserNick=Bruce_Liuxiaowei; AU=E1F; UN=weixin_41905135; BT=1642230175210; p_uid=U110000; Hm_up_6bcd52f51e9b3dce32bec4a3997715ac={"islogin":{"value":"1","scope":1},"isonline":{"value":"1","scope":1},"isvip":{"value":"1","scope":1},"uid_":{"value":"weixin_41905135","scope":1}}; Hm_ct_6bcd52f51e9b3dce32bec4a3997715ac=6525*1*10_20562014200-1642230088053-480005!5744*1*weixin_41905135; c_dl_prid=1643325931307_582254; c_dl_rid=1644247742278_429472; c_dl_fref=https://blog.csdn.net/orange_xiang/article/details/82924296; c_dl_fpage=/download/weixin_38701952/12853345; c_dl_um=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-0-82924296.first_rank_v2_pc_rank_v29; dc_session_id=10_1644408299161.775025; dc_sid=8287d83a7595ae0d8779cb4fcedbe31a; csrfToken=Lb84yS2_Mi8haar-7oGjC4bZ; c_first_ref=default; c_first_page=https://www.csdn.net/; c_segment=5; Hm_lvt_6bcd52f51e9b3dce32bec4a3997715ac=1644373995,1644395322,1644401531,1644409224; c_page_id=default; log_Id_click=364; c_pref=https://www.csdn.net/; c_ref=https://blog.csdn.net/weixin_41905135?spm=1000.2115.3001.5343; dc_tos=r71fd2; log_Id_pv=444; Hm_lpvt_6bcd52f51e9b3dce32bec4a3997715ac=1644410632; log_Id_view=722'
}
# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 建立driver
driver = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'https://www.csdn.net/'
driver.get(url)
try:
    element = WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, 'sousuo')) # 引數以元組形式傳入
    )
finally:
    driver.quit()

3、一些其他等待條件

presence_of_element_located:某個元素已經載入完畢
presence_of_all_element_located:網頁中所有滿足條件的元素載入完畢
element_to_be_cliable:某個元素可以點選了

切換頁面

有時候視窗中有很多子tab頁面。這時候需要切換頁面的。selenium提供了一個叫做switch_to_window來進行切換，具體切換到哪個頁面，可以從driver.window_handles中找到。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/9/22 9:57 PM
# 檔案      ：切換頁面.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

# 指定url
url1 = 'https://www.douban.com/'
url2 = 'https://www.baidu.com/'
driver = webdriver.Chrome(executable_path=execute_path)

driver.get(url1)
driver.execute_script("window.open('https://www.baidu.com/')")

程式執行結果如下圖：

切換到指定視窗，程式碼如下：

# 從driver.window_handlers中取出具體的視窗，driver.window_handlers是一個列表，裡面是視窗控制程式碼
# 此案例只開啟兩個視窗，所以handles列表取‘1'
driver.switch_to_window(driver.window_handles[1])

設定代理ip

爬取網頁的時候為避免被封掉ip地址，我們就會需要代理ip，隨時更換ip地址。不同的瀏覽器有不同的實現方式，本文以Chrome瀏覽器為例。範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/10/22 9:57 AM
# 檔案      ：設定代理ip.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver

# chromedriver路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'

options = webdriver.ChromeOptions()
options.add_argument("--proxy-server=http://218.75.69.50:57903")

driver = webdriver.Chrome(executable_path=execute_path, chrome_options=options)
driver.get('http://httpbin.org/ip')

程式執行結果如下圖：

WebElement元素

從網頁中獲取網頁元素，from selenium.webdriver.remote.webelement import WebElement 類是每個獲取的元素的所屬類。一些常用的屬性如下：

1、get_attribute:獲取標籤某個屬性值

2、screenshot:獲取當前頁面的截圖，並儲存為png格式，此方法只能在driver上使用。

範例程式碼如下：

#_*_coding:utf-8_*_
# 作者      ：liuxiaowei
# 建立時間   ：2/10/22 10:28 AM
# 檔案      ：獲取網頁元素.py
# IDE      ：PyCharm

# 匯入模組
from selenium import webdriver
from selenium.webdriver.remote.webelement import WebElement

# 指定瀏覽器驅動的路徑
execute_path = '../使用selenium爬取動態載入的資訊/chromedriver'
# 建立瀏覽器驅動
brower = webdriver.Chrome(executable_path=execute_path)

# 指定url
url = 'http://www.baidu.com/'

# 開啟網頁
brower.get(url)

# 找到搜尋按鈕"百度一下"
baidu_Btn = brower.find_element_by_id('su')
print(type(baidu_Btn))
# 獲取按鈕屬性value的值
print(baidu_Btn.get_attribute('value'))

# 儲存網頁截圖
brower.save_screenshot('baidu.png')

執行結果如下：