dataapp-resource-proxy 0.1.0__py3-none-any.whl

dataapp_resource_proxy/errors.py

@@ -0,0 +1,16 @@
+"""Public client errors."""
+
+
+class ResourceProxyError(RuntimeError):
+    """A managed resource request failed."""
+
+    def __init__(
+        self,
+        message: str,
+        *,
+        code: str = "resource-proxy-error",
+        status_code: int | None = None,
+    ) -> None:
+        super().__init__(message)
+        self.code = code
+        self.status_code = status_code

dataapp_resource_proxy/handles.py

@@ -0,0 +1,199 @@
+"""Owned resource handles returned by :mod:`dataapp_resource_proxy`."""
+
+from __future__ import annotations
+
+import asyncio
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from .client import ResourceProxy
+
+
+@dataclass
+class ProxyResource:
+    """One DataOS proxy lease owned by this client."""
+
+    client: ResourceProxy = field(repr=False)
+    lease_id: str
+    proxy_url: str = field(repr=False)
+    expires_at: datetime | None = None
+    credential_generation: int | None = None
+    network: str | None = None
+    node_id: str | None = None
+    line_tier: str | None = None
+    proxy_group: str | None = None
+    supplier_code: str | None = None
+    supplier_route_id: str | None = None
+    supplier_account_ref: str | None = None
+    line_policy_version: str | None = None
+    _closed: bool = field(default=False, init=False, repr=False)
+    _lock: asyncio.Lock = field(default_factory=asyncio.Lock, init=False, repr=False)
+
+    def __repr__(self) -> str:
+        return f"ProxyResource(lease_id={self.lease_id!r}, proxy_url=<redacted>)"
+
+    async def renew(self) -> None:
+        async with self._lock:
+            self._require_open()
+            value = await self.client._renew_proxy(self.lease_id)
+            self._update(value)
+
+    async def refresh(self) -> None:
+        """Refresh safe lease metadata without changing credentials."""
+        async with self._lock:
+            self._require_open()
+            value = await self.client._get_proxy(self.lease_id)
+            self._update(value)
+
+    async def rotate_credentials(self) -> None:
+        """Rotate only the DataOS gateway credential, not the upstream exit IP."""
+        async with self._lock:
+            self._require_open()
+            value = await self.client._rotate_proxy(
+                self.lease_id, credential_generation=self.credential_generation
+            )
+            self._update(value)
+
+    async def close(self) -> None:
+        async with self._lock:
+            if self._closed:
+                return
+            await self.client._release_proxy(self.lease_id)
+            self._closed = True
+            self.client._forget(self)
+
+    def _update(self, value: dict[str, Any]) -> None:
+        if "proxy_url" in value:
+            self.proxy_url = _required_text(value, "proxy_url")
+        if "expires_at" in value:
+            self.expires_at = _datetime(value.get("expires_at"))
+        if "credential_generation" in value:
+            generation = value.get("credential_generation")
+            self.credential_generation = int(generation) if generation is not None else None
+        if "network" in value:
+            self.network = _optional_text(value.get("network"))
+        if "node_id" in value:
+            self.node_id = _optional_text(value.get("node_id"))
+        if "line_tier" in value:
+            self.line_tier = _optional_text(value.get("line_tier"))
+        if "proxy_group" in value:
+            self.proxy_group = _optional_text(value.get("proxy_group"))
+        if "supplier_code" in value:
+            self.supplier_code = _optional_text(value.get("supplier_code"))
+        if "supplier_route_id" in value:
+            self.supplier_route_id = _optional_text(value.get("supplier_route_id"))
+        if "supplier_account_ref" in value:
+            self.supplier_account_ref = _optional_text(value.get("supplier_account_ref"))
+        if "line_policy_version" in value:
+            self.line_policy_version = _optional_text(value.get("line_policy_version"))
+
+    def _require_open(self) -> None:
+        if self._closed:
+            raise RuntimeError("proxy resource is closed")
+
+
+@dataclass(frozen=True)
+class CaptchaTask:
+    """Node-owned task snapshot; solver output still needs site acceptance."""
+
+    task_id: str
+    status: str
+    token: str | None = field(default=None, repr=False)
+    text: str | None = field(default=None, repr=False)
+    user_agent: str | None = field(default=None, repr=False)
+    error_code: str | None = None
+    node_id: str | None = None
+    network: str | None = None
+    data: dict[str, Any] | None = field(default=None, repr=False)
+
+
+@dataclass(frozen=True)
+class CaptchaPluginGrant:
+    """Short-lived permission for the managed extension adapter only."""
+
+    api_url: str
+    client_key: str = field(repr=False)
+    expires_at: datetime
+    node_id: str
+    network: str
+    protocol_version: int
+
+
+@dataclass
+class BrowserResource:
+    """One managed browser session owned by this client."""
+
+    client: ResourceProxy = field(repr=False)
+    resource_id: str
+    status: str
+    session_deadline: datetime | None = None
+    network: str | None = None
+    node_id: str | None = None
+    _closed: bool = field(default=False, init=False, repr=False)
+    _lock: asyncio.Lock = field(default_factory=asyncio.Lock, init=False, repr=False)
+
+    async def connection(self) -> tuple[str, dict[str, str]]:
+        async with self._lock:
+            self._require_open()
+            value = await self.client._browser_connection(self.resource_id)
+            ws_url = _required_text(value, "ws_url")
+            raw_headers = value.get("headers")
+            if not isinstance(raw_headers, dict) or not all(
+                isinstance(key, str) and isinstance(item, str)
+                for key, item in raw_headers.items()
+            ):
+                raise ValueError("resource service returned invalid browser headers")
+            return ws_url, dict(raw_headers)
+
+    async def refresh(self) -> None:
+        """Refresh the browser status and deadline."""
+        async with self._lock:
+            self._require_open()
+            value = await self.client._get_browser(self.resource_id)
+            if "status" in value:
+                self.status = str(value["status"])
+            if "session_deadline" in value or "expires_at" in value:
+                self.session_deadline = _datetime(
+                    value.get("session_deadline") or value.get("expires_at")
+                )
+            if "network" in value:
+                self.network = _optional_text(value.get("network"))
+            if "node_id" in value:
+                self.node_id = _optional_text(value.get("node_id"))
+
+    async def close(self) -> None:
+        async with self._lock:
+            if self._closed:
+                return
+            await self.client._release_browser(self.resource_id)
+            self._closed = True
+            self.client._forget(self)
+
+    def _require_open(self) -> None:
+        if self._closed:
+            raise RuntimeError("browser resource is closed")
+
+
+def _required_text(value: dict[str, Any], name: str) -> str:
+    result = value.get(name)
+    if not isinstance(result, str) or not result:
+        raise ValueError(f"resource service returned invalid {name}")
+    return result
+
+
+def _datetime(value: Any) -> datetime | None:
+    if value is None:
+        return None
+    if not isinstance(value, str):
+        raise ValueError("resource service returned an invalid timestamp")
+    return datetime.fromisoformat(value.replace("Z", "+00:00"))
+
+
+def _optional_text(value: Any) -> str | None:
+    if value is None:
+        return None
+    if not isinstance(value, str) or not value:
+        raise ValueError("resource service returned invalid resource metadata")
+    return value

