rustwright-core 0.1.1

Rust CDP core for a Python Playwright-compatible automation API
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

"""Explicit opt-in Playwright/Patchright/Cloakbrowser import compatibility."""

from __future__ import annotations

import importlib
import sys
from types import ModuleType
from typing import Optional


_ALIASES = (
    ("playwright", "rustwright._compat.playwright"),
    ("playwright.__main__", "rustwright._compat.playwright.__main__"),
    ("playwright._impl", "rustwright._compat.playwright._impl"),
    ("playwright._impl._errors", "rustwright._compat.playwright._impl._errors"),
    ("playwright.async_api", "rustwright._compat.playwright.async_api"),
    ("playwright.pytest_plugin", "rustwright._compat.playwright.pytest_plugin"),
    ("playwright.sync_api", "rustwright._compat.playwright.sync_api"),
    ("patchright", "rustwright._compat.patchright"),
    ("patchright.__main__", "rustwright._compat.patchright.__main__"),
    ("patchright._impl", "rustwright._compat.patchright._impl"),
    ("patchright._impl._errors", "rustwright._compat.patchright._impl._errors"),
    ("patchright.async_api", "rustwright._compat.patchright.async_api"),
    ("patchright.pytest_plugin", "rustwright._compat.patchright.pytest_plugin"),
    ("patchright.sync_api", "rustwright._compat.patchright.sync_api"),
    ("cloakbrowser", "rustwright._compat.cloakbrowser"),
    ("pytest_playwright", "rustwright._compat.pytest_playwright"),
    ("pytest_playwright.pytest_playwright", "rustwright._compat.pytest_playwright.pytest_playwright"),
)

_PREVIOUS_MODULES: dict[str, Optional[ModuleType]] = {}
_ENABLED = False


def _set_parent_attribute(module_name: str, module: ModuleType) -> None:
    parent_name, _, child_name = module_name.rpartition(".")
    if not parent_name:
        return
    parent = sys.modules.get(parent_name)
    if parent is not None:
        setattr(parent, child_name, module)


def enable_playwright_compat() -> None:
    """Enable legacy Playwright-compatible import names for this Python process.

    After this is called, subsequent imports such as ``playwright.sync_api`` or
    ``patchright.async_api`` resolve to Rustwright's compatibility shims.
    """

    global _ENABLED
    if _ENABLED:
        return

    loaded_modules = [(alias_name, importlib.import_module(target_name)) for alias_name, target_name in _ALIASES]
    for alias_name, module in loaded_modules:
        _PREVIOUS_MODULES[alias_name] = sys.modules.get(alias_name)
        sys.modules[alias_name] = module
        _set_parent_attribute(alias_name, module)

    _ENABLED = True


def disable_playwright_compat() -> None:
    """Undo aliases installed by :func:`enable_playwright_compat`."""

    global _ENABLED
    if not _ENABLED:
        return

    for alias_name, _target_name in _ALIASES:
        previous = _PREVIOUS_MODULES.get(alias_name)
        if previous is None:
            sys.modules.pop(alias_name, None)
        else:
            sys.modules[alias_name] = previous
            _set_parent_attribute(alias_name, previous)

    _PREVIOUS_MODULES.clear()
    _ENABLED = False


__all__ = ["disable_playwright_compat", "enable_playwright_compat"]