11

core/adapters/router.py

@@ -0,0 +1,74 @@
+from collections.abc import Mapping
+from typing import Any
+
+from ..api_config import APIConfigError, APIEndpointRegistry
+from ..request import APIClient
+from .base import EndpointAdapter
+from .jx3api import JX3ApiAdapter
+from .jx3box import JX3BoxAdapter
+
+
+class APIAdapterRouter:
+    """依据服务配置把端点请求路由到对应 Adapter。"""
+
+    def __init__(
+        self,
+        client: APIClient,
+        registry: APIEndpointRegistry,
+        *,
+        jx3api_token: str = "",
+        jx3api_ticket: str = "",
+    ):
+        self._registry = registry
+        self._adapters: dict[str, EndpointAdapter] = {
+            "generic": EndpointAdapter(client, registry),
+            "jx3api": JX3ApiAdapter(
+                client,
+                registry,
+                token=jx3api_token,
+                ticket=jx3api_ticket,
+            ),
+            "jx3box": JX3BoxAdapter(client, registry),
+        }
+
+    def get_adapter(self, endpoint_name: str) -> EndpointAdapter:
+        adapter_name = self._resolve_adapter_name(endpoint_name)
+        try:
+            return self._adapters[adapter_name]
+        except KeyError as exc:
+            raise APIConfigError(
+                f"端点引用了不支持的 Adapter: {endpoint_name} -> {adapter_name}"
+            ) from exc
+
+    async def request(
+        self,
+        endpoint_name: str,
+        *,
+        params: Mapping[str, Any] | None = None,
+        out_key: str | None = "data",
+        path_params: Mapping[str, Any] | None = None,
+    ) -> Any:
+        adapter = self.get_adapter(endpoint_name)
+        return await adapter.request(
+            endpoint_name,
+            params=params,
+            out_key=out_key,
+            path_params=path_params,
+        )
+
+    async def request_all_pages(
+        self,
+        endpoint_name: str,
+        **kwargs,
+    ) -> list[Any]:
+        adapter = self.get_adapter(endpoint_name)
+        if not isinstance(adapter, JX3BoxAdapter):
+            raise APIConfigError(
+                f"端点不属于 JX3BOX，不能使用 JX3BOX 分页规则: {endpoint_name}"
+            )
+        return await adapter.request_all_pages(endpoint_name, **kwargs)
+
+    def _resolve_adapter_name(self, endpoint_name: str) -> str:
+        endpoint = self._registry.get(endpoint_name)
+        service = self._registry.get_service(str(endpoint["service"]))
+        return str(service.get("adapter", "generic")).lower()