selenium中整个页面的截图及局部截图（方法分析）

selenium中整个页面的截图及局部截图（方法分析）

from selenium import webdriver
from PIL import Image
from time import sleep
import time
def test1():
    browser = webdriver.Chrome()
    #打开一个注册界面
    browser.get('http://localhost:8080/jpress/user/register')
    #最大化浏览器
    browser.maximize_window()
	#下面三行是为截图加上时间
    t = time.time()
    picture_name1 = str(t) + '.png'
    #下面一行为保存整个页面的截图
    browser.save_screenshot(picture_name1)
    ce = browser.find_element_by_id('captchaimg')#获取验证码区域左上角的位置
    print(ce.location)#打印ce的x,y坐标
    left = ce.location['x']
    top = ce.location['y']
    right = ce.size['width'] + left
    height = ce.size['height'] + top
    im = Image.open(picture_name1)#打开刚才保存的截图
    #抠图
    # sleep(2)
    img = im.crop((left,top,right,height))#截取验证码部分
    t = time.time()
    picture_name2 = str(t) + '.png'
    img.save(picture_name2)#保存
    browser.close()#关闭浏览器

上面的代码是视频教程上的源码，但是有点复杂了，而且想要得到局部截图的时候，总是得不到预期的结果，好在经过自己的一番查找，终于找到了一种比较好的方法。
整个页面的截图和上面一样，没什么问题，关键是局部截图，只需使用screenshot方法即可，如下

from selenium import webdriver
import time
def test2():
	browser = webdriver.Chrome()
	#打开一个注册界面
    browser.get('http://localhost:8080/jpress/user/register')
    #最大化浏览器
    browser.maximize_window()
	t = time.time()
    # t = t.strftime('%Y-%m-%d %H:%M:%S')思考一下怎么改为这种时间格式命名？
    picture_name2 = str(t) + '.png'
    #下面是显示等待的方法
    WebDriverWait(browser, 10, 0.5).until(lambda el: browser.find_element_by_id('captchaimg'))
    browser.find_element_by_id('captchaimg').screenshot(picture_name2)
    browser.close()