Python selenium 截图-全屏和指定元素

UI自动化测试中，为了更好保留错误，配合复现问题，我们都会采取截图的方式。所以总结了下目前常用的两种截图方式，第一种是全屏截图，第二种是指定元素截图。
首先全屏截图，代码如下：

driver = webdriver.Chrome()
driver.save_screenshot('all.png') #保存在当前路径

然后是指定元素截图，这种方式需要我们首先定位到元素，然后再进行截图，代码如下：

driver = webdriver.Chrome()
element = driver.find_element_by_id("checkimg")
element.screenshot("11.png") 

网上搜索到的很多截图都是需要根据元素坐标进行截图，比较复杂。这种就使用起来很方便了。指定元素的截图还可以用作识别一些图片验证码时截取验证码，这样就可以实现在线上环境做一些回归的测试