liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化超时时间

Browse Source
This commit is contained in:
liuwei
2025-07-11 16:10:35 +08:00
parent de2bb3619a
commit b685ed3b09
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
utils/markdown_to_image.py
View File

@@ -243,7 +243,8 @@ async def html_to_image(html_file, output_image):
# 业务逻辑不变 # 业务逻辑不变
try: try:
page = await browser.new_page() page = await browser.new_page()
await page.goto(f'file://{os.path.abspath(html_file)}') # 增加超时时间到60秒或更长
await page.goto(f'file://{os.path.abspath(html_file)}', timeout=60000)
await page.set_viewport_size({"width": 750, "height": 800}) await page.set_viewport_size({"width": 750, "height": 800})
await page.wait_for_timeout(500) await page.wait_for_timeout(500)
await page.screenshot(path=output_image, full_page=True) await page.screenshot(path=output_image, full_page=True)
@@ -305,6 +306,14 @@ async def convert_md_str_to_image(md_content: str, output_image: str) -> str:
# 将 Markdown 转换为 HTML # 将 Markdown 转换为 HTML
await md_str_to_html(md_content, str(temp_html_path)) await md_str_to_html(md_content, str(temp_html_path))
# 添加更长的等待时间确保文件系统同步
await asyncio.sleep(1.0)
# 检查文件是否存在和可读
if not os.path.exists(str(temp_html_path)):
logger.error(f"HTML文件不存在: {temp_html_path}")
raise FileNotFoundError(f"HTML文件不存在: {temp_html_path}")
# 将 HTML 转换为图片 # 将 HTML 转换为图片
await html_to_image(str(temp_html_path), str(output_image_path)) await html_to_image(str(temp_html_path), str(output_image_path))