nonebot-plugin-htmlrender 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.whl

nonebot_plugin_htmlrender/browser.py

@@ -61,7 +61,7 @@ async def init_browser(**kwargs) -> Browser:
 @asynccontextmanager
 async def get_new_page(device_scale_factor: float = 2, **kwargs) -> AsyncIterator[Page]:
     """
-    获取一个新的页面的上下文管理器。
+    获取一个新的页面的上下文管理器, 这里的 page 默认使用设备缩放因子为 2。
 
     Args:
         device_scale_factor (float): 设备缩放因子。

nonebot_plugin_htmlrender/data_source.py

@@ -284,17 +284,32 @@ async def template_to_pic(
 async def capture_element(
     url: str,
     element: str,
-    timeout: float = 0,
-    type: Literal["jpeg", "png"] = "png",
-    quality: Union[int, None] = None,
-    screenshot_timeout: Optional[float] = 30_000,
-    **kwargs,
+    page_kwargs: Optional[dict] = None,
+    goto_kwargs: Optional[dict] = None,
+    screenshot_kwargs: Optional[dict] = None,
 ) -> bytes:
-    async with get_new_page(**kwargs) as page:
-        page.on("console", lambda msg: logger.debug(f"浏览器控制台: {msg.text}"))
-        await page.goto(url, timeout=timeout)
-        return await page.locator(element).screenshot(
-            type=type,
-            quality=quality,
-            timeout=screenshot_timeout,
+    """捕获网页中指定元素的截图, 通过CSS选择器或XPath表达式指定元素。
+
+    Args:
+        url: 目标网页URL
+        element: CSS选择器或XPath表达式
+        page_kwargs: 传递给get_new_page的参数
+        goto_kwargs: 传递给page.goto方法的额外参数
+        screenshot_kwargs: 传递给screenshot方法的额外参数
+
+    Returns:
+        bytes: 元素截图数据
+    """
+    page_kwargs = page_kwargs or {}
+    goto_kwargs = goto_kwargs or {}
+    screenshot_kwargs = screenshot_kwargs or {}
+
+    async with get_new_page(**page_kwargs) as page:
+        page.on(
+            "console",
+            lambda msg: logger.opt(colors=True).debug(
+                f"<cyan>[Browser Console]</cyan> {msg.text}"
+            ),
         )
+        await page.goto(url, **goto_kwargs)
+        return await page.locator(element).screenshot(**screenshot_kwargs)

{nonebot_plugin_htmlrender-0.6.1.dist-info → nonebot_plugin_htmlrender-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nonebot-plugin-htmlrender
-Version: 0.6.1
+Version: 0.6.2
 Summary: 通过浏览器渲染图片
 Project-URL: Homepage, https://github.com/kexue-z/nonebot-plugin-htmlrender
 Project-URL: Bug Tracker, https://github.com/kexue-z/nonebot-plugin-htmlrender/issues

{nonebot_plugin_htmlrender-0.6.1.dist-info → nonebot_plugin_htmlrender-0.6.2.dist-info}/RECORD

@@ -1,9 +1,9 @@
 nonebot_plugin_htmlrender/__init__.py,sha256=fzqf-XaZ_lxzr18kHgxD16Zq4BaU2e-UidoHGVzDyHg,1825
-nonebot_plugin_htmlrender/browser.py,sha256=jjj43q82-zMZsmR2gJx9tymwoG8piODuaZz8Sp6FUAY,4873
+nonebot_plugin_htmlrender/browser.py,sha256=zMdFvDQrZ1sZtV6c3Kh7RAwj4FBcoWA85jt147-4jaA,4925
 nonebot_plugin_htmlrender/compat.py,sha256=eeAOMbJB0qMYkUmO8pbM11hxGCZuVUQ2sYEy8XTV81g,826
 nonebot_plugin_htmlrender/config.py,sha256=-fywpLhQHVCZ2s6c3wG5ywREZbUFcW9ZKrrRj8ptEfQ,2272
 nonebot_plugin_htmlrender/consts.py,sha256=eMujEJaBDsQw8wwIJH3sb3twS0eo7jCiQd-TD671950,930
-nonebot_plugin_htmlrender/data_source.py,sha256=OBzIYZHTIzffD9Jsl7GBkpc3jYuYaMmJWusEFi_ckyE,9757
+nonebot_plugin_htmlrender/data_source.py,sha256=rFHyW0JYFRI9bhnbGHRw8c7sMWGMkvpb66LziFn2Ns0,10270
 nonebot_plugin_htmlrender/install.py,sha256=ejfoPkyixHN83ldFhNihCGA25IjOLqtncEjEbh_C6Sw,8186
 nonebot_plugin_htmlrender/process.py,sha256=BqU0GBj_NYl5mSxyVIpF35PApcQDjiF3jT6hdhLqniA,4526
 nonebot_plugin_htmlrender/signal.py,sha256=sOOqIns5QIi5404mZoEtt3ssliXS3IGndkt0WMreNe0,3920
@@ -17,7 +17,7 @@ nonebot_plugin_htmlrender/templates/katex/katex.min.b64_fonts.css,sha256=8KXk7iV
 nonebot_plugin_htmlrender/templates/katex/katex.min.js,sha256=cXQ4tsltOzinSxpNqnPldAdfpstIPXBDKihVb7ccZL0,270288
 nonebot_plugin_htmlrender/templates/katex/mathtex-script-type.min.js,sha256=d-R93Fs6-ZBEU3VEH7u_qz-b02iCPKEPmLJEgRf2-MU,1290
 nonebot_plugin_htmlrender/templates/katex/mhchem.min.js,sha256=8MoD3xlLjD1gF_9FXbag75iFeQVmP6MRps3teIsVNAs,33730
-nonebot_plugin_htmlrender-0.6.1.dist-info/METADATA,sha256=QbO5wbRG-EgEbL4uzFvJTE4KbKKzkK_n6_s_hadSdBI,6721
-nonebot_plugin_htmlrender-0.6.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nonebot_plugin_htmlrender-0.6.1.dist-info/licenses/LICENSE,sha256=4HHgpu3ihnBZ9AVZDgOzGnAJRt8W7IQo5VSBEJKRr5g,1062
-nonebot_plugin_htmlrender-0.6.1.dist-info/RECORD,,
+nonebot_plugin_htmlrender-0.6.2.dist-info/METADATA,sha256=7S6E1LOFfirWqx_jpJyzesvje_g23NKUJuRnPSX_hzM,6721
+nonebot_plugin_htmlrender-0.6.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nonebot_plugin_htmlrender-0.6.2.dist-info/licenses/LICENSE,sha256=4HHgpu3ihnBZ9AVZDgOzGnAJRt8W7IQo5VSBEJKRr5g,1062
+nonebot_plugin_htmlrender-0.6.2.dist-info/RECORD,,