Python爬虫之Selenium实现窗口截图

前言：由程序去执行的操作不允许有任何误差，有些时候在测试的时候未出现问题，但是放到服务器上就会报错，而且打印的错误信息并不十分明确。这时，我在想如果在脚本执行出错的时候能对当前窗口截图保存，那么通过图片就可以非常直观地看出出错的原因。WebDriver提供了截图函数get_screenshot_as_file()来截取当前窗口。

本章中用到的关键方法如下：

get_screenshot_as_file()：截图

from selenium import webdriver

driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')

# 截图，图片后缀最好为.png，如果是其他的执行的时候会有警告，但不会报错
driver.get_screenshot_as_file("D:\\baidu_index.png")

driver.quit()

实例：

#窗口截图操作
#coding utf-8

from selenium import webdriver
from time import sleep

driver = webdriver.Firefox()

url = "http://www.baidu.com"

driver.get(url)

driver.find_element_by_id('kw').send_keys('selenium python')
driver.find_element_by_id('su').click()

sleep(2)

driver.get_screenshot_as_file("D:\\baidu_error.jpg")

driver.quit()

运行结果：

有个warning，什么意思呢，就是说截图最好是使用 .png格式的图片，而我的代码中使用的是.jpg格式，但是，不影响最终结果

 file type. It should end with a `.png` extension
  "type. It should end with a `.png` extension", UserWarning)　　

网页截图： – 不出意外，渣度第一时间给你推送广告！！呵呵呵！-