python3.0 + selenium命名截图并保存到指定文件夹

1.被注释掉的是截屏到指定的文件下，指定保存文件名称
2.没有被注释的是不指定文件名称截图并保存到指定的文件夹下
import unittest
from selenium import webdriver
from pathlib import Path
import os

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)
i = 1
scrpath = ‘D:\potho’ # 指定的保存目录
capturename = ‘\’ + str(i) + ‘.png’ # 自定义命名截图
wholepath = scrpath + capturename
if Path(scrpath).is_dir(): # 判断文件夹路径是否已经存在
pass
else:
Path(scrpath).mkdir() # 如果不存在，创建文件夹
while Path(wholepath).exists(): # 判断文件是否已经存在，也可使用is_file()判断
i += 1
capturename = ‘\’ + str(i) + ‘.png’
wholepath = scrpath + capturename
self.dr