python3.6.3 对当前浏览器窗口截屏

在这个过程中只需要注意一个问题，就是python3.0的异常捕获和3.0以下版本有所不同
import unittest
from selenium import webdriver

class MyTestCase(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome(executable_path=’/python/driver/chromedriver’)
def testcaptureScreen(self):
url=“http://www.sogou.com”
self.driver.get(url)
try:
result=self.driver.get_screenshot_as_file(r"D:\photo\photo1.png")
print(result)
except IOError as e:
print(e)

if name == ‘main’:
unittest.main()