Selenium webdriver的基本操作

目錄

• 常用類與方法
• 選擇器
• Keys（鍵盤相關事件）
• 鼠標相關事件
• EC
• 其他案例
  • 瀏覽器的前進后退和刷新
  • 執行js代碼

常用類與方法

返回頂部

from selenium import webdriver   # 驅動瀏覽器
from selenium.webdriver import ActionChains   # 鼠標的相關操作，比如滑動驗證
from selenium.webdriver.common.by import By   # 選擇器，以什么方式選擇標簽元素
from selenium.webdriver.common.keys import Keys   # 鍵盤相關
from selenium.webdriver.support import expected_conditions as EC  # 各種判斷，一般跟等待事件連用，比如說等待某個元素加載出來
from selenium.webdriver.support.wait import WebDriverWait  # 等待事件，可以與EC連用
browser = webdriver.Chrome()
 wait = WebDriverWait(browser, 10)
 browser.get('https://www.baidu.com')
 browser.maximize_window() # 窗口最大化
 print(browser.current_url) # 獲取當前頁URL
 print(browser.title) # 獲取頁面的title
 print(browser.name) # 獲取driver對象：chrome
 print(browser.current_window_handle) # 獲取當前窗口
 print(browser.get_cookies()) # 獲取cookies
 print(browser.page_source) # 獲取當前頁面內容
 browser.refresh() # 刷新
 browser.save_screenshot('error.png') # 保存截圖
 browser.close() # 關閉當前窗口
 browser.quit() # 退出瀏覽器驅動，關閉所有關聯窗口

選擇器

返回頂部

from selenium import webdriver   # 驅動瀏覽器

from selenium.webdriver.support.wait import WebDriverWait  # 等待事件

browser = webdriver.Chrome()

wait = WebDriverWait(browser, 10)

browser.get('https://www.baidu.com')
 browser.find_element_by_id('su') # id選擇器
 browser.find_element_by_class_name('xx') # 類選擇器，1個
 browser.find_elements_by_class_name('xx') # 類選擇器，多個
 browser.find_element_by_link_text('xxx') # 鏈接文本選擇器
 browser.find_element_by_xpath('xxxx') # xpath選擇器
 browser.find_element_by_tag_name('h1') # 標簽選擇器，獲取1個
 browser.find_elements_by_tag_name('h1') # 標簽選擇器，獲取多個
 browser.find_element_by_css_selector('xxx') # 樣式選擇器

在選擇器的選用時，要記得獲取一個還是多個，要區分開find_elements還是find_element。這個s容易忽略。

另外，上面的選擇器，都可以寫成下面的形式：

from selenium.webdriver.common.by import By  # 選擇器，以什么方式選擇標簽元素

browser.find_element(by=By.ID, value=None)  # 選擇器類型， value值

常用的選擇器有：

By.ID

By.LINK_TEXT

By.CLASS_NAME

By.XPATH

By.CSS_SELECTOR

By.TAG_NAME

我們可以通過標簽選擇器來做：

import time

from selenium import webdriver  # 驅動瀏覽器

from selenium.webdriver import ActionChains  # 鼠標的相關操作，比如滑動驗證

from selenium.webdriver.common.by import By  # 選擇器，以什么方式選擇標簽元素

from selenium.webdriver.common.keys import Keys  # 鍵盤相關

from selenium.webdriver.support import expected_conditions as EC  # 各種判斷，一般跟等待事件連用，比如說等待某個元素加載出來

from selenium.webdriver.support.wait import WebDriverWait  # 等待事件

browser = webdriver.Chrome()
 wait = WebDriverWait(browser, 10)
 browser.get('https://www.baidu.com')
