python+selenium 屏幕截屏

#encoding=utf-8
import unittest
import time
import chardet
from selenium import webdriver
 
class VisitSogouByIE(unittest.TestCase):

    def setUp(self):
        #启动IE浏览器
        self.driver = webdriver.Firefox(executable_path = "e:\\geckodriver")
        # self.driver = webdriver.Ie(executable_path = "e:\\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:\screenPicture2.png")
            print (result)
        except IOError as e:
            print (e)


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

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