如何在HTMLTestRunner报表中包含截图？

from selenium import webdriver

# 创建浏览器实例
driver = webdriver.Chrome()

# 打开网页
driver.get("http://www.example.com")

# 进行测试操作
# ...

# 如果测试失败，进行截图
if test_failed:
    driver.save_screenshot("screenshot.png")

# 关闭浏览器
driver.quit()

import HTMLTestRunner

# 创建测试套件
suite = unittest.TestSuite()

# 将测试用例添加到测试套件中
# ...

# 创建HTMLTestRunner实例
runner = HTMLTestRunner.HTMLTestRunner(output="report.html")

# 运行测试套件并生成报告
runner.run(suite)

# 在报告中添加截图
with open("report.html", "r+") as file:
    content = file.read()
    # 将截图的HTML代码插入到报告中的适当位置
    content = content.replace("</body>", '<img src="screenshot.png" alt="screenshot"><br></body>')
    file.seek(0)
    file.write(content)
    file.truncate()