Python+Selenium笔记（十六）屏幕截图

 (一) 方法

方法	简单说明
save_screenshot(filename)	获取当前屏幕截图并保存为指定文件 filename:路径/文件名
get_screenshot_as_base64()	获取当前屏幕截图base64编码字符串
get_screenshot_as_file(filename)	获取当前屏幕截图，使用完整的路径。如果有任何IOerror，返回false，否则返回true filename:路径/文件名
get_screenshot_as_png()	获取当前屏幕截图的二进制路径

(二) 屏幕截图示例（代码肯定会报错的哈，这里是为了演示测试失败时，自动截图的功能（比如有的时候弹出不应该有的警告或者错误信息什么的，就可以截图保留记录)）

from selenium import webdriver
import unittest
from selenium.common.exceptions import NoSuchElementException
class Login(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.driver.implicitly_wait(5)
        cls.driver.maximize_window()
        cls.driver.get("https://www.cnblogs.com/")

    def test_login(self):
        try:
            login_area = self.driver.find_element_by_css_selector('#login_area')
            register = login_area.find_element_by_link_text('注册111')
            register.click()
        except NoSuchElementException:
            #找不到元素时，截图
            self.driver.save_screenshot('image/test_login/test_login_area.png')
            raise

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

相应的文件目录下可以看到图片