58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from dataclasses import dataclass
|
|
|
|
from backend.data.contracts import DataUsage
|
|
from backend.data.policy import DataSourcePolicy
|
|
from backend.data.providers import IfindProvider, TushareProvider
|
|
from chart_data_provider import EastmoneyChartClient, MarketChartClient
|
|
from ifind_client import IfindHttpClient
|
|
from realtime_aggregator import WebRealtimeAggregator
|
|
from tushare_client import TushareClient
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class DataGateway:
|
|
policy: DataSourcePolicy
|
|
tushare_provider: TushareProvider
|
|
ifind_provider: IfindProvider
|
|
chart_data: MarketChartClient
|
|
realtime_observer: WebRealtimeAggregator
|
|
|
|
@property
|
|
def ifind(self) -> IfindHttpClient:
|
|
return self.ifind_provider.client
|
|
|
|
def tushare(
|
|
self,
|
|
dataset_id: str = "",
|
|
usage: DataUsage = "calculation",
|
|
) -> TushareClient:
|
|
if dataset_id:
|
|
self.policy.assert_allowed(dataset_id, "tushare", usage)
|
|
return self.tushare_provider.client()
|
|
|
|
def assert_source(self, dataset_id: str, provider_id: str, usage: DataUsage) -> None:
|
|
self.policy.assert_allowed(dataset_id, provider_id, usage)
|
|
|
|
|
|
def build_data_gateway(
|
|
credentials: dict[str, object],
|
|
tushare_token_supplier: Callable[[], str] | None = None,
|
|
) -> DataGateway:
|
|
ifind = IfindHttpClient(
|
|
str(credentials.get("ifind_refresh_token") or ""),
|
|
str(credentials.get("ifind_access_token") or ""),
|
|
)
|
|
token_supplier = tushare_token_supplier or (
|
|
lambda: str(credentials.get("tushare_token") or "")
|
|
)
|
|
return DataGateway(
|
|
policy=DataSourcePolicy.load(),
|
|
tushare_provider=TushareProvider(token_supplier),
|
|
ifind_provider=IfindProvider(ifind),
|
|
chart_data=MarketChartClient(ifind, EastmoneyChartClient()),
|
|
realtime_observer=WebRealtimeAggregator(),
|
|
)
|