一.整个页面截图
代码语言:javascript复制driver = webdriver.Chrome()
driver.get(url)
diver.save_screenshot('保存路径')
二.局部截图
代码语言:javascript复制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('保存路径'))