input = browser.find_element_by_id('kw') # id選擇器
 input.send_keys('張開嘴') # 如果獲取到的是個input框，就可以往里面填值
 time.sleep(1)
 input.clear() # 清空值
 submit = browser.find_element(By.ID, 'su')
 submit.click() # 點擊事件
 # 有的時候，click事件可能會有問題，我們使用鍵盤事件來處理
 submit.send_keys(Keys.ENTER)
 print(input.get_attribute('name')) # 獲取name屬性
 print(input.text) # 獲取內容，但這里的input沒有text
 print(input.tag_name)
 print(input.get_property('name')) # 獲取給定屬性，用法類似get_attribute
 print(browser.find_element_by_class_name('mnav').text) # 找個有text的

當獲取了標簽對象后，我們可以獲取這個標簽的屬性、內容、css樣式和子標簽（子標簽也支持這些操作），示例來自pythonav登錄頁：

import time

from selenium import webdriver

driver = webdriver.Chrome()
 driver.implicitly_wait(10)
try:
 driver.get('https://pythonav.com/login/')
<span class="hljs-comment"># 根據 class 屬性獲取 a 標簽對象</span>
a_obj = driver.find_element_by_class_name(<span class="hljs-string">'navbar-brand'</span>)

<span class="hljs-comment"># 獲取 a 標簽的href屬性</span>
print(a_obj.get_attribute(<span class="hljs-string">'href'</span>))
print(a_obj.text)  <span class="hljs-comment"># 獲取標簽的文本內容</span>
print(a_obj.tag_name)   <span class="hljs-comment"># 獲取標簽的名稱</span>
<span class="hljs-comment"># 獲取 css 樣式</span>
print(a_obj.value_of_css_property(<span class="hljs-string">'padding'</span>))
print(a_obj.value_of_css_property(<span class="hljs-string">'font-size'</span>))
<span class="hljs-comment"># 如果這個標簽有 submit 屬性，那么就可以點擊它了，比如我們點擊form表單中有submit屬性的標簽</span>
driver.find_element_by_xpath(<span class="hljs-string">'//*[@id="fm"]/div[5]/div/input'</span>).submit()

<span class="hljs-comment"># 獲取該標簽下的子標簽，或者子標簽元素</span>
div_obj = driver.find_element_by_class_name(<span class="hljs-string">'col-xs-3'</span>)
print(div_obj.find_element_by_tag_name(<span class="hljs-string">'input'</span>))
print(div_obj.find_element_by_tag_name(<span class="hljs-string">'input'</span>).get_attribute(<span
            class="hljs-string">'value'</span>))

<span class="hljs-comment"># 獲取驗證碼圖片的大小</span>
print(driver.find_element_by_id(<span class="hljs-string">'image_code'</span>).size)
<span class="hljs-comment"># # 保存驗證碼圖片</span>
driver.find_element_by_id(<span class="hljs-string">'image_code'</span>).screenshot(<span class="hljs-string">'./a.png'</span>)

except Exception as e:
 print(e)
finally:
 time.sleep(10)
 driver.quit()
 # 代碼截止到2019/12/19號有效

Keys

返回頂部

鍵盤事件有：

from selenium.webdriver.common.keys import Keys  # 鍵盤相關

print(dir(Keys))

'''
 [
 'ADD', 'ALT', 'ARROW_DOWN', 'ARROW_LEFT', 'ARROW_RIGHT', 'ARROW_UP', 'BACKSPACE', 'BACK_SPACE', 'CANCEL', 'CLEAR', 'COMMAND', 'CONTROL', 'DECIMAL', 'DELETE', 'DIVIDE', 'DOWN', 'END', 'ENTER', 'EQUALS', 'ESCAPE', 'F1', 'F10', 'F11', 'F12', 'F2', 'F3', 'F4', 'F5', 'F6', 'F7', 'F8', 'F9', 'HELP', 'HOME', 'INSERT', 'LEFT', 'LEFT_ALT', 'LEFT_CONTROL', 'LEFT_SHIFT', 'META', 'MULTIPLY', 'NULL', 'NUMPAD0', 'NUMPAD1', 'NUMPAD2', 'NUMPAD3', 'NUMPAD4', 'NUMPAD5', 'NUMPAD6', 'NUMPAD7', 'NUMPAD8', 'NUMPAD9', 'PAGE_DOWN', 'PAGE_UP', 'PAUSE', 'RETURN', 'RIGHT', 'SEMICOLON', 'SEPARATOR', 'SHIFT', 'SPACE', 'SUBTRACT', 'TAB', 'UP'
 ]
 '''

