mirror of
https://github.com/shy3130/tick-stock-panel.git
synced 2026-11-02 10:26:13 +00:00
kevin9327
bd85d56cae
助手工具 get_stock_daily / get_stock_analysis 已用 cn_today(), 个股分析页的 _load_kline 与 GET /levels 仍走 date.today()。 美西主机整个 A 股交易时段、UTC 主机北京 00:00-08:00, 本地日历日 比北京早一天, 管道已写入的当日官方 K 被排除, 分析与价位停在昨天。 Closes #390
66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
"""个股分析 / 关键价位的日K窗口右端必须是北京日期, 不能用服务器本地 date.today()。
|
|
|
|
CONTRIBUTING §3.3: A 股交易时段按北京时间, 服务器时区不能成为隐式输入。
|
|
助手工具 get_stock_daily / get_stock_analysis 已改用 cn_today()
|
|
(test_assistant_beijing_today), 但个股分析页的两条读路径仍是 date.today():
|
|
|
|
- stock_analyzer._load_kline: AI 四维分析注入的最近 N 根日 K
|
|
- stock_analysis.get_levels: 图表关键价位 / markLine 数据源
|
|
|
|
美西主机整个 A 股交易时段、UTC 主机北京 00:00-08:00, 本地日历日比北京
|
|
早一天: 管道已写入的当日官方 K 被排除, 分析与价位停在昨天。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import date
|
|
from types import SimpleNamespace
|
|
|
|
import polars as pl
|
|
|
|
from app.api import stock_analysis as stock_analysis_api
|
|
from app.services import stock_analyzer
|
|
|
|
BJ = date(2026, 3, 2) # 钉死的北京日期, 不会碰巧等于跑测试那天的 date.today()
|
|
|
|
|
|
class _Repo:
|
|
def __init__(self) -> None:
|
|
self.windows: list[tuple[date, date]] = []
|
|
|
|
def resolve_asset_type(self, symbol: str) -> str:
|
|
return "stock"
|
|
|
|
def get_daily_asset(self, asset_type, symbol, start, end, columns=None) -> pl.DataFrame:
|
|
self.windows.append((start, end))
|
|
return pl.DataFrame()
|
|
|
|
|
|
def test_load_kline_window_ends_on_beijing_today(monkeypatch) -> None:
|
|
"""AI 个股分析读日K的窗口右端必须是北京今天。
|
|
|
|
raising=False: 未修复代码没有调用 cn_today, 钉了也不会被用到,
|
|
仍走 date.today() → 与 2026-03-02 不等。
|
|
"""
|
|
monkeypatch.setattr(stock_analyzer, "cn_today", lambda: BJ, raising=False)
|
|
repo = _Repo()
|
|
stock_analyzer._load_kline(repo, "600000.SH")
|
|
assert repo.windows, "应查询日K"
|
|
_start, end = repo.windows[0]
|
|
assert end == BJ, f"窗口右端必须是北京日期 {BJ}, 实际 {end} (服务器本地 {date.today()})"
|
|
assert end != date.today()
|
|
|
|
|
|
def test_levels_window_ends_on_beijing_today(monkeypatch) -> None:
|
|
"""/api/stock-analysis/levels 未传日期时窗口右端必须是北京今天。
|
|
|
|
raising=False: 未修复代码没有调用 cn_today, 钉了也不会被用到。
|
|
"""
|
|
monkeypatch.setattr(stock_analysis_api, "cn_today", lambda: BJ, raising=False)
|
|
repo = _Repo()
|
|
req = SimpleNamespace(app=SimpleNamespace(state=SimpleNamespace(repo=repo)))
|
|
stock_analysis_api.get_levels(req, symbol="600000.SH", days=120)
|
|
assert repo.windows, "应查询日K"
|
|
_start, end = repo.windows[0]
|
|
assert end == BJ, f"窗口右端必须是北京日期 {BJ}, 实际 {end} (服务器本地 {date.today()})"
|
|
assert end != date.today()
|