Spaces:
Running on Zero
Running on Zero
| """Command-line entry point for GrowNXT institutional PDF reports. | |
| Usage: | |
| venv/Scripts/python.exe scripts/generate_report.py WIPRO | |
| venv/Scripts/python.exe scripts/generate_report.py WIPRO INFY --refresh | |
| """ | |
| import argparse | |
| import logging | |
| from pathlib import Path | |
| import sys | |
| sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) | |
| from core.config import safe_ticker # noqa: E402 | |
| from reporting.engine import ReportError, generate_report # noqa: E402 | |
| from scripts import cli # noqa: E402 | |
| def resolve_symbol(ticker_or_name: str) -> str: | |
| """Resolves an exchange symbol or company name to the canonical ticker symbol.""" | |
| raw = (ticker_or_name or "").strip() | |
| if not raw: | |
| return "" | |
| upper = safe_ticker(raw) | |
| # 1. Check if local directory cache already exists | |
| try: | |
| from core.config import OUTPUT_DIR | |
| if (OUTPUT_DIR / upper / "api" / "summary.json").exists() or (OUTPUT_DIR / upper / "summary.json").exists(): | |
| return upper | |
| except Exception: | |
| pass | |
| # 2. Lookup via stock discovery search API | |
| try: | |
| from api.search import find | |
| hits = find(raw) | |
| if hits: | |
| resolved = str(hits[0].get("ticker") or hits[0].get("sid") or upper).upper() | |
| if resolved != upper: | |
| logging.getLogger("generate_report").info("Resolved company '%s' -> ticker '%s'", raw, resolved) | |
| return safe_ticker(resolved) | |
| except Exception: | |
| pass | |
| return upper | |
| def _parser() -> argparse.ArgumentParser: | |
| """Builds the argument parser.""" | |
| p = argparse.ArgumentParser(description="Generate an institutional PDF equity report.") | |
| p.add_argument("tickers", nargs="+", help="Ticker symbols or company names, e.g. WIPRO, Infosys") | |
| p.add_argument("--output-dir", type=Path, | |
| help="Root of the per-ticker build workspace (default: output/).") | |
| p.add_argument("--keep-build", action="store_true", | |
| help="Keep the chart SVGs and Typst source beside the PDF.") | |
| p.add_argument("--refresh", action="store_true", | |
| help="Re-request collector data instead of using the local cache.") | |
| p.add_argument("--as-of", help="Display date for the header (default: today).") | |
| p.add_argument("--quiet", action="store_true", help="Log warnings and errors only.") | |
| return p | |
| def main() -> int: | |
| """Generates one report per ticker. | |
| Returns: | |
| 0 when every report was written, 1 otherwise. | |
| """ | |
| args = _parser().parse_args() | |
| cli.setup(logging.WARNING if args.quiet else logging.INFO, | |
| cli.PLAIN, stream=sys.stderr) | |
| failed = [] | |
| for ticker_input in args.tickers: | |
| sym = resolve_symbol(ticker_input) | |
| try: | |
| path = generate_report(sym, output_dir=args.output_dir, | |
| refresh=args.refresh, as_of=args.as_of, | |
| keep_build=args.keep_build) | |
| except Exception as exc: # noqa: BLE001 - report and continue the batch | |
| # A ReportError is about this company's data; anything else is a | |
| # defect, and the label says which so a batch log stays readable. | |
| label = "FAILED" if isinstance(exc, ReportError) else "ERROR" | |
| print("%-12s %s: %s" % (sym, label, exc), file=sys.stderr) | |
| failed.append(sym) | |
| else: | |
| print("%-12s %s" % (sym, path)) | |
| if failed: | |
| print("failed: %s" % ", ".join(failed), file=sys.stderr) | |
| return 1 if failed else 0 | |
| if __name__ == "__main__": | |
| sys.exit(main()) | |