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.driver.get_screenshot_as_file(wholepath)
‘’‘try:
#调用get_screenshot_as_file（filename)方法，对浏览器当前打开页面进行截图保存，并保为D:\photo目录下，命名为photo1.png
result=self.driver.get_screenshot_as_file(r"D:\photo\photo1.png")
print(result)
i=1
srcpath=‘D:\photo’
capturename=’\’+str(i)+’.png’
wholepath=srcpath+capturename
if Path(srcpath).is_dir():
pass
else:
Path(srcpath).mkdir()
while Path(wholepath).exists():
i+=1
capturename=’\’+str(i)+‘png’
wholepath=srcpath+capturename
self.driver.get_screenshot_as_file(wholepath)
except IOError as e:
print(e)’’’

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