75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
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()
|