hoodini.utils.playwright_utils

 1"""
 2Playwright utilities for browser automation with proper environment setup.
 3
 4This module provides async helpers for Playwright Firefox that handle
 5LD_LIBRARY_PATH setup for conda/pixi/mamba environments automatically.
 6"""
 7
 8from collections.abc import AsyncGenerator
 9from contextlib import asynccontextmanager
10
11from playwright.async_api import Page, async_playwright
12
13from .runtime_env import apply_ld_library_path
14
15
16@asynccontextmanager
17async def get_browser_context() -> AsyncGenerator[Page, None]:
18    """
19    Context manager for async Playwright browser with proper environment setup.
20
21    Automatically:
22    - Sets up LD_LIBRARY_PATH for conda/pixi/mamba environments
23    - Launches Firefox browser with appropriate arguments
24    - Provides a page context for automation
25    - Cleans up browser on exit
26
27    Yields:
28        playwright.async_api.Page: Browser page for automation
29
30    Example:
31        >>> async with get_browser_context() as page:
32        ...     await page.goto("https://example.com")
33        ...     await page.fill("#search", "query")
34    """
35    # Apply environment setup before launching browser
36    apply_ld_library_path()
37
38    async with async_playwright() as p:
39        browser = await p.firefox.launch(headless=True)
40        page = await browser.new_page()
41        try:
42            yield page
43        finally:
44            await browser.close()
45
46
47async def open_blast_page(url: str) -> tuple[Page, str | None]:
48    """
49    Open NCBI BLAST page and return page object for automation.
50
51    This is a utility function for testing the Playwright setup.
52
53    Args:
54        url: The NCBI BLAST URL to open
55
56    Returns:
57        Tuple of (page object, error message if any)
58
59    Example:
60        >>> page, error = await open_blast_page("https://blast.ncbi.nlm.nih.gov/...")
61        >>> if not error:
62        ...     # Automate BLAST workflow
63        ...     pass
64    """
65    try:
66        apply_ld_library_path()
67        async with async_playwright() as p:
68            browser = await p.firefox.launch(headless=True)
69            page = await browser.new_page()
70            await page.goto(url)
71            return page, None
72    except Exception as e:
73        return None, str(e)
74
75
76__all__ = [
77    "get_browser_context",
78    "open_blast_page",
79]