124 lines
4.0 KiB
Python
124 lines
4.0 KiB
Python
from pathlib import Path
|
||
|
||
from jinja2 import Template
|
||
from playwright.async_api import async_playwright
|
||
|
||
from zhenxun.configs.path_config import DATA_PATH
|
||
from zhenxun.services.log import logger
|
||
|
||
from ..utils.config import g_sResourcePath
|
||
|
||
|
||
class CHelpManager:
|
||
@classmethod
|
||
def renderHtmlToFile(
|
||
cls, path: Path | str, context: dict, output: Path | str
|
||
) -> None:
|
||
"""
|
||
使用 Jinja2 渲染 HTML 模板并保存到指定文件，会自动创建父目录
|
||
|
||
Args:
|
||
path (str): 模板 HTML 路径
|
||
context (dict): 用于渲染的上下文字典
|
||
output (str): 输出 HTML 文件路径
|
||
"""
|
||
templatePath = str(path)
|
||
outputPath = str(output)
|
||
|
||
templateStr = Path(templatePath).read_text(encoding="utf-8")
|
||
template = Template(templateStr)
|
||
rendered = template.render(**context)
|
||
|
||
# 自动创建目录
|
||
Path(outputPath).parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
Path(outputPath).write_text(rendered, encoding="utf-8")
|
||
|
||
@classmethod
|
||
async def screenshotHtmlToBytes(cls, path: str) -> bytes:
|
||
"""
|
||
使用 Playwright 截图本地 HTML 文件并返回 PNG 图片字节数据
|
||
|
||
Args:
|
||
path (str): 本地 HTML 文件路径
|
||
|
||
Returns:
|
||
bytes: PNG 图片的原始字节内容
|
||
"""
|
||
async with async_playwright() as p:
|
||
browser = await p.chromium.launch()
|
||
page = await browser.new_page(
|
||
viewport={"width": 1200, "height": 900}, device_scale_factor=1
|
||
)
|
||
file_url = Path(path).resolve().as_uri()
|
||
await page.goto(file_url, wait_until="networkidle")
|
||
await page.evaluate("""() => {
|
||
return new Promise(r => setTimeout(r, 200));
|
||
}""")
|
||
image_bytes = await page.screenshot(full_page=True)
|
||
await browser.close()
|
||
|
||
return image_bytes
|
||
|
||
@classmethod
|
||
async def screenshotSave(
|
||
cls, path: str, save: str, width: int, height: int
|
||
) -> None:
|
||
"""
|
||
使用 Playwright 渲染本地 HTML 并将截图保存到指定路径
|
||
|
||
Args:
|
||
path (str): HTML 文件路径
|
||
save (str): PNG 保存路径（如 output/image.png）
|
||
width (int): 图片宽度
|
||
height (int): 图片高度
|
||
"""
|
||
async with async_playwright() as p:
|
||
browser = await p.chromium.launch()
|
||
page = await browser.new_page(
|
||
viewport={"width": width, "height": height}, device_scale_factor=1
|
||
)
|
||
|
||
file_url = Path(path).resolve().as_uri()
|
||
await page.goto(file_url, wait_until="networkidle")
|
||
await page.evaluate("""() => {
|
||
return new Promise(r => setTimeout(r, 200));
|
||
}""")
|
||
|
||
# 确保保存目录存在
|
||
Path(save).parent.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 截图并保存到本地文件
|
||
await page.screenshot(path=save, full_page=True)
|
||
await browser.close()
|
||
|
||
@classmethod
|
||
async def createHelpImage(cls) -> bool:
|
||
templatePath = g_sResourcePath / "html/help.html"
|
||
outputPath = g_sResourcePath / "temp_html/help.html"
|
||
savePath = DATA_PATH / "farm_res/html/help.png"
|
||
|
||
context = {
|
||
"main_title": "真寻农场帮助菜单",
|
||
"subtitle": "[]中为可选参数",
|
||
"page_title": "真寻农场帮助菜单",
|
||
"font_family": "MyFont",
|
||
"contents": [
|
||
{"title": "主要指令", "commands": ["指令A", "指令B"]},
|
||
{"title": "B", "commands": ["指令D", "指令E", "指令M", "指令i"]},
|
||
],
|
||
}
|
||
|
||
try:
|
||
cls.renderHtmlToFile(templatePath, context, outputPath)
|
||
|
||
bytes = await cls.screenshotSave(str(outputPath), str(savePath), 1500, 2300)
|
||
except Exception as e:
|
||
logger.warning("绘制农场帮助菜单失败", e=e)
|
||
return False
|
||
|
||
return True
|
||
|
||
|
||
g_pHelpManager = CHelpManager()
|