爬蟲神器:超炫酷的自動化工具 Selenium

Hello,我是 Alex 007,爲啥是007呢?因爲叫 Alex 的人太多了,再加上每天007的生活,Alex 007就誕生了。

今天給大家分享一個小白看見大呼WC的工具Selenium,在我第一次看見別人演示這個工具的時候,真的是忍不住說了一句:WCNB,好了,正文開始。

一、初識Selenium

Selenium其實是一個用於Web應用程序測試的工具,也就是說,當我們開發了一個應用程序之後,需要進行必要的測試,通過測試可以找出很多bug

假如說我們要測試一個登陸功能,每次測試都需要輸入用戶名和密碼,這顯然是非常浪費時間的,需求創造出產品,Selenium就可以滿足我們這個需求。

Selenium測試直接運行在瀏覽器中,就像真正的用戶在操作一樣,你可以看到瀏覽器自己打開了一個網址,自己輸入了賬號和密碼,一切都是自動完成的,模擬真實用戶對瀏覽器進行操作。

1.環境安裝

首先,在PythonSelenium屬於第三方庫,因此需要安裝,在這裏直接提供使用阿里源的安裝命令:

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)MozillaFirefoxSafariGoogle ChromeOpera等,還有AndroidBlackBerry等手機端瀏覽器,另外也支持無界面瀏覽器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()

注意:

  1. find_element_by_xxx找的是第一個符合條件的標籤,find_elements_by_xxx找的是所有符合條件的標籤。

  2. 根據IDCSS選擇器和XPath獲取,它們返回的結果完全一致。

  3. 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 Souppyquery等)來提取信息了。

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)

在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章