初始化工程

backend/app/api/financials.py

@@ -0,0 +1,127 @@
+"""财务数据 API — 独立路由, Cap.FINANCIAL 门控。"""
+from __future__ import annotations
+
+import logging
+
+import polars as pl
+from fastapi import APIRouter, HTTPException, Request
+
+from app.services.financial_sync import get_financial_df
+from app.tickflow.capabilities import Cap
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/api/financials", tags=["financials"])
+
+
+@router.get("/status")
+def financial_status(request: Request):
+    """返回各财务表的同步状态。无需 FINANCIAL 权限（前端根据 available 决定是否展示）。"""
+    capset = request.app.state.capabilities
+    if not capset.has(Cap.FINANCIAL):
+        return {"available": False, "tables": {}}
+
+    data_dir = request.app.state.repo.store.data_dir
+    tables = {}
+
+    for table in ("metrics", "income", "balance_sheet", "cash_flow"):
+        path = data_dir / "financials" / table / "part.parquet"
+        if path.exists():
+            try:
+                df = pl.read_parquet(path, columns=["symbol"])
+                tables[table] = {
+                    "rows": len(df),
+                    "symbols": df["symbol"].n_unique() if not df.is_empty() else 0,
+                }
+            except Exception:
+                tables[table] = {"rows": 0, "symbols": 0}
+        else:
+            tables[table] = {"rows": 0, "symbols": 0}
+
+    fs = getattr(request.app.state, "financial_scheduler", None)
+    last_sync = fs.last_sync if fs else {}
+
+    return {
+        "available": True,
+        "tables": tables,
+        "last_sync": last_sync,
+        # 服务端是否正在同步(手动触发)——前端据此显示"同步中"并防重复点击,
+        # 且刷新页面后仍能正确反映服务端状态。
+        "syncing": bool(fs and fs.is_syncing),
+    }
+
+
+@router.get("/metrics")
+def get_metrics(request: Request, symbol: str | None = None):
+    """查询核心财务指标。"""
+    capset = request.app.state.capabilities
+    capset.require(Cap.FINANCIAL)
+
+    df = get_financial_df(request.app.state.repo.store.data_dir, "metrics")
+    if df.is_empty():
+        return {"data": []}
+    if symbol:
+        df = df.filter(pl.col("symbol") == symbol)
+    return {"data": df.to_dicts()}
+
+
+@router.get("/income")
+def get_income(request: Request, symbol: str | None = None):
+    """查询利润表。"""
+    capset = request.app.state.capabilities
+    capset.require(Cap.FINANCIAL)
+
+    df = get_financial_df(request.app.state.repo.store.data_dir, "income")
+    if df.is_empty():
+        return {"data": []}
+    if symbol:
+        df = df.filter(pl.col("symbol") == symbol)
+    return {"data": df.to_dicts()}
+
+
+@router.get("/balance-sheet")
+def get_balance_sheet(request: Request, symbol: str | None = None):
+    """查询资产负债表。"""
+    capset = request.app.state.capabilities
+    capset.require(Cap.FINANCIAL)
+
+    df = get_financial_df(request.app.state.repo.store.data_dir, "balance_sheet")
+    if df.is_empty():
+        return {"data": []}
+    if symbol:
+        df = df.filter(pl.col("symbol") == symbol)
+    return {"data": df.to_dicts()}
+
+
+@router.get("/cash-flow")
+def get_cash_flow(request: Request, symbol: str | None = None):
+    """查询现金流量表。"""
+    capset = request.app.state.capabilities
+    capset.require(Cap.FINANCIAL)
+
+    df = get_financial_df(request.app.state.repo.store.data_dir, "cash_flow")
+    if df.is_empty():
+        return {"data": []}
+    if symbol:
+        df = df.filter(pl.col("symbol") == symbol)
+    return {"data": df.to_dicts()}
+
+
+@router.post("/sync/{table}")
+def sync_table(request: Request, table: str):
+    """手动触发同步。table: metrics / income / balance_sheet / cash_flow / all"""
+    capset = request.app.state.capabilities
+    capset.require(Cap.FINANCIAL)
+
+    valid_tables = {"metrics", "income", "balance_sheet", "cash_flow", "all"}
+    if table not in valid_tables:
+        raise HTTPException(400, f"invalid table: {table}, expected one of {valid_tables}")
+
+    fs = getattr(request.app.state, "financial_scheduler", None)
+    if not fs:
+        return {"status": "error", "message": "FinancialScheduler not available"}
+
+    target = None if table == "all" else table
+    result = fs.run_now(target)
+
+    return {"status": "ok", "synced": result}