intuned-runtime 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

runtime/run/setup_context_hook.py

@@ -0,0 +1,40 @@
+from collections.abc import Callable
+from typing import Any
+from typing import Awaitable
+from typing import cast
+from typing import Protocol
+from typing import TYPE_CHECKING
+from typing import Union
+
+from runtime.errors.run_api_errors import ApiNotFoundError
+
+from .types import ImportFunction
+
+if TYPE_CHECKING:
+    from playwright.async_api import BrowserContext
+    from playwright.async_api import Page
+
+
+setup_context_hook_path = "hooks/setup_context"
+setup_context_hook_function_name = "setup_context"
+
+SetupContextHookReturn = Union[
+    "None",
+    "BrowserContext",
+    "tuple[BrowserContext, Page | None]",
+    "tuple[BrowserContext, Page | None, Callable[..., Awaitable[None]]]",
+]
+
+
+class SetupContextHook(Protocol):
+    def __call__(self, *, cdp_url: str, api_name: str, api_parameters: Any) -> Awaitable[SetupContextHookReturn]: ...
+
+
+def load_setup_context_hook(*, import_function: ImportFunction):
+    try:
+        setup_context_hook = cast(
+            SetupContextHook, import_function(setup_context_hook_path, setup_context_hook_function_name)
+        )
+        return setup_context_hook
+    except ApiNotFoundError:
+        return None

runtime/run/types.py

@@ -0,0 +1,14 @@
+from collections.abc import Callable
+from typing import Any
+from typing import Awaitable
+from typing import Optional
+from typing import Protocol
+
+from git import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    pass
+
+
+class ImportFunction(Protocol):
+    def __call__(self, file_path: str, name: Optional[str] = None, /) -> Callable[..., Awaitable[Any]]: ...

runtime/types/run_types.py

@@ -126,7 +126,6 @@ class TracingDisabled(CamelBaseModel):
 
 class Auth(CamelBaseModel):
     session: RunApiSession
-    run_check: bool = False
 
 
 class IntunedRunContext(CamelBaseModel):

runtime_helpers/__init__.py

@@ -1,5 +1,6 @@
 from runtime.helpers import extend_payload
 from runtime.helpers import extend_timeout
+from runtime.helpers.attempt_store import attempt_store
 from runtime.helpers.get_auth_session_parameters import get_auth_session_parameters
 
-__all__ = ["extend_payload", "extend_timeout", "get_auth_session_parameters"]
+__all__ = ["extend_payload", "extend_timeout", "get_auth_session_parameters", "attempt_store"]

runtime/run/playwright_constructs.py

@@ -1,20 +0,0 @@
-from contextlib import asynccontextmanager
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from playwright.async_api import ProxySettings
-from ..browser import launch_browser
-
-
-@asynccontextmanager
-async def get_production_playwright_constructs(
-    proxy: "ProxySettings | None" = None,
-    headless: bool = False,
-    *,
-    cdp_address: str | None = None,
-):
-    async with launch_browser(headless=headless, cdp_address=cdp_address, proxy=proxy) as (context, page):
-        try:
-            yield context, page
-        finally:
-            await context.close()