mirror of
https://ghfast.top/https://github.com/aeroxw/easy-tdx.git
synced 2026-09-12 13:24:15 +08:00
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
"""Dependency injection for Web API routers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from functools import lru_cache
|
|
from typing import Any
|
|
|
|
from fastapi import Request
|
|
|
|
from easy_tdx.client import AsyncTdxClient
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def get_ths_web_client() -> Any:
|
|
"""返回共享的同花顺公开网页客户端（仅行情快照短时缓存）。"""
|
|
from easy_tdx.ths_web import ThsWebClient
|
|
|
|
return ThsWebClient()
|
|
|
|
|
|
def get_client(request: Request) -> AsyncTdxClient:
|
|
"""从 app.state 获取共享的 AsyncTdxClient 实例。"""
|
|
client: AsyncTdxClient = request.app.state.tdx_client
|
|
return client
|
|
|
|
|
|
def get_mac_client(request: Request) -> Any:
|
|
"""从 app.state 获取共享的 AsyncMacClient 实例。"""
|
|
from easy_tdx.exceptions import TdxConnectionError
|
|
|
|
client: Any | None = request.app.state.mac_client
|
|
if client is None:
|
|
raise TdxConnectionError("MAC 客户端未连接")
|
|
return client
|
|
|
|
|
|
def get_mac_client_optional(request: Request) -> Any | None:
|
|
"""从 app.state 获取 AsyncMacClient 实例，未连接时返回 None（不抛异常）。
|
|
|
|
供需要"MAC 不可用时自动回退标准 TdxClient"的端点使用（如 ``/bars``）。
|
|
其他强制依赖 MAC 的端点（``/mac/*``）仍用 :func:`get_mac_client`。
|
|
"""
|
|
return request.app.state.mac_client
|
|
|
|
|
|
def get_ex_client(request: Request) -> Any:
|
|
"""从 app.state 获取共享的 AsyncExTdxClient 实例（可选）。"""
|
|
client: Any | None = request.app.state.ex_client
|
|
if client is None:
|
|
from easy_tdx.exceptions import TdxConnectionError
|
|
|
|
raise TdxConnectionError("扩展市场客户端未启用")
|
|
return client
|