一 . 認識Phantomjs
1.Phantomjs:無界面的瀏覽器
Selenium: 可以根據我們的指令,讓瀏覽器自動加載頁面,獲取需要的數據,甚至頁面截屏,或者判斷網站上某些動作是否發生。Selenium 自己不帶瀏覽器,不支持瀏覽器的功能,它需要與第三方瀏覽器結合在一起才能使用。但是我們有時候需要讓它內嵌在代碼中運行,所以我們可以用一個叫 Phantomjs 的工具代替真實的瀏覽器。
文檔地址:http://selenium-python.readthedocs.io/index.html
# 導入 webdriver
from selenium import webdriver
# 要想調用鍵盤按鍵操作需要引入keys包
from selenium.webdriver.common.keys
import Keys
# 調用環境變量指定的Phantomjs瀏覽器創建瀏覽器對象
driver = webdriver.Phantomjs()
# 如果沒有在環境變量指定Phantomjs位置# driver = webdriver.Phantomjs(executable_path="./phantomjs"))
# get方法會一直等到頁面被完全加載,然后才會繼續程序,通常測試會在這里選擇 time.sleep(2)
driver.get("http://www.baidu.com/")
# 獲取頁面名為 wrapper的id標簽的文本內容
data = driver.find_element_by_id("wrapper").text
# 打印數據內容
print data
# 打印頁面標題 "百度一下,你就知道
"print driver.title
# 生成當前頁面快照並保存
driver.save_screenshot("baidu.png")
# id="kw"是百度搜索輸入框,輸入字符串"長城"
driver.find_element_by_id("kw").send_keys("長城")
# id="su"是百度搜索按鈕,click() 是模擬點擊
driver.find_element_by_id("su").click()
# 獲取新的頁面快照
driver.save_screenshot("長城.png")
# 打印網頁渲染后的源代碼
print driver.page_source
# 獲取當前頁面Cookie
print driver.get_cookies()
# ctrl+a 全選輸入框內容
driver.find_element_by_id("kw").send_keys(Keys.CONTROL,'a')
# ctrl+x 剪切輸入框內容
driver.find_element_by_id("kw").send_keys(Keys.CONTROL,'x')
# 輸入框重新輸入內容
driver.find_element_by_id("kw").send_keys("atguigu")
# 模擬Enter回車鍵
driver.find_element_by_id("su").send_keys(Keys.RETURN)
# 清除輸入框內容
driver.find_element_by_id("kw").clear()
# 生成新的頁面快照
driver.save_screenshot("atguigu.png")
# 獲取當前url
print driver.current_url
# 關閉當前頁面,如果只有一個頁面,會關閉瀏覽器# driver.close()
# 關閉瀏覽器
driver.quit()
2.頁面操作大體內容
# 獲取id標簽值
element = driver.find_element_by_id("passwd-id")
# 獲取name標簽值
element = driver.find_element_by_name("user-name")
# 獲取標簽名值
element = driver.find_elements_by_tag_name("input")
# 也可以通過XPath來匹配
element = driver.find_element_by_xpath("//input[@id='passwd-id']")
3. 關於元素的選取,有如下的API 單個元素選取
# 定位UI元素 (WebElements) find_element_by_id find_elements_by_name find_elements_by_xpath find_elements_by_link_text find_elements_by_partial_link_text find_elements_by_tag_name find_elements_by_class_name find_elements_by_css_selector
虎課網https://www.wode007.com/sites/73267.html 設計塢https://www.wode007.com/sites/73738.html
4. 鼠標動作鏈
# 導入 webdriver
#導入 ActionChains 類
from selenium.webdriver import ActionChains
# 鼠標移動到 ac 位置
ac = driver.find_element_by_xpath('element')
ActionChains(driver).move_to_element(ac).perform()
# 在 ac 位置單擊
ac = driver.find_element_by_xpath("elementA")
ActionChains(driver).move_to_element(ac).click(ac).perform()
# 在 ac 位置雙擊
ac = driver.find_element_by_xpath("elementB")
ActionChains(driver).move_to_element(ac).double_click(ac).perform()
# 在 ac 位置右擊
ac = driver.find_element_by_xpath("elementC")
ActionChains(driver).move_to_element(ac).context_click(ac).perform()
# 在 ac 位置左鍵單擊hold住
ac = driver.find_element_by_xpath('elementF')
ActionChains(driver).move_to_element(ac).click_and_hold(ac).perform()
# 將 ac1 拖拽到 ac2 位置
ac1 = driver.find_element_by_xpath('elementD')
ac2 = driver.find_element_by_xpath('elementE')
ActionChains(driver).drag_and_drop(ac1, ac2).perform()
5 . 下拉菜單的操作(導入select類)
# 導入 Select 類
from selenium import webdriver
from selenium.webdriver.support.ui import Select
driver = webdriver.Chrome()
driver.get("http://127.0.0.1:8080")
# 找到 name 的選項卡
select = Select(driver.find_element_by_name('status'))
#根據位置下標
# select.select_by_index(1)
#根據值找到對應的選擇
# select.select_by_value("3")
#根據顯示值找到對應的選擇
select.select_by_visible_text("審核不通過")
6. 頁面切換
driver.switch_to.window("window name")
7. 操作頁面的前進和后退
driver.forward() #前進 driver.back() # 后退
8. 頁面等待
顯式等待是等待特定的時間,隱式等待是指定某一條件直到這個條件成立時繼續執行。
8.1 隱式等待比較簡單,就是簡單地設置一個等待時間,單位為秒
from selenium import webdriver
driver = webdriver.Chrome()
driver.implicitly_wait(10) # seconds
driver.get("http://www.xxxxx.com/loading")
8.2 顯式等待指定某個條件,然后設置最長等待時間。如果在這個時間還沒有找到元素,那么便會拋出異常了
from selenium import webdriver
from selenium.webdriver.common.by import By# WebDriverWait 庫,負責循環等待
from selenium.webdriver.support.ui import WebDriverWait# expected_conditions 類,負責條件出發
from selenium.webdriver.support import expected_conditions as EC
driver = webdriver.Chrome()
driver.get("http://www.xxxxx.com/loading")
try:
# 頁面一直循環,直到 id="myElement" 出現
element = WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "myElement")))
finally:
driver.quit()
二、網站模擬登錄豆瓣網
from selenium import webdriver
import time
# 創建一個瀏覽器客戶端,並且指定配置
# 如果是Phantomjs做瀏覽器客戶端還要設置路徑
# driver = webdriver.Phantomjs(executable_path="/usr/local/bin/phantomjs")
driver = webdriver.Chrome()
driver.get("https://www.douban.com/")
time.sleep(1)
driver.save_screenshot("豆瓣首頁.png")
#輸入賬號
driver.find_element_by_id("form_email").send_keys("username")
#輸入密碼
driver.find_element_by_name("form_password").send_keys("password")
#保存驗證碼的圖片
driver.save_screenshot("驗證碼.png")
#輸入驗證碼
check_code = input("請輸入驗證碼:")
print(r"驗證碼是多少:%s" % check_code)
driver.find_element_by_id("captcha_field").send_keys(check_code)
#點擊登錄按鈕
driver.find_element_by_xpath("//input[@class='bn-submit']").click()
#休眠一下等待登錄成功
time.sleep(3)
#保存登錄成功的快照
driver.save_screenshot("登錄成功.png")
#保存成功登錄好的html到本地 with open("douban.html","w",encoding="utf-8") as f: f.write(driver.page_source) #退出成功 driver.quit()
三、動態頁面模擬點擊(unittest -python測試模塊)
import time
#導入python測試模塊
import unittest
#類名任意,但必須繼承unittest.TestCase
class DouyuTest(unittest.TestCase):
#固定寫法,通常做初始化
def setUp(self):
print("setUp()....")
self.num1 = 1
self.num2 = 1
def testTest(self):
for i in range(1,3):
print("testTest()==",i)
self.num1 += 1
time.sleep(1)
def testTest2(self):
for i in range(1,3):
print("testTest2()==",i)
self.num2 += 1
time.sleep(1)
#固定寫法,但每個自定義方法接收后都會調用一次該方法
def tearDown(self):
print("tearDown()...")
print("num1==",self.num1)
print("num2==", self.num2)
if __name__ == "__main__":
#調用的時候只需要寫上main,固定的調用方式
unittest.main()
四、執行JavaScript代碼from selenium import webdriver
import time
driver = webdriver.PhantomJS()
driver.get("https://movie.douban.com/typerank?type_name=劇情&type=11&interval_id=100:90&action=")
# 向下滾動10000像素
js = "document.body.scrollTop=10000"
time.sleep(5)
#查看頁面快照
driver.save_screenshot("douban.png")
# 執行JS語句
driver.execute_script(js)
time.sleep(5)
#查看頁面快照
driver.save_screenshot("newdouban.png")
#退出瀏覽器
driver.quit()
五、模擬最新無界面瀏覽器(絕招在這,缺陷性能太慢)
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from bs4 import BeautifulSoup
from selenium.webdriver.chrome.options import Options
#不加載圖片,不緩存在硬盤(內存)
SERVICE_ARGS = ['--load-images=false', '--disk-cache=false']
chrome_options = Options()
chrome_options.add_argument("--headless")
# 創建瀏覽器, 添加參數設置為無界面瀏覽器
driver = webdriver.Chrome(service_args=SERVICE_ARGS, chrome_options=chrome_options)
#設置等待時間
waite = WebDriverWait(driver, 5)
driver.get('https://taobao.com/')
def get_page_num():
# 等待搜索框出現
input = waite.until(EC.presence_of_element_located((By.css_SELECTOR, '#q')))
# 輸入美食
input.send_keys('美食')
# 點擊搜索
driver.find_element_by_css_selector("#J_TSearchForm > div > button").click()
# 共計多少頁
text = driver.find_element_by_css_selector('#mainsrp-pager > div > div > div > div[class="total"]').text
data = text[2:6]
get_product_info()
return data
#得到某一個寶貝,商品的大體信息
def get_product_info():
waite.until(EC.presence_of_element_located((By.css_SELECTOR, '#mainsrp-itemlist .items .item')))
# 通過BeautifulSoup取數據
soup = BeautifulSoup(driver.page_source, 'lxml')
#取所有的列表數據
item_lists = soup.select("#mainsrp-itemlist .items .item")
for item_list in item_lists:
item_dict = {}
image = item_list.select('.J_ItemPic.img')[0].attrs["data-src"]
if not image:
image = item_list.select('.J_ItemPic.img')[0].attrs["data-ks-lazyload"]
# 銷售地
location = item_list.select(".location")[0].text
# 價格
price = item_list.select(".price")[0].text
# 商店名稱
shopname = item_list.select(".shopname")[0].text.strip()
# 寶貝名稱
title = item_list.select('a[class="J_ClickStat"]')[0].text.strip()
# 鏈接
data_link = item_list.select('a[class="J_ClickStat"]')[0].attrs["href"]
item_dict["image"] = "https:" + image
item_dict["location"] = location
item_dict["shopname"] = shopname
item_dict["title"] = title
item_dict["data_link"] = "https:" + data_link
item_dict["price"] = price
print(item_dict)
# 切換下一頁
def next_page(page):
print("當前正在加載第%s頁的數據--------" % page)
try:
input = waite.until(EC.presence_of_element_located((By.css_SELECTOR, '#mainsrp-pager > div > div > div > div > input')))
input.clear() # 清空輸入框
# 頁面框添加頁碼
input.send_keys(page)
#找到確定按鈕,點擊確定
driver.find_element_by_css_selector("#mainsrp-pager > div > div > div > div > span.btn.J_Submit").click()
waite.until(EC.text_to_be_present_in_element(
(By.css_SELECTOR, "#mainsrp-pager > div > div > div > ul.items > li.item.active"), str(page)))
except Exception as e:
print(e)
next_page(page)
# 當前切換后的頁面的數據
get_product_info()
def main():
data = get_page_num()
print('總頁數是=', data)
for page in range(2, int(data) + 1):
next_page(page)
# 退出瀏覽器
driver.quit()
if __name__ == '__main__':
main()
