Selenium - Selenium webdriver的基本操作

目錄

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

• 返回Selenium目錄
• 返回隨筆首頁

常用類與方法

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/')

    # 根據 class 屬性獲取 a 標簽對象
    a_obj = driver.find_element_by_class_name('navbar-brand')

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

    # 獲取該標簽下的子標簽，或者子標簽元素
    div_obj = driver.find_element_by_class_name('col-xs-3')
    print(div_obj.find_element_by_tag_name('input'))
    print(div_obj.find_element_by_tag_name('input').get_attribute('value'))

    # 獲取驗證碼圖片的大小
    print(driver.find_element_by_id('image_code').size)
    # # 保存驗證碼圖片
    driver.find_element_by_id('image_code').screenshot('./a.png')

except Exception as e:
    print(e)

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

需要注意的是：按鈕提交時，除了click之外，還有一個submit，區別是click就是單純的點一下。而submit是完成了表單提交，需要攜帶表單信息進行提交的。如果是button按鈕的話，只能使用click，而不能使用submit。

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('https://pythonav.com/login/')

    input_obj = driver.find_element_by_id('id_username')
    time.sleep(2)
    input_obj.send_keys('selenium is cool')
    time.sleep(2)
    input_obj.send_keys(Keys.CONTROL, 'a')
    time.sleep(2)
    input_obj.send_keys(Keys.DELETE)
    time.sleep(3)
    input_obj.send_keys('no 。。。。。')
    time.sleep(3)
    
    pwd_obj = driver.find_element_by_id('id_password')
    pwd_obj.send_keys('selenium is very cool')
    pwd = pwd_obj.get_attribute('value')
    print(pwd)

    js = "return $('#id_password').val()"
    res = driver.execute_script(js)
    print(res)
    js2 = "alert('{}')".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'
]
'''

示例：

import time
from selenium import webdriver
from selenium.webdriver.common.keys import Keys  # 鍵盤事件
from selenium.webdriver import ActionChains  # 鼠標事件

driver = webdriver.Chrome()
driver.get('https://www.baidu.com')
driver.find_element_by_id('kw').send_keys('聽雨危樓')
driver.find_element_by_id('su').click()
time.sleep(2)
ActionChains(driver).click(driver.find_element_by_partial_link_text('- 博客園')).perform()

time.sleep(10)
driver.quit()

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")')

---

歡迎斧正，that's all see also：[selenium ExpectedConditions常用方法](https://www.jianshu.com/p/e470a89ff125) | [Python+Selenium練習篇之10-刷新當前頁面](https://www.cnblogs.com/wangyinghao/p/10532606.html)