mirror of
https://github.com/Nezumi-2711/astrbot_plugin_qq_group_daily_analysis.git
synced 2026-09-22 13:38:43 +00:00
134 lines
4.7 KiB
Python
134 lines
4.7 KiB
Python
"""
|
|
HTML模板模块
|
|
使用Jinja2加载外部HTML模板文件
|
|
"""
|
|
|
|
import asyncio
|
|
import os
|
|
import threading
|
|
|
|
from jinja2 import Environment, FileSystemLoader, select_autoescape
|
|
|
|
from ..utils.logger import logger
|
|
|
|
|
|
class HTMLTemplates:
|
|
"""HTML模板管理类"""
|
|
|
|
def __init__(self, config_manager):
|
|
"""初始化Jinja2环境"""
|
|
self.config_manager = config_manager
|
|
# 设置模板根目录
|
|
self.base_dir = os.path.join(os.path.dirname(__file__), "templates")
|
|
# 缓存不同模板的Jinja2环境（多线程安全）
|
|
self._envs = {}
|
|
self._env_lock = threading.Lock()
|
|
|
|
def _get_env_sync(self) -> Environment:
|
|
"""获取当前配置的模板环境（同步版本，供 asyncio.to_thread 调用）"""
|
|
template_name = self.config_manager.get_report_template()
|
|
|
|
# 如果环境已缓存且配置未变（使用锁保证多线程安全）
|
|
with self._env_lock:
|
|
env = self._envs.get(template_name)
|
|
if env is not None:
|
|
return env
|
|
|
|
template_dir = os.path.join(self.base_dir, template_name)
|
|
if not os.path.exists(template_dir):
|
|
logger.warning(f"模板目录不存在: {template_dir}，回退到 scrapbook")
|
|
template_dir = os.path.join(self.base_dir, "scrapbook")
|
|
|
|
env = Environment(
|
|
loader=FileSystemLoader(template_dir),
|
|
autoescape=select_autoescape(["html", "xml"]),
|
|
trim_blocks=True,
|
|
lstrip_blocks=True,
|
|
)
|
|
|
|
# 使用双重检查锁定，避免在高并发下重复创建相同 template_name 的 env
|
|
with self._env_lock:
|
|
existing = self._envs.get(template_name)
|
|
if existing is not None:
|
|
return existing
|
|
self._envs[template_name] = env
|
|
|
|
return env
|
|
|
|
async def _get_env_async(self) -> Environment:
|
|
"""获取当前配置的模板环境（异步版本）"""
|
|
return await asyncio.to_thread(self._get_env_sync)
|
|
|
|
def _get_env(self) -> Environment:
|
|
"""获取当前配置的模板环境（同步版本，向后兼容）"""
|
|
return self._get_env_sync()
|
|
|
|
def _read_template_file_sync(self, filename: str) -> str:
|
|
"""同步读取模板文件内容"""
|
|
with open(filename, encoding="utf-8") as f:
|
|
return f.read()
|
|
|
|
async def get_image_template_async(self) -> str:
|
|
"""获取图片报告的HTML模板（异步版本，返回原始模板字符串）"""
|
|
try:
|
|
env = await self._get_env_async()
|
|
template = env.get_template("image_template.html")
|
|
return await asyncio.to_thread(
|
|
self._read_template_file_sync, template.filename
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"加载图片模板失败: {e}")
|
|
return ""
|
|
|
|
def get_image_template(self) -> str:
|
|
"""获取图片报告的HTML模板（同步版本，向后兼容）"""
|
|
try:
|
|
env = self._get_env()
|
|
template = env.get_template("image_template.html")
|
|
with open(template.filename, encoding="utf-8") as f:
|
|
return f.read()
|
|
except Exception as e:
|
|
logger.error(f"加载图片模板失败: {e}")
|
|
return ""
|
|
|
|
async def get_pdf_template_async(self) -> str:
|
|
"""获取PDF报告的HTML模板（异步版本，返回原始模板字符串）"""
|
|
try:
|
|
env = await self._get_env_async()
|
|
template = env.get_template("pdf_template.html")
|
|
return await asyncio.to_thread(
|
|
self._read_template_file_sync, template.filename
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"加载PDF模板失败: {e}")
|
|
return ""
|
|
|
|
def get_pdf_template(self) -> str:
|
|
"""获取PDF报告的HTML模板（同步版本，向后兼容）"""
|
|
try:
|
|
env = self._get_env()
|
|
template = env.get_template("pdf_template.html")
|
|
with open(template.filename, encoding="utf-8") as f:
|
|
return f.read()
|
|
except Exception as e:
|
|
logger.error(f"加载PDF模板失败: {e}")
|
|
return ""
|
|
|
|
def render_template(self, template_name: str, **kwargs) -> str:
|
|
"""渲染指定的模板文件
|
|
|
|
Args:
|
|
template_name: 模板文件名
|
|
**kwargs: 传递给模板的变量
|
|
|
|
Returns:
|
|
渲染后的HTML字符串
|
|
"""
|
|
try:
|
|
env = self._get_env()
|
|
template = env.get_template(template_name)
|
|
return template.render(**kwargs)
|
|
except Exception as e:
|
|
logger.error(f"渲染模板 {template_name} 失败: {e}")
|
|
return ""
|