WebDriver內置了測試中捕獲屏幕并保存的方法。

示例腳本：

（1）save_screenshot(filename)：保存屏幕截圖

from selenium import webdriver
from time import sleep,strftime,localtime,time
import os
class TestScreenShot(object):
  def setup(self):
      self.driver = webdriver.Chrome()
      self.driver.get("http://www.baidu.com")
  def test_screen(self):
      self.driver.save_screenshot("testbaidu.png") 
  def teardown(self):
      self.driver.quit()
if __name__ == '__main__':
  shot=TestScreenShot()
  shot.test_screen()

（2）get_screenshot_as_base64()：獲取當前屏幕截圖base64編碼字符串

from selenium import webdriver
from time import sleep,strftime,localtime,time
import os 
class TestScreenShot(object):
  def setup(self):
      self.driver = webdriver.Chrome()
      self.driver.get("http://www.baidu.com")
  def test_screen(self):
      self.driver.save_screenshot("testbaidu.png")
      print(self.driver.get_screenshot_as_base64()) 
  def teardown(self):
      self.driver.quit()
if __name__ == '__main__':
  shot=TestScreenShot()
  shot.test_screen()

運行結果：

（3）get_screenshot_as_file(filename)：獲取當前屏幕截圖，使用完整路徑

from selenium import webdriver
from time import sleep,strftime,localtime,time
import os 
class TestScreenShot(object):
  def setup(self):
      self.driver = webdriver.Chrome()
      self.driver.get("http://www.baidu.com")
  def test_screen(self):
      self.driver.get_screenshot_as_file("testbaidu2.png") 
  def teardown(self):
      self.driver.quit()
if __name__ == '__main__':
  shot=TestScreenShot()
  shot.test_screen()

（4）get_screenshot_as_file(filename)：獲取當前屏幕截圖，使用完整路徑，文件名以時間命名

from selenium import webdriver
from time import sleep,strftime,localtime,time
import os
class TestScreenShot(object):
  def setup(self):
      self.driver = webdriver.Chrome()
      self.driver.get("http://www.baidu.com")
  def test_screen(self):
      # print(self.driver.get_screenshot_as_png())
      # self.driver.get_screenshot_as_file("testbaidu2.png")
      # self.driver.save_screenshot("testbaidu.png")
      # print(self.driver.get_screenshot_as_base64())
      self.driver.find_element_by_id("kw").send_keys("selenium test")
      self.driver.find_element_by_id("su").click()
      str = strftime("%Y-%m-%d-%H-%M-%S",localtime(time()))
      filename = str + ".png"
      path =os.path.abspath("screenshots")
      filepath=path + '/'+filename
      self.driver.get_screenshot_as_file(filepath)
      print(filepath) 
  def teardown(self):
      self.driver.quit()
if __name__ == '__main__':
  shot=TestScreenShot()
  shot.test_screen()

（5）get_screenshot_as_png()：獲取當前屏幕截圖的二進制文件數據

from selenium import webdriver
from time import sleep,strftime,localtime,time
import os 
class TestScreenShot(object):
  def setup(self):
      self.driver = webdriver.Chrome()
      self.driver.get("http://www.baidu.com")
  def test_screen(self):
      print(self.driver.get_screenshot_as_png()) 
  def teardown(self):
      self.driver.quit()
if __name__ == '__main__':
  shot=TestScreenShot()
  shot.test_screen()

運行結果：

以上就是python自動化測試selenium屏幕截圖示例的詳細內容，更多關于selenium屏幕截圖示例的資料請關注服務器之家其它相關文章！

原文鏈接：https://blog.csdn.net/aovenus/article/details/121180311