每日备份 2026-03-27

skills/daily-stock-analysis/scripts/_report_utils.py

@@ -0,0 +1,162 @@
+#!/usr/bin/env python3
+"""Shared helpers for daily-stock-analysis scripts."""
+
+from __future__ import annotations
+
+import os
+import re
+from dataclasses import dataclass
+from datetime import date
+from typing import Dict, List, Optional
+
+
+FILENAME_RE = re.compile(
+    r"^(?P<run_date>\d{4}-\d{2}-\d{2})-(?P<ticker>[A-Za-z0-9._-]+)-analysis(?:-v(?P<version>\d+))?\.md$",
+    re.IGNORECASE,
+)
+
+
+@dataclass(frozen=True)
+class ReportFile:
+    path: str
+    run_date: str
+    ticker: str
+    version: int
+    in_canonical_dir: bool
+
+
+def canonical_reports_dir(workdir: str) -> str:
+    return os.path.join(os.path.abspath(workdir), "daily-stock-analysis", "reports")
+
+
+def compatible_dirs(workdir: str) -> List[str]:
+    root = os.path.abspath(workdir)
+    return [
+        canonical_reports_dir(root),
+        os.path.join(root, "daily-stock-analysis"),
+        root,
+    ]
+
+
+def is_within_workdir(path: str, workdir: str) -> bool:
+    root = os.path.realpath(os.path.abspath(workdir))
+    target = os.path.realpath(os.path.abspath(path))
+    return target == root or target.startswith(root + os.sep)
+
+
+def parse_filename(name: str) -> Optional[Dict[str, str]]:
+    match = FILENAME_RE.match(name)
+    if not match:
+        return None
+    return {
+        "run_date": match.group("run_date"),
+        "ticker": match.group("ticker").upper(),
+        "version": str(int(match.group("version") or "1")),
+    }
+
+
+def discover_reports(workdir: str, ticker: str) -> List[ReportFile]:
+    root = os.path.abspath(workdir)
+    ticker_upper = ticker.upper()
+    canonical_dir = canonical_reports_dir(root)
+    seen = set()
+    records: List[ReportFile] = []
+
+    for directory in compatible_dirs(root):
+        if not is_within_workdir(directory, root):
+            continue
+        if not os.path.isdir(directory):
+            continue
+        for entry in os.scandir(directory):
+            # Never follow symlinks for safety/privacy.
+            if not entry.is_file(follow_symlinks=False):
+                continue
+            parsed = parse_filename(entry.name)
+            if not parsed:
+                continue
+            if parsed["ticker"] != ticker_upper:
+                continue
+            abs_path = os.path.abspath(entry.path)
+            real_path = os.path.realpath(abs_path)
+            if real_path in seen:
+                continue
+            seen.add(real_path)
+            records.append(
+                ReportFile(
+                    path=abs_path,
+                    run_date=parsed["run_date"],
+                    ticker=parsed["ticker"],
+                    version=int(parsed["version"]),
+                    in_canonical_dir=os.path.dirname(abs_path) == canonical_dir,
+                )
+            )
+
+    def sort_key(record: ReportFile):
+        try:
+            d = date.fromisoformat(record.run_date)
+        except ValueError:
+            d = date.min
+        return (d, record.version, 1 if record.in_canonical_dir else 0)
+
+    return sorted(records, key=sort_key, reverse=True)
+
+
+def read_frontmatter(path: str) -> Dict[str, str]:
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            first_line = f.readline()
+            if first_line.strip() != "---":
+                return {}
+
+            # Read only a bounded header section to avoid loading large files.
+            frontmatter: Dict[str, str] = {}
+            total_chars = len(first_line)
+            for _ in range(200):
+                line = f.readline()
+                if not line:
+                    break
+                total_chars += len(line)
+                if total_chars > 64 * 1024:
+                    break
+                raw = line.rstrip("\n")
+                if raw.strip() == "---":
+                    break
+                if not raw.strip():
+                    continue
+                if raw.startswith("  - "):
+                    continue
+                if ":" not in raw:
+                    continue
+                key, value = raw.split(":", 1)
+                frontmatter[key.strip()] = value.strip()
+            return frontmatter
+    except (OSError, UnicodeDecodeError):
+        return {}
+
+
+def parse_float(value: Optional[str]) -> Optional[float]:
+    if value is None:
+        return None
+    text = value.strip()
+    if not text:
+        return None
+    if text.upper() in {"N/A", "NA", "NONE", "NULL", "PENDING"}:
+        return None
+    text = text.replace(",", "")
+    if text.endswith("%"):
+        text = text[:-1]
+    try:
+        return float(text)
+    except ValueError:
+        return None
+
+
+def parse_bool(value: Optional[str]) -> Optional[bool]:
+    if value is None:
+        return None
+    text = value.strip().lower()
+    if text in {"true", "yes", "1"}:
+        return True
+    if text in {"false", "no", "0"}:
+        return False
+    return None