截图主要用到的方法是:
get_screenshot_as_file()
实验步骤:
1,打开百度,进行截图
2,在输入框中输入 selenium 关键字,进行截图
3,点击“百度一下”,进行截图
from selenium import webdriver import time driver = webdriver.Chrome() driver.get("https://www.baidu.com/") driver.maximize_window() # 截取百度首页 driver.get_screenshot_as_file("baidu.png") # 输入关键字 selenium 后截图 driver.find_element_by_id("kw").send_keys("selenium") time.sleep(2) driver.get_screenshot_as_file("input_keyword.png") # 点击 百度一下 之后截图 driver.find_element_by_id("su").click() time.sleep(2) driver.get_screenshot_as_file("result.png")
操作完成后,会在当前目录保存图片
总结:
1.get_screenshot_as_file()函数截图成功后会返回True,出现IOError异常,会返回False
2.存放图片的路径,可以是相对路径,也可以是绝对路径
3.只能保存png格式
4.在执行用例中,可以在try..except语句中结合使用,用例报异常,可以进行截图