nonebot-plugin-htmlrender 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

nonebot_plugin_htmlrender/data_source.py

@@ -144,6 +144,7 @@ async def read_tpl(path: str) -> str:
 async def template_to_html(
     template_path: str,
     template_name: str,
+    filters: Optional[Dict[str, Any]] = None,
     **kwargs,
 ) -> str:
     """使用jinja2模板引擎通过html生成图片
@@ -151,6 +152,7 @@ async def template_to_html(
     Args:
         template_path (str): 模板路径
         template_name (str): 模板名
+        filters (Optional[Dict[str, Any]]): 自定义过滤器
         **kwargs: 模板内容
     Returns:
         str: html
@@ -160,6 +162,12 @@ async def template_to_html(
         loader=jinja2.FileSystemLoader(template_path),
         enable_async=True,
     )
+
+    if filters:
+        for filter_name, filter_func in filters.items():
+            template_env.filters[filter_name] = filter_func
+            logger.debug(f"加载自定义过滤器 {filter_name}")
+    
     template = template_env.get_template(template_name)
 
     return await template.render_async(**kwargs)
@@ -207,6 +215,7 @@ async def template_to_pic(
     template_path: str,
     template_name: str,
     templates: Dict[Any, Any],
+    filters: Optional[Dict[str, Any]] = None,
     pages: Optional[Dict[Any, Any]] = None,
     wait: int = 0,
     type: Literal["jpeg", "png"] = "png",  # noqa: A002
@@ -219,6 +228,7 @@ async def template_to_pic(
         template_path (str): 模板路径
         template_name (str): 模板名
         templates (Dict[Any, Any]): 模板内参数 如: {"name": "abc"}
+        filters (Optional[Dict[str, Any]]): 自定义过滤器
         pages (Optional[Dict[Any, Any]]): 网页参数 Defaults to
             {"base_url": f"file://{getcwd()}", "viewport": {"width": 500, "height": 10}}
         wait (int, optional): 网页载入等待时间. Defaults to 0.
@@ -238,6 +248,12 @@ async def template_to_pic(
         loader=jinja2.FileSystemLoader(template_path),
         enable_async=True,
     )
+
+    if filters:
+        for filter_name, filter_func in filters.items():
+            template_env.filters[filter_name] = filter_func
+            logger.debug(f"加载自定义过滤器 {filter_name}")
+
     template = template_env.get_template(template_name)
 
     return await html_to_pic(

{nonebot_plugin_htmlrender-0.3.3.dist-info → nonebot_plugin_htmlrender-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-htmlrender
-Version: 0.3.3
+Version: 0.3.5
 Summary: 通过浏览器渲染图片
 Author-Email: kexue <xana278@foxmail.com>
 License: MIT License

{nonebot_plugin_htmlrender-0.3.3.dist-info → nonebot_plugin_htmlrender-0.3.5.dist-info}/RECORD

@@ -1,10 +1,10 @@
-nonebot_plugin_htmlrender-0.3.3.dist-info/METADATA,sha256=6IsAtzmoYZ1E8y0VEG8_WaXWLUvtRouZ0sUmoJTBe3w,5987
-nonebot_plugin_htmlrender-0.3.3.dist-info/WHEEL,sha256=mbxFTmdEUhG7evcdMkR3aBt9SWcoFBJ4CDwnfguNegA,90
-nonebot_plugin_htmlrender-0.3.3.dist-info/licenses/LICENSE,sha256=4HHgpu3ihnBZ9AVZDgOzGnAJRt8W7IQo5VSBEJKRr5g,1062
+nonebot_plugin_htmlrender-0.3.5.dist-info/METADATA,sha256=I-ymdXT1rLAGXQkIt7HIiChwZgp4rFUaV_1XdTzDIQk,5987
+nonebot_plugin_htmlrender-0.3.5.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
+nonebot_plugin_htmlrender-0.3.5.dist-info/licenses/LICENSE,sha256=4HHgpu3ihnBZ9AVZDgOzGnAJRt8W7IQo5VSBEJKRr5g,1062
 nonebot_plugin_htmlrender/__init__.py,sha256=JBxs90TaroVrq5EFo0Yib5_c7OmBCeUqOyQuJvFiXrU,1349
 nonebot_plugin_htmlrender/browser.py,sha256=0AfOO7P3JVv14qkaN8iyvQmpRnIdr4Slu1-SYVGjKpc,3335
 nonebot_plugin_htmlrender/config.py,sha256=M3JLglX0yzo7slebS4w3riciEsqmfruzrQ9sPFz1yzM,357
-nonebot_plugin_htmlrender/data_source.py,sha256=3llCipPPk4wJSwCdqAbOcUBQ3ie3ad7lQQ0FJWiFJMA,8605
+nonebot_plugin_htmlrender/data_source.py,sha256=Fy4VEEINXnqNwyfhFb1-iZvAYpVHSEil3R_O9JQV7fo,9233
 nonebot_plugin_htmlrender/templates/github-markdown-light.css,sha256=bWV9QiirgHYgauZBcR9YJ3IvEoA3fP2Cf5aOdshBVrI,17297
 nonebot_plugin_htmlrender/templates/katex/katex.min.b64_fonts.css,sha256=8KXk7iVhFZ-h2LuWlhI6Eq2za2vuDW-n4MrHQJ2eRAU,1458204
 nonebot_plugin_htmlrender/templates/katex/katex.min.js,sha256=cXQ4tsltOzinSxpNqnPldAdfpstIPXBDKihVb7ccZL0,270288
@@ -13,4 +13,4 @@ nonebot_plugin_htmlrender/templates/markdown.html,sha256=y-_F5YMVLHceU9nuuEAtw2t
 nonebot_plugin_htmlrender/templates/pygments-default.css,sha256=lwkE40qSsvPe52nGAZW1luJ7qNeIWpgw7vDX-MZpCDA,4889
 nonebot_plugin_htmlrender/templates/text.css,sha256=zhb6Yd3zndvojU4-qQR46dRg9CsCRvmyWU1-Uhm-2LQ,117
 nonebot_plugin_htmlrender/templates/text.html,sha256=QhoOXE1111YNfxu_b6OYhgm4ygFpIvW0VI6VwYo4Fig,220
-nonebot_plugin_htmlrender-0.3.3.dist-info/RECORD,,
+nonebot_plugin_htmlrender-0.3.5.dist-info/RECORD,,

{nonebot_plugin_htmlrender-0.3.3.dist-info → nonebot_plugin_htmlrender-0.3.5.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: pdm-backend (2.3.2)
+Generator: pdm-backend (2.3.3)
 Root-Is-Purelib: true
 Tag: py3-none-any