From dffcd9f792ba19d55dbc331455a8fa4c384f51a6 Mon Sep 17 00:00:00 2001 From: GitHub Date: Tue, 1 Sep 2026 17:58:12 +0800 Subject: [PATCH] =?UTF-8?q?style:=20ruff=20format=20=E4=B8=89=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E9=87=8D=E6=8E=92=EF=BC=88quote=5Fstreamer/watchlist?= =?UTF-8?q?=5Fstore/test=5Fex=5Ftick=5Fchart=5Fdate=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 本地预演 CI 全部四道门禁通过:ruff check / ruff format --check / mypy (226 files) / 1078 tests --- src/easy_tdx/web/quote_streamer.py | 48 +++++++++++++-------------- src/easy_tdx/web/watchlist_store.py | 8 +++-- tests/unit/test_ex_tick_chart_date.py | 3 +- 3 files changed, 32 insertions(+), 27 deletions(-) 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)),