來個鍵盤事件示例：

import time

from selenium import webdriver

from selenium.webdriver.common.keys import Keys

driver = webdriver.Chrome()
try:
driver.get(<span class="hljs-string">'https://pythonav.com/login/'</span>)

input_obj = driver.find_element_by_id(<span class="hljs-string">'id_username'</span>)
time.sleep(<span class="hljs-number">2</span>)
input_obj.send_keys(<span class="hljs-string">'selenium is cool'</span>)
time.sleep(<span class="hljs-number">2</span>)
input_obj.send_keys(Keys.CONTROL, <span class="hljs-string">'a'</span>)
time.sleep(<span class="hljs-number">2</span>)
input_obj.send_keys(Keys.DELETE)
time.sleep(<span class="hljs-number">3</span>)
input_obj.send_keys(<span class="hljs-string">'no 。。。。。'</span>)
time.sleep(<span class="hljs-number">3</span>)

pwd_obj = driver.find_element_by_id(<span class="hljs-string">'id_password'</span>)
pwd_obj.send_keys(<span class="hljs-string">'selenium is very cool'</span>)
pwd = pwd_obj.get_attribute(<span class="hljs-string">'value'</span>)
print(pwd)

js = <span class="hljs-string">"return $('#id_password').val()"</span>
res = driver.execute_script(js)
print(res)
js2 = <span class="hljs-string">"alert('{}')"</span>.format(res)
driver.execute_script(js2)

finally:
 time.sleep(10)
 driver.quit()

鼠標相關事件

返回頂部

from selenium.webdriver import ActionChains  # 鼠標的相關操作，比如滑動驗證

print(dir(ActionChains))

'''
 [
 'click', 'click_and_hold', 'context_click', 'double_click', 'drag_and_drop', 'drag_and_drop_by_offset', 'key_down', 'key_up', 'move_by_offset', 'move_to_element', 'move_to_element_with_offset', 'pause', 'perform', 'release', 'reset_actions', 'send_keys', 'send_keys_to_element'
 ]
 '''

EC

返回頂部

再來看EC：

from selenium.webdriver.support import expected_conditions as EC  # 各種判斷，一般跟等待事件連用，比如說等待某個元素加載出來

print(dir(EC))

'''
 ['NoAlertPresentException', 'NoSuchElementException', 'NoSuchFrameException', 'StaleElementReferenceException', 'WebDriverException', 'WebElement', 'builtins', 'cached', 'doc', 'file', 'loader', 'name', 'package', 'spec', '_element_if_visible', '_find_element', '_find_elements', 'alert_is_present', 'element_located_selection_state_to_be', 'element_located_to_be_selected', 'element_selection_state_to_be', 'element_to_be_clickable', 'element_to_be_selected', 'frame_to_be_available_and_switch_to_it', 'invisibility_of_element', 'invisibility_of_element_located', 'new_window_is_opened', 'number_of_windows_to_be', 'presence_of_all_elements_located', 'presence_of_element_located', 'staleness_of', 'text_to_be_present_in_element', 'text_to_be_present_in_element_value', 'title_contains', 'title_is', 'url_changes', 'url_contains', 'url_matches', 'url_to_be', 'visibility_of', 'visibility_of_all_elements_located', 'visibility_of_any_elements_located', 'visibility_of_element_located']
 '''

expected_conditions中封裝了各種判斷，一般常應用於斷言中和等待機制搭配使用。比如判斷某個元素是否存在，存在則返回該標簽對象，不存在則報報錯，並且可以自定制報錯：

import time

from selenium import webdriver  # 驅動瀏覽器

from selenium.webdriver import ActionChains  # 鼠標的相關操作，比如滑動驗證

from selenium.webdriver.common.by import By  # 選擇器，以什么方式選擇標簽元素

from selenium.webdriver.common.keys import Keys  # 鍵盤相關

from selenium.webdriver.support import expected_conditions as EC  # 各種判斷，一般跟等待事件連用，比如說等待某個元素加載出來

from selenium.webdriver.support.expected_conditions import NoSuchElementException

from selenium.webdriver.support.wait import WebDriverWait  # 等待事件

browser = webdriver.Chrome()
 # wait = WebDriverWait(browser, 10)
 #
 browser.get('https://www.baidu.com')
try:
 print(EC.presence_of_element_located((By.ID, 'xxx'))(browser)) # 判斷某個元素是否被加到了dom樹里，並不代表該元素一定可見
 except NoSuchElementException as e:
 print('找不到這個標簽')

如上例，當前頁面中沒有id名為xxx的，所以報錯，需要注意的是，報錯也是一個對象，要想得到False結果，需要執行這個對象並且將瀏覽器驅動對象傳進去。

其他的判斷：

import time

from selenium import webdriver  # 驅動瀏覽器

from selenium.webdriver import ActionChains  # 鼠標的相關操作，比如滑動驗證

from selenium.webdriver.common.by import By  # 選擇器，以什么方式選擇標簽元素

from selenium.webdriver.common.keys import Keys  # 鍵盤相關

from selenium.webdriver.support import expected_conditions as EC  # 各種判斷，一般跟等待事件連用，比如說等待某個元素加載出來

from selenium.webdriver.support.wait import WebDriverWait  # 等待事件

browser = webdriver.Chrome()
 wait = WebDriverWait(browser, 10)
 #
 browser.get('https://www.baidu.com')
print(EC.title_contains('XX')(browser)) # 判斷當前頁面的title是否包含預期字符串
 print(EC.visibility_of_element_located((By.ID, 'su'))(browser)) # 判斷某個元素是否可見（代表元素非隱藏，元素的寬和高都不等於0）
 print(EC.invisibility_of_element_located(((By.ID, 'xxx')))(browser)) # 判斷某個元素中是否不存在於dom樹或不可見
 print(EC.text_to_be_present_in_element((By.ID, 'su'), '百度一下')(browser)) # 判斷某個元素中的text是否包含了預期的字符串， 示例中的標簽沒有text值
 print(EC.text_to_be_present_in_element_value((By.ID, 'su'), '百度一下')) # 判斷某個元素中的value屬性是否包含了預期的字符串
 print(EC.alert_is_present()(browser)) # 判斷頁面上是否存在alert
 print(EC.element_to_be_clickable((By.ID, 'su'))(browser)) # 判斷某個元素中是否可見並且是enable的，這樣的話才叫clickable

其他案例

返回頂部

瀏覽器的前進后退和刷新

返回頂部

我們可以使用selenium自動的模擬前進（forward）后退（back）和刷新（refresh）三個動作：

import time

from selenium import webdriver  # 驅動瀏覽器

browser = webdriver.Chrome()

browser.get('https://www.baidu.com')

browser.get('https://www.taobao.com')

browser.get('http://www.sina.com.cn/')

time.sleep(5)

browser.back()  # 后退

time.sleep(3)

browser.forward()  # 前進

time.sleep(2)

browser.refresh()  # 刷新

browser.close()  # 關閉當前窗口

time.sleep(3)

browser.quit()

執行js代碼

返回頂部

在有些時候，需要執行一些js代碼來完成任務。

from selenium import webdriver  # 驅動瀏覽器

browser = webdriver.Chrome()

browser.get('https://www.zhihu.com/explore')

browser.execute_script('alert("xxoo")')

---

see also： selenium ExpectedConditions常用方法 | Python+Selenium練習篇之10-刷新當前頁面