selenium+webdriver框架 bbs项目function.py解释：

selenium+webdriver框架 bbs项目function.py解释：

from selenium import webdriver
import os

def insert_img(driver,file_name):
base_dir = os.path.dirname(os.path.dirname(file)) #获取文件所在目录的完整路径
print(base_dir)
base_dir = str(base_dir) #将地址转为字符串
print(base_dir)
base_dir = base_dir.replace(’\’,’/’)
base = base_dir.split(’/test_case’)[0] # 以/test_case为分隔符，取第一个分割的部分
print(base)
file_path = base + “/report/image” +file_name
driver.get_screenshot_as_file(file_path) #保存截图
#以下为功能实验，实际项目时可删除
if name == ‘main’:
driver = webdriver.Chrome()
driver.get(“https://www.baidu.com”)
insert_img(driver,‘baidu.jpg’)
driver.quit()