diff --git a/src/easy_tdx/web/quote_streamer.py b/src/easy_tdx/web/quote_streamer.py index a960712..8343ada 100644 --- a/src/easy_tdx/web/quote_streamer.py +++ b/src/easy_tdx/web/quote_streamer.py @@ -50,30 +50,30 @@ INDEX_NAMES: dict[str, str] = { _MARKET_NAMES = {Market.SZ: "SZ", Market.SH: "SH", Market.BJ: "BJ"} # 推送给前端的字段白名单(SecurityQuote 全字段中挑展示需要的,避免 unknown_* 噪音)。 -_QUOTE_FIELDS = [ - "market", - "code", - "price", - "pre_close", - "open", - "high", - "low", - "vol", - "cur_vol", - "amount", - "s_vol", - "b_vol", - "rise_speed", - "limit_up", - "limit_down", - "decimal_point", - "server_time", - "trading_status", -] + [ - f"{side}{i}" for side in ("bid", "ask") for i in range(1, 6) -] + [ - f"{side}_vol{i}" for side in ("bid", "ask") for i in range(1, 6) -] +_QUOTE_FIELDS = ( + [ + "market", + "code", + "price", + "pre_close", + "open", + "high", + "low", + "vol", + "cur_vol", + "amount", + "s_vol", + "b_vol", + "rise_speed", + "limit_up", + "limit_down", + "decimal_point", + "server_time", + "trading_status", + ] + + [f"{side}{i}" for side in ("bid", "ask") for i in range(1, 6)] + + [f"{side}_vol{i}" for side in ("bid", "ask") for i in range(1, 6)] +) _SH_TZ = dt_timezone(timedelta(hours=8)) # Asia/Shanghai diff --git a/src/easy_tdx/web/watchlist_store.py b/src/easy_tdx/web/watchlist_store.py index 28872b3..8a24238 100644 --- a/src/easy_tdx/web/watchlist_store.py +++ b/src/easy_tdx/web/watchlist_store.py @@ -151,8 +151,12 @@ class WatchlistStore: (market, code, name, group, _now_iso(), next_order), ) return WatchItem( - market=market, code=code, name=name, group_name=group, - created_at=_now_iso(), sort_order=next_order, + market=market, + code=code, + name=name, + group_name=group, + created_at=_now_iso(), + sort_order=next_order, ) def remove(self, market: str, code: str) -> bool: diff --git a/tests/unit/test_ex_tick_chart_date.py b/tests/unit/test_ex_tick_chart_date.py index 8606d39..6d89d31 100644 --- a/tests/unit/test_ex_tick_chart_date.py +++ b/tests/unit/test_ex_tick_chart_date.py @@ -26,7 +26,8 @@ from easy_tdx.ex.mac_client import MacExClient, _coerce_query_date # --------------------------------------------------------------------------- -@pytest.mark.parametrize("raw,expected", +@pytest.mark.parametrize( + "raw,expected", [ (20260827, date(2026, 8, 27)), (20250101, date(2025, 1, 1)),