selenium中整个页面的截图及局部截图（方法分析）
from selenium import webdriverfrom PIL import Imagefrom time import sleepimport timedef 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'] + leftheight = ce.size['height'] + topim = 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 webdriverimport timedef 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()
