截图两种方法:
get_screenshot_as_file(filepath)
save_screenshot(filepath)
我用的第二种:save_screenshot
代码如下:
def snip_screenshot(driver: WebDriver):
screenshot = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))+ os.sep + "screenshot" # 截图文件所在目录(完全路径)
now_time = time.strftime("%Y%m%d%H%M%S", time.localtime()) + "_" + str(randint(100, 999)) # 截图文件命名(当前时间_3位随机数)
pic = screenshot + os.sep + "%s.png" % now_time # 截图文件完全路径
# driver.get_screenshot_as_file(pic)
print("截图结果:", driver.save_screenshot(pic)) # 截图并打印结果
return pic
结果:
总是截图失败
原因:
screenshot值中的“//screenshot”目录未被创建,代码不会自动创建此目录,需要手动创建。
手动创建//screenshot文件夹后,再运行,截图就成功了