Hello,我是 Alex 007,爲啥是007呢?因爲叫 Alex 的人太多了,再加上每天007的生活,Alex 007就誕生了。
今天給大家分享一個小白看見大呼WC的工具
Selenium
,在我第一次看見別人演示這個工具的時候,真的是忍不住說了一句:WCNB,好了,正文開始。
文章目錄
一、初識Selenium
Selenium
其實是一個用於Web
應用程序測試的工具,也就是說,當我們開發了一個應用程序之後,需要進行必要的測試,通過測試可以找出很多bug
。
假如說我們要測試一個登陸功能,每次測試都需要輸入用戶名和密碼,這顯然是非常浪費時間的,需求創造出產品,Selenium
就可以滿足我們這個需求。
Selenium
測試直接運行在瀏覽器中,就像真正的用戶在操作一樣,你可以看到瀏覽器自己打開了一個網址,自己輸入了賬號和密碼,一切都是自動完成的,模擬真實用戶對瀏覽器進行操作。
1.環境安裝
首先,在Python
中Selenium
屬於第三方庫,因此需要安裝,在這裏直接提供使用阿里源的安裝命令:
pip install -i http://mirrors.aliyun.com/pypi/simple --trusted-host mirrors.aliyun.com selenium
然後我們需要下載一個瀏覽器驅動程序:下載地址,需要下載跟自己瀏覽器匹配的版本,查看瀏覽器的版本可以通過地址欄訪問chrome://version/。
2.簡單使用
from time import sleep
from selenium import webdriver
if __name__ == '__main__':
# 後面是你的瀏覽器驅動位置,記得前面加r''防止字符轉義的
driver = webdriver.Chrome(r'../chromedriver.exe')
# 用 get 打開百度頁面
driver.get("http://www.baidu.com")
# 找到百度的輸入框,並輸入 美女
driver.find_element_by_id('kw').send_keys('美女')
sleep(1)
# 點擊搜索按鈕
driver.find_element_by_id('su').click()
sleep(1)
# 在打開的頁面中找到“美女_海量精選高清圖片_百度圖片”,並打開這個頁面
driver.find_elements_by_link_text('美女_海量精選高清圖片_百度圖片')[0].click()
sleep(1)
# 關閉瀏覽器
driver.quit()
運行之後就可以發現,Python
自動打開了一個Chrome
瀏覽器,然後按照我們的指令搜索指定的內容並打開指定的網頁。
二、基本知識
1.瀏覽器的創建
Selenium
支持非常多的瀏覽器,如IE(7, 8, 9, 10, 11)
,Mozilla
、Firefox
,Safari
,Google Chrome
,Opera
等,還有Android
、BlackBerry
等手機端瀏覽器,另外也支持無界面瀏覽器PhantomJS
。
from selenium import webdriver
browser = webdriver.Chrome()
browser = webdriver.Firefox()
browser = webdriver.Edge()
browser = webdriver.PhantomJS()
browser = webdriver.Safari()
2.元素定位
webdriver
提供了一系列的元素定位方法,常用的方法有以下幾種:
find_element_by_id()
find_element_by_name()
find_element_by_class_name()
find_element_by_tag_name()
find_element_by_link_text()
find_element_by_partial_link_text()
find_element_by_xpath()
find_element_by_css_selector()
注意:
-
find_element_by_xxx
找的是第一個符合條件的標籤,find_elements_by_xxx
找的是所有符合條件的標籤。 -
根據
ID
、CSS
選擇器和XPath
獲取,它們返回的結果完全一致。 -
Selenium
還提供了通用方法find_element()
,它需要傳入兩個參數:查找方式By
和值。
3.節點交互
Selenium
可以驅動瀏覽器來執行一些操作,也就是說可以讓瀏覽器模擬執行一些動作。
比較常見的用法有:輸入文字時用send_keys()
方法,清空文字時用clear()
方法,點擊按鈕時用click()
方法。
from selenium import webdriver
import time
if __name__ == '__main__':
browser = webdriver.Chrome(r'../chromedriver.exe')
browser.get('https://www.taobao.com')
search = browser.find_element_by_id('q')
search.send_keys('MAC')
time.sleep(1)
search.clear()
search.send_keys('IPhone')
button = browser.find_element_by_class_name('btn-search')
button.click()
browser.quit()
4.動作鏈
在網頁中還有一些連續性的動作,沒有特定的執行對象,比如鼠標拖拽、鍵盤按鍵等,這些動作在Selenium
中就需要使用動作鏈來完成。
import time
from selenium import webdriver
from selenium.webdriver import ActionChains
if __name__ == '__main__':
browser = webdriver.Chrome(r'../chromedriver.exe')
url = 'http://www.runoob.com/try/try.php?filename=jqueryui-api-droppable'
browser.get(url)
browser.switch_to.frame('iframeResult')
source = browser.find_element_by_css_selector('#draggable')
target = browser.find_element_by_css_selector('#droppable')
actions = ActionChains(browser)
actions.click_and_hold(source)
time.sleep(1)
for i in range(5):
actions.move_by_offset(xoffset=17, yoffset=0).perform()
time.sleep(0.5)
actions.release()
browser.quit()
5.執行JavaScript
對於Selenium API
沒有提供的操作,比如下拉進度條等,可以通過直接模擬運行JavaScript
來達到目的。
import time
from selenium import webdriver
if __name__ == '__main__':
browser = webdriver.Chrome(r'../chromedriver.exe')
browser.get('https://www.jd.com/')
browser.execute_script('window.scrollTo(0, document.body.scrollHeight)')
browser.execute_script('alert("123")')
time.sleep(2)
browser.quit()
6.獲取頁面源碼數據
通過page_source
屬性可以獲取網頁的源代碼,接着就可以使用解析庫(如正則表達式、Beautiful Soup
、pyquery
等)來提取信息了。
7.前進和後退
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/')
browser.back()
time.sleep(10)
browser.forward()
browser.close()
8.Cookie
處理
使用Selenium,還可以方便地對Cookies進行操作,例如獲取、添加、刪除Cookies等。示例如下:
from selenium import webdriver
browser = webdriver.Chrome()
browser.get('https://www.zhihu.com/explore')
print(browser.get_cookies())
browser.add_cookie({'name': 'name', 'domain': 'www.zhihu.com', 'value': 'germey'})
print(browser.get_cookies())
browser.delete_all_cookies()
print(browser.get_cookies())
9.異常處理
from selenium import webdriver
from selenium.common.exceptions import TimeoutException,NoSuchElementException,NoSuchFrameException
try:
browser=webdriver.Chrome()
browser.get('http://www.runoob.com/try/try.php?filename=jqueryui-api-droppable')
browser.switch_to.frame('iframssseResult')
except TimeoutException as e:
print(e)
except NoSuchFrameException as e:
print(e)
finally:
browser.close()
10.無頭瀏覽器
谷歌的無頭瀏覽器,是一款無界面的谷歌瀏覽器。
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import time
# 創建一個參數對象,用來控制chrome以無界面模式打開
chrome_options = Options()
chrome_options.add_argument('--headless')
chrome_options.add_argument('--disable-gpu')
# 驅動路徑
path = r'C:\Users\ZBLi\Desktop\1801\day05\ziliao\chromedriver.exe'
# 創建瀏覽器對象
browser = webdriver.Chrome(executable_path=path, chrome_options=chrome_options)
# 上網
url = 'http://www.baidu.com/'
browser.get(url)
time.sleep(3)
browser.save_screenshot('baidu.png')
browser.quit()
11.規避檢測識別
現在不少大網站有對selenium採取了監測機制。
比如正常情況下我們用瀏覽器訪問淘寶等網站的 window.navigator.webdriver的值爲undefined。而使用selenium訪問則該值爲true。那麼如何解決這個問題呢?
只需要設置Chromedriver的啓動參數即可解決問題。在啓動Chromedriver之前,爲Chrome開啓實驗性功能參數excludeSwitches,它的值爲[‘enable-automation’],完整代碼如下:
from selenium.webdriver import Chrome
from selenium.webdriver import ChromeOptions
option = ChromeOptions()
option.add_experimental_option('excludeSwitches', ['enable-automation'])
driver = Chrome(options=option)