diff --git a/unicorn_websocket.py b/unicorn_websocket.py index e5c9188..1656e85 100644 --- a/unicorn_websocket.py +++ b/unicorn_websocket.py @@ -4,7 +4,22 @@ Unicorn WebSocket模块 - 提供高性能实时数据流 import asyncio import logging from typing import Dict, List, Optional, Callable -from unicorn_binance_websocket_api.unicorn_binance_websocket_api_manager import BinanceWebSocketApiManager + +try: + # 新版本导入路径 + from unicorn_binance_websocket_api.manager import BinanceWebSocketApiManager +except ImportError: + try: + # 兼容旧版本路径 + from unicorn_binance_websocket_api.unicorn_binance_websocket_api_manager import ( + BinanceWebSocketApiManager, + ) + except ImportError as e: # pragma: no cover - 环境缺失依赖时提示 + raise ImportError( + "缺少依赖 unicorn-binance-websocket-api,请执行 " + "'pip install unicorn-binance-websocket-api==2.4.0'" + ) from e + import config logger = logging.getLogger(__name__)