爬虫selenium中截图

一.整个页面截图

driver = webdriver.Chrome()
driver.get(url)
diver.save_screenshot('保存路径')

二.局部截图

driver = webdriver.Chrome()
driver.get(url)
#比如获得类名为img的元素截图
#首先要等待他加载出来2秒一次
WebDriverWait(driver, 2,1).until(EC.presence_of_element_located((By.CLASS_NAME, 'img')))

#获取元素保存
img = driver.find_element_by_class_name('img').screenshot('保存路径'))