dataapp_resource_proxy-0.1.0.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.5
+Name: dataapp-resource-proxy
+Version: 0.1.0
+Summary: Run-scoped client for managed browser, proxy and CAPTCHA resources
+Requires-Python: >=3.11
+Requires-Dist: httpx<1,>=0.28

dataapp_resource_proxy-0.1.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+dataapp_resource_proxy/__init__.py,sha256=RHgtBZrNRZpzD5xTD8UaUCtOeT4M0rEYIAJqZg56UM0,405
+dataapp_resource_proxy/__main__.py,sha256=nLRIRYJdKbIbfIa9aZUiHij28aS8cT-waGVWA0CUlzc,2417
+dataapp_resource_proxy/_local_config.py,sha256=VeevtAmHCJZC1Gy4Xc6T0wXHweMDY5RopOBMwwcBNzo,11134
+dataapp_resource_proxy/client.py,sha256=_ACYOpltkUsP8-pM_e3ZINCjnktORuQajXASYjmzSC8,76512
+dataapp_resource_proxy/errors.py,sha256=ASFQ56uxUHlPx7wFySpk5sLO5-k4FcydCgW0dxY4zwo,378
+dataapp_resource_proxy/handles.py,sha256=Y-bSgl7uCA-I01HhCTjMcNLCa4nCl081Bv6KLT0TgdI,7354
+dataapp_resource_proxy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dataapp_resource_proxy-0.1.0.dist-info/METADATA,sha256=FYg6XLXJMoLya6hpyEBqdTcz5_EqDqnS-GiY6PrazAE,196
+dataapp_resource_proxy-0.1.0.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+dataapp_resource_proxy-0.1.0.dist-info/RECORD,,

dataapp_resource_proxy-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.33.0
+Root-Is-Purelib: true
+Tag: py3-none-any