nonebot-plugin-htmlrender 0.6.8__tar.gz → 0.7.0a1__tar.gz

nonebot_plugin_htmlrender-0.7.0a1/.github/workflows/_test-os-arch.yml

@@ -0,0 +1,67 @@
+on:
+  workflow_call:
+    inputs:
+      python-version:
+        required: true
+        type: string
+
+jobs:
+  test:
+    name: Test (Py ${{ inputs.python-version }} — ${{ matrix.os_label }})
+    runs-on: ${{ matrix.runs_on }}
+    strategy:
+      fail-fast: true
+      matrix:
+        include:
+          - runs_on: ubuntu-latest
+            os_label: ubuntu-latest x64
+            arch: x64
+            platform: linux
+
+          - runs_on: macos-latest
+            os_label: macos-latest arm64
+            arch: arm64
+            platform: macos
+
+          - runs_on: windows-latest
+            os_label: windows-latest x64
+            arch: x64
+            platform: windows
+
+          - runs_on: ubuntu-24.04-arm
+            os_label: ubuntu-24.04-arm arm64
+            arch: arm64
+            platform: linux
+
+    env:
+      OS: ${{ matrix.os_label }}
+      PYTHON_VERSION: ${{ inputs.python-version }}
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v6
+        with:
+          python-version: ${{ inputs.python-version }}
+          enable-cache: true
+          cache-suffix: ${{ matrix.runs_on }}-${{ inputs.python-version }}
+
+      - name: Sync deps
+        run: uv sync --locked --all-extras --dev
+
+      - name: Install Playwright (Linux)
+        if: matrix.platform == 'linux'
+        run: uv run playwright install --with-deps && uv run playwright install --with-deps
+      - name: Install Playwright (non-Linux)
+        if: matrix.platform != 'linux'
+        run: uv run playwright install && uv run playwright install
+
+      - name: Run tests
+        run: uv run pytest -s -n auto
+
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v5
+        with:
+          env_vars: OS,PYTHON_VERSION
+          verbose: true

nonebot_plugin_htmlrender-0.7.0a1/.github/workflows/codecov.yml

@@ -0,0 +1,20 @@
+name: Code Coverage
+
+on:
+  push:
+    branches: [ master, dev , feat/*]
+  pull_request:
+    branches: [ master ]
+
+jobs:
+  per-python:
+    name: Python ${{ matrix.python-version }}
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13" ]
+
+    uses: ./.github/workflows/_test-os-arch.yml
+    with:
+      python-version: ${{ matrix.python-version }}
+    secrets: inherit

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nonebot-plugin-htmlrender
-Version: 0.6.8
+Version: 0.7.0a1
 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
@@ -37,6 +37,7 @@ Requires-Dist: playwright>=1.48.0
 Requires-Dist: pygments>=2.10.0
 Requires-Dist: pymdown-extensions>=9.1
 Requires-Dist: python-markdown-math>=0.8
+Requires-Dist: tenacity>=9.1.2
 Description-Content-Type: text/markdown
 
 # nonebot-plugin-htmlrender

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/nonebot_plugin_htmlrender/browser.py

@@ -1,5 +1,5 @@
 from collections.abc import AsyncIterator
-from contextlib import asynccontextmanager
+from contextlib import AsyncExitStack, asynccontextmanager
 from typing import Optional
 
 from nonebot.log import logger
@@ -10,6 +10,7 @@ from playwright.async_api import (
     Playwright,
     async_playwright,
 )
+from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_fixed
 
 from nonebot_plugin_htmlrender.config import plugin_config
 from nonebot_plugin_htmlrender.install import install_browser
@@ -127,7 +128,30 @@ async def _connect(browser_type: str, **kwargs) -> Browser:
     if _playwright is not None:
         return await _browser_cls.connect(**kwargs)
     else:
-        raise RuntimeError("Playwright 未初始化")
+        raise RuntimeError("Playwright is not initialized")
+
+
+@retry(
+    retry=retry_if_exception_type(RuntimeError),
+    stop=stop_after_attempt(4),
+    wait=wait_fixed(1),
+    reraise=True,
+    before_sleep=lambda retry_state: logger.warning(
+        f"Attempt {retry_state.attempt_number} failed, retrying..."
+    ),
+)
+async def _check_env_with_install_retry(**kwargs):
+    try:
+        return await check_playwright_env(**kwargs)
+    except RuntimeError:
+        if plugin_config.htmlrender_ci_mode:
+            raise
+        try:
+            await install_browser()
+        except Exception as e:
+            logger.error(f"Browser installation failed: {e!s}")
+            raise RuntimeError(f"install_browser failed: {e}") from e
+        raise
 
 
 @with_lock
@@ -144,8 +168,10 @@ async def startup_htmlrender(**kwargs) -> Browser:
     global _browser, _playwright
 
     await shutdown_htmlrender()
-    clean_playwright_cache()
-    _prepare_playwright_env_vars()
+
+    if not plugin_config.htmlrender_ci_mode:
+        clean_playwright_cache()
+        _prepare_playwright_env_vars()
 
     _playwright = await async_playwright().start()
     logger.debug("Playwright started")
@@ -177,31 +203,69 @@ async def startup_htmlrender(**kwargs) -> Browser:
                     f" '{plugin_config.htmlrender_browser_executable_path}': {e}"
                 ) from e
         else:
-            try:
-                _browser = await check_playwright_env(**kwargs)
-            except RuntimeError:
-                await install_browser()
-                _browser = await check_playwright_env(**kwargs)
+            _browser = await _check_env_with_install_retry(**kwargs)
 
     return _browser
 
 
 async def shutdown_htmlrender() -> None:
-    """关闭浏览器和 Playwright 实例。"""
-    if _browser:
-        if not _browser.is_connected():
-            logger.info("Browser was already disconnected, skipping close.")
-        else:
-            with suppress_and_log():
-                logger.debug("Disconnecting browser...")
-                await _browser.close()
-                logger.info("Disconnected browser.")
-    if _playwright:
+    is_remote = bool(
+        plugin_config.htmlrender_connect or plugin_config.htmlrender_connect_over_cdp
+    )
+    async with AsyncExitStack() as stack:
+        await _schedule_browser_shutdown(stack, is_remote=is_remote)
+        await _schedule_playwright_shutdown(stack)
+    _clear_globals()
+
+
+async def _schedule_browser_shutdown(stack: AsyncExitStack, *, is_remote: bool) -> None:
+    global _browser
+    if not _browser:
+        return
+
+    should_close = (not is_remote) and plugin_config.htmlrender_shutdown_browser_on_exit
+    if not should_close:
+        logger.info(
+            "Skipping browser shutdown due to configuration or remote connection."
+        )
+        return
+
+    browser = _browser
+    if not browser.is_connected():
+        logger.info("Browser was already disconnected.")
+        return
+
+    logger.debug("Disconnecting browser...")
+
+    async def _close_browser():
+        with suppress_and_log():
+            await browser.close()
+            logger.info("Disconnected browser.")
+
+    stack.push_async_callback(_close_browser)
+
+
+async def _schedule_playwright_shutdown(stack: AsyncExitStack) -> None:
+    global _playwright
+    if not _playwright:
+        return
+
+    pw = _playwright
+    logger.debug("Stopping Playwright...")
+
+    async def _stop_pw():
         with suppress_and_log():
-            logger.debug("Stopping Playwright...")
-            await _playwright.stop()
+            await pw.stop()
             logger.info("Playwright stopped.")
 
+    stack.push_async_callback(_stop_pw)
+
+
+def _clear_globals() -> None:
+    global _browser, _playwright
+    _browser = None
+    _playwright = None
+
 
 async def check_playwright_env(**kwargs) -> Browser:
     """

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/nonebot_plugin_htmlrender/config.py

@@ -3,6 +3,7 @@ from typing import Any, Optional
 
 from nonebot import get_driver, get_plugin_config
 from nonebot.compat import model_validator
+from nonebot.log import logger
 import nonebot_plugin_localstore as store
 from pydantic import BaseModel, Field
 
@@ -32,6 +33,14 @@ class Config(BaseModel):
         default=plugin_config_dir,
         description="配置路径，不填则使用 `nonebot-plugin-localstore` 管理",
     )
+    htmlrender_shutdown_browser_on_exit: bool = Field(
+        default=True,
+        description="在插件关闭时关闭浏览器实例，在连接到远程浏览器时默认为 False。",
+    )
+    htmlrender_ci_mode: bool = Field(
+        default=False,
+        description="启用CI模式，跳过浏览器安装和环境变量",
+    )
     htmlrender_download_host: Optional[str] = Field(
         default=None, description="下载Playwright浏览器时的主机地址。"
     )
@@ -93,3 +102,8 @@ class Config(BaseModel):
 
 global_config = get_driver().config
 plugin_config = get_plugin_config(Config)
+
+if plugin_config.htmlrender_ci_mode:
+    logger.info(
+        "CI mode enabled, skipping browser installation and environment variable setup."
+    )

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/nonebot_plugin_htmlrender/install.py

@@ -202,7 +202,7 @@ async def execute_install_command(timeout: int) -> tuple[bool, str]:
                 asyncio.gather(stdout_task, stderr_task), timeout=timeout
             )
         except asyncio.TimeoutError:
-            logger.error(f"Timed out ({timeout}秒)")
+            logger.error(f"Timed out ({timeout}s)")
             await terminate_process(process)
             return False, f"Timed out ({timeout}s)"
 

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nonebot-plugin-htmlrender"
-version = "0.6.8"
+version = "0.7.0.alpha.1"
 description = "通过浏览器渲染图片"
 readme = "README.md"
 authors = [{ name = "kexue", email = "x@kexue-cloud.cn" }]
@@ -16,6 +16,7 @@ dependencies = [
     "pygments>=2.10.0",
     "pymdown-extensions>=9.1",
     "python-markdown-math>=0.8",
+    "tenacity>=9.1.2",
 ]
 
 [project.urls]

nonebot_plugin_htmlrender-0.7.0a1/tests/conftest.py

@@ -0,0 +1,42 @@
+import gc
+
+import anyio
+import nonebot
+from nonebug import NONEBOT_INIT_KWARGS
+import pytest
+from pytest_asyncio import is_async_test
+
+
+def pytest_configure(config: pytest.Config) -> None:
+    config.stash[NONEBOT_INIT_KWARGS] = {
+        "superusers": {"10001"},
+        "command_start": {""},
+        "log_level": "DEBUG",
+        "htmlrender_ci_mode": True,
+    }
+
+
+def pytest_collection_modifyitems(items: list[pytest.Item]):
+    pytest_asyncio_tests = (item for item in items if is_async_test(item))
+    session_scope_marker = pytest.mark.asyncio(loop_scope="session")
+    for async_test in pytest_asyncio_tests:
+        async_test.add_marker(session_scope_marker, append=False)
+
+
+@pytest.fixture(scope="session", autouse=True)
+async def after_nonebot_init(after_nonebot_init: None):
+    nonebot.require("nonebot_plugin_htmlrender")
+
+
+@pytest.fixture(scope="session", autouse=True)
+async def _cleanup_playwright_session():
+    from nonebot_plugin_htmlrender import shutdown_htmlrender
+
+    yield
+
+    await shutdown_htmlrender()
+
+    gc.collect()
+    gc.collect()
+
+    await anyio.lowlevel.checkpoint()

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/tests/test_htmlrender.py

@@ -152,7 +152,8 @@ async def test_capture_element(mocker: MockerFixture) -> None:
 
     mock_page = mocker.AsyncMock()
     mock_page.goto = mocker.AsyncMock()
-    mock_page.on = mocker.AsyncMock()
+    mock_page.on = mocker.MagicMock()
+    mock_page.off = mocker.MagicMock()
     mock_page.locator = mocker.MagicMock(return_value=mock_locator)
 
     mock_cm = mocker.MagicMock()

{nonebot_plugin_htmlrender-0.6.8 → nonebot_plugin_htmlrender-0.7.0a1}/uv.lock

@@ -771,7 +771,7 @@ wheels = [
 
 [[package]]
 name = "nonebot-plugin-htmlrender"
-version = "0.6.8"
+version = "0.7.0a1"
 source = { editable = "." }
 dependencies = [
     { name = "aiofiles" },
@@ -783,6 +783,7 @@ dependencies = [
     { name = "pygments" },
     { name = "pymdown-extensions" },
     { name = "python-markdown-math" },
+    { name = "tenacity" },
 ]
 
 [package.dev-dependencies]
@@ -808,6 +809,7 @@ requires-dist = [
     { name = "pygments", specifier = ">=2.10.0" },
     { name = "pymdown-extensions", specifier = ">=9.1" },
     { name = "python-markdown-math", specifier = ">=0.8" },
+    { name = "tenacity", specifier = ">=9.1.2" },
 ]
 
 [package.metadata.requires-dev]
@@ -1145,7 +1147,7 @@ wheels = [
 
 [[package]]
 name = "pydantic"
-version = "2.11.7"
+version = "2.11.9"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "annotated-types" },
@@ -1153,9 +1155,9 @@ dependencies = [
     { name = "typing-extensions" },
     { name = "typing-inspection" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/00/dd/4325abf92c39ba8623b5af936ddb36ffcfe0beae70405d456ab1fb2f5b8c/pydantic-2.11.7.tar.gz", hash = "sha256:d989c3c6cb79469287b1569f7447a17848c998458d49ebe294e975b9baf0f0db", size = 788350, upload-time = "2025-06-14T08:33:17.137Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/ff/5d/09a551ba512d7ca404d785072700d3f6727a02f6f3c24ecfd081c7cf0aa8/pydantic-2.11.9.tar.gz", hash = "sha256:6b8ffda597a14812a7975c90b82a8a2e777d9257aba3453f973acd3c032a18e2", size = 788495, upload-time = "2025-09-13T11:26:39.325Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/6a/c0/ec2b1c8712ca690e5d61979dee872603e92b8a32f94cc1b72d53beab008a/pydantic-2.11.7-py3-none-any.whl", hash = "sha256:dde5df002701f6de26248661f6835bbe296a47bf73990135c7d07ce741b9623b", size = 444782, upload-time = "2025-06-14T08:33:14.905Z" },
+    { url = "https://files.pythonhosted.org/packages/3e/d3/108f2006987c58e76691d5ae5d200dd3e0f532cb4e5fa3560751c3a1feba/pydantic-2.11.9-py3-none-any.whl", hash = "sha256:c42dd626f5cfc1c6950ce6205ea58c93efa406da65f479dcb4029d5934857da2", size = 444855, upload-time = "2025-09-13T11:26:36.909Z" },
 ]
 
 [[package]]
@@ -1330,16 +1332,16 @@ wheels = [
 
 [[package]]
 name = "pytest-asyncio"
-version = "1.1.0"
+version = "1.2.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "backports-asyncio-runner", marker = "python_full_version < '3.11'" },
     { name = "pytest" },
-    { name = "typing-extensions", marker = "python_full_version < '3.10'" },
+    { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/4e/51/f8794af39eeb870e87a8c8068642fc07bce0c854d6865d7dd0f2a9d338c2/pytest_asyncio-1.1.0.tar.gz", hash = "sha256:796aa822981e01b68c12e4827b8697108f7205020f24b5793b3c41555dab68ea", size = 46652, upload-time = "2025-07-16T04:29:26.393Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/42/86/9e3c5f48f7b7b638b216e4b9e645f54d199d7abbbab7a64a13b4e12ba10f/pytest_asyncio-1.2.0.tar.gz", hash = "sha256:c609a64a2a8768462d0c99811ddb8bd2583c33fd33cf7f21af1c142e824ffb57", size = 50119, upload-time = "2025-09-12T07:33:53.816Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/c7/9d/bf86eddabf8c6c9cb1ea9a869d6873b46f105a5d292d3a6f7071f5b07935/pytest_asyncio-1.1.0-py3-none-any.whl", hash = "sha256:5fe2d69607b0bd75c656d1211f969cadba035030156745ee09e7d71740e58ecf", size = 15157, upload-time = "2025-07-16T04:29:24.929Z" },
+    { url = "https://files.pythonhosted.org/packages/04/93/2fa34714b7a4ae72f2f8dad66ba17dd9a2c793220719e736dda28b7aec27/pytest_asyncio-1.2.0-py3-none-any.whl", hash = "sha256:8e17ae5e46d8e7efe51ab6494dd2010f4ca8dae51652aa3c8d55acf50bfb2e99", size = 15095, upload-time = "2025-09-12T07:33:52.639Z" },
 ]
 
 [[package]]
@@ -1492,6 +1494,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/ce/fd/901cfa59aaa5b30a99e16876f11abe38b59a1a2c51ffb3d7142bb6089069/starlette-0.47.3-py3-none-any.whl", hash = "sha256:89c0778ca62a76b826101e7c709e70680a1699ca7da6b44d38eb0a7e61fe4b51", size = 72991, upload-time = "2025-08-24T13:36:40.887Z" },
 ]
 
+[[package]]
+name = "tenacity"
+version = "9.1.2"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/0a/d4/2b0cd0fe285e14b36db076e78c93766ff1d529d70408bd1d2a5a84f1d929/tenacity-9.1.2.tar.gz", hash = "sha256:1169d376c297e7de388d18b4481760d478b0e99a777cad3a9c86e556f4b697cb", size = 48036, upload-time = "2025-04-02T08:25:09.966Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/e5/30/643397144bfbfec6f6ef821f36f33e57d35946c44a2352d3c9f0ae847619/tenacity-9.1.2-py3-none-any.whl", hash = "sha256:f77bf36710d8b73a50b2dd155c97b870017ad21afe6ab300326b0371b3b05138", size = 28248, upload-time = "2025-04-02T08:25:07.678Z" },
+]
+
 [[package]]
 name = "tomli"
 version = "2.2.1"

nonebot_plugin_htmlrender-0.6.8/.github/workflows/codecov.yml

@@ -1,43 +0,0 @@
-name: Code Coverage
-
-on:
-  push:
-    branches:
-      - "master"
-      - "dev"
-  pull_request:
-    branches:
-      - "master"
-
-jobs:
-  test:
-    name: Test
-    runs-on: ${{ matrix.os }}
-    strategy:
-      matrix:
-        python-version: ["3.10", "3.11", "3.12", "3.13"]
-        os: [ubuntu-latest, windows-latest, macos-latest]
-      fail-fast: false
-    env:
-      OS: ${{ matrix.os }}
-      PYTHON_VERSION: ${{ matrix.python-version }}
-    steps:
-      - name: Checkout
-        uses: actions/checkout@v4
-
-      - name: Install the latest version of uv
-        uses: astral-sh/setup-uv@v3
-        with:
-          enable-cache: true
-          python-version: ${{ matrix.python-version }}
-
-      - run: uv sync
-        shell: bash
-
-      - name: Run tests
-        run: uv run pytest -s
-
-      - name: Upload coverage to Codecov
-        uses: codecov/codecov-action@v2
-        with:
-          env_vars: OS,PYTHON_VERSION

nonebot_plugin_htmlrender-0.6.8/tests/conftest.py

@@ -1,15 +0,0 @@
-import nonebot
-import pytest
-from pytest_asyncio import is_async_test
-
-
-def pytest_collection_modifyitems(items: list[pytest.Item]):
-    pytest_asyncio_tests = (item for item in items if is_async_test(item))
-    session_scope_marker = pytest.mark.asyncio(loop_scope="session")
-    for async_test in pytest_asyncio_tests:
-        async_test.add_marker(session_scope_marker, append=False)
-
-
-@pytest.fixture(scope="session", autouse=True)
-async def after_nonebot_init(after_nonebot_init: None):
-    nonebot.require("nonebot_plugin_htmlrender")