from selenium import webdriver
import os
import time
def test_currentDate():
'''生成当前日期字符串'''
currentDate = time.localtime()
return '-'.join([str(currentDate.tm_year), str(currentDate.tm_mon), str(currentDate.tm_mday)])
def test_currentTime():
'''生成当前时间字符串'''
currentTime = time.localtime()
return '-'.join([str(currentTime.tm_hour), str(currentTime.tm_min), str(currentTime.tm_sec)])
def test_createDir():
'''创建当前日期文件夹'''
base_dir = 'D:\\PyCharm\\script\\Test_CRM\\report\\image\\'+test_currentDate()
#如果当前目录不存在就创建
if not os.path.exists(base_dir):
os.mkdir(base_dir)
return base_dir
def test_takeScreenshot(driver, imgName):
'''截图函数'''
driver.get_screenshot_as_file(test_createDir()+'\\'+imgName+'.png')
if __name__ == '__main__':
driver = webdriver.Chrome()
driver.get('http://..........')
time.sleep(3)
test_takeScreenshot(driver, test_currentTime())
driver.quit()
执行结束的结构图如下