Selenium自动化之截屏操作

#截屏，截取整个网站页面

import unittest
import time
from selenium import webdriver

class VisitSogouByIE(unittest.TestCase):
    def setUp(self):
        # 启动IE浏览器
        self.driver = webdriver.Ie(executable_path="g:\\IEDriverServer")

    def test_captureScreenInCurrentWindow(self):
        url = "http://www.sohu.com"
        # 访问搜狗首页
        self.driver.get(url)
        try:
            '''
            调用get_screenshot_as_file(filename)方法，对浏览器当前打开页面
            进行截图,并保为e盘下的screenPicture.png文件。
            '''
            result = self.driver.get_screenshot_as_file(r"e:\screenPicture.png")
            print(result)
        except IOError as e:
            print(e)

    def tearDown(self):
        # 退出IE浏览器
        self.driver.quit()

if __name__ == '__main__':
    unittest.main()