Selenium系列（八） - 截取完整頁面和截取指定元素並保存為圖片

如果你還想從頭學起Selenium，可以看看這個系列的文章哦！

https://www.cnblogs.com/poloyy/category/1680176.html

 

其次，如果你不懂前端基礎知識，需要自己去補充哦，博主暫時沒有總結（雖然我也會，所以我學selenium就不用復習前端了哈哈哈...）

 

截圖操作

• 截取整個頁面
• 截取指定元素

只有兩個方法，比較簡單，直接上代碼

# !/usr/bin/env python
# -*- coding: utf-8 -*-

"""
__title__  =
__Time__   = 2020/3/25 17:52
__Author__ = 小菠蘿測試筆記
__Blog__   = https://www.cnblogs.com/poloyy/
"""

from selenium import webdriver

driver = webdriver.Chrome("../resources/chromedriver.exe")

driver.get("https://www.baidu.com")
driver.maximize_window()

# 截取整個頁面
driver.get_screenshot_as_file("test.png")
driver.save_screenshot("tests.png")

# 找到搜索框
inputElement = driver.find_element_by_id("kw")

# 截取搜索框元素
inputElement.screenshot("inputElement.png")

知識點

• get_screenshot_as_file()、save_screenshot() 效果一樣， save_screenshot() 最終調用的函數就是前者
• 后綴名建議寫 .png ，否則會有Warning提示