selenium模块提供了截图函数get_screenshot_as_file()来截取当前窗口
# 定义截图函数 采取相对路径的方式将测试截图保存到.\reports\image\中
from selenium import webdriver #导入webdriver模块
import os
def insert_img(driver,file_name):
# 获取当前文件所在的绝对路径 E:/learn/erds_test/erds_ui/ui_test/models
base_dir=os.path.dirname(__file__)
# 转换成str格式
base_dir=str(base_dir)
# 以'/ui_test'将路径拆分两部分,获取第一部分:E:/learn/erds_test/erds_ui
base=base_dir.split('/ui_test')[0]
# 设置截图保存的路径 E:/learn/erds_test/erds_ui/reports/image/+file_name
file_path=base+'/reports/image/'+file_name
# 截图
image=driver.get_screenshot_as_file(file_path)
if __name__ == '__main__':
#打开浏览器,并截图
driver=webdriver.Chrome()
driver.get("https://rzrk.thinkfunds.cn:8001/login/")
driver.maximize_window()
#E:/learn/erds_test/erds_ui/reports/image/1.png
insert_img(driver,"1.png")
driver.quit()
运行程序:
打开1.png: