stock_advisor.py

#!/usr/bin/env python3
from __future__ import annotations

import argparse

from stock_analysis.advisor import build_advice
from stock_analysis.report import advice_to_json, render_text_report
from stock_analysis.technical import analyze_technical


def parse_args(argv=None):
    parser = argparse.ArgumentParser(description="街哥核心战法股票分析")
    parser.add_argument("ticker", help="股票代码，如 NVDA、0700.HK")
    parser.add_argument("--horizons", default="short,medium,long", help="分析周期")
    parser.add_argument("--json", action="store_true", dest="as_json", help="输出JSON")
    return parser.parse_args(argv)


def main(argv=None) -> int:
    args = parse_args(argv)
    timeframes = ["daily", "weekly"]
    if "short" in args.horizons:
        timeframes.insert(0, "4h")

    technical = analyze_technical(args.ticker, timeframes=timeframes)
    advice = build_advice(args.ticker, technical=technical)

    if args.as_json:
        print(advice_to_json(advice))
    else:
        print(render_text_report(advice))
    return 0


if __name__ == "__main__":
    raise SystemExit(main())