74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
"""数据覆盖开机检查器。"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from dataclasses import dataclass
|
||
from datetime import date, timedelta
|
||
from typing import Callable, Dict
|
||
|
||
from app.data.schema import initialize_database
|
||
from app.ingest.tushare import collect_data_coverage, ensure_data_coverage
|
||
|
||
LOGGER = logging.getLogger(__name__)
|
||
|
||
|
||
@dataclass
|
||
class CoverageReport:
|
||
start: str
|
||
end: str
|
||
expected_trading_days: int
|
||
tables: Dict[str, Dict[str, object]]
|
||
stock_basic: Dict[str, object]
|
||
|
||
def to_dict(self) -> Dict[str, object]:
|
||
return {
|
||
"start": self.start,
|
||
"end": self.end,
|
||
"expected_trading_days": self.expected_trading_days,
|
||
"tables": self.tables,
|
||
"stock_basic": self.stock_basic,
|
||
}
|
||
|
||
|
||
def _default_window(days: int = 365) -> tuple[date, date]:
|
||
end = date.today()
|
||
start = end - timedelta(days=days)
|
||
return start, end
|
||
|
||
|
||
def run_boot_check(
|
||
days: int = 365,
|
||
auto_fetch: bool = True,
|
||
progress_hook: Callable[[str, float], None] | None = None,
|
||
) -> CoverageReport:
|
||
"""执行开机自检，必要时自动补数据。"""
|
||
|
||
initialize_database()
|
||
start, end = _default_window(days)
|
||
LOGGER.info("开机检查覆盖窗口：%s 至 %s", start, end)
|
||
|
||
if auto_fetch:
|
||
ensure_data_coverage(start, end, progress_hook=progress_hook)
|
||
|
||
coverage = collect_data_coverage(start, end)
|
||
|
||
report = CoverageReport(
|
||
start=coverage["period"]["start"],
|
||
end=coverage["period"]["end"],
|
||
expected_trading_days=coverage["period"]["expected_trading_days"],
|
||
tables={k: v for k, v in coverage.items() if k not in ("period", "stock_basic")},
|
||
stock_basic=coverage["stock_basic"],
|
||
)
|
||
|
||
LOGGER.info(
|
||
"数据覆盖情况：日线[%s,%s]，Distinct=%s，目标交易日=%s",
|
||
report.tables["daily"].get("min"),
|
||
report.tables["daily"].get("max"),
|
||
report.tables["daily"].get("distinct_days"),
|
||
report.expected_trading_days,
|
||
)
|
||
if progress_hook:
|
||
progress_hook("数据覆盖检查完成", 1.0)
|
||
|
||
return report
|