Source code for disclosure_alpha.cli

#!/usr/bin/env python3
"""CLI for deterministic SEC filing analytics."""

from __future__ import annotations

import argparse
import json
import sys
from pathlib import Path

from disclosure_alpha.pipeline import (
    compute_section_metrics,
    extract_sections_from_html,
    score_deterministic,
    score_filing_html,
    score_filing_ticker,
)


def _load_html(path: str | None) -> str:
    if path and path != "-":
        return Path(path).read_text(encoding="utf-8", errors="ignore")
    return sys.stdin.read()



[docs]
def main() -> None:
    parser = argparse.ArgumentParser(
        description="Disclosure Alpha — deterministic SEC filing analytics"
    )
    sub = parser.add_subparsers(dest="command", required=True)

    extract_p = sub.add_parser("extract", help="Extract sections from filing HTML")
    extract_p.add_argument("--html", required=True, help="Path to HTML file or '-' for stdin")
    extract_p.add_argument("--form", required=True, help="Form type, e.g. 10-K")

    score_p = sub.add_parser("score", help="Full pipeline → deterministic scores")
    src = score_p.add_mutually_exclusive_group(required=True)
    src.add_argument("--html", help="Path to HTML file or '-' for stdin")
    src.add_argument("--ticker", help="Ticker symbol (fetches from SEC EDGAR)")
    score_p.add_argument(
        "--form",
        default="10-K",
        help="10-K or 10-Q (EDGAR/ticker); 8-K supported with --html only",
    )
    score_p.add_argument("--fiscal-year", type=int, help="Fiscal year (with --ticker)")
    score_p.add_argument("--quarter", choices=["Q1", "Q2", "Q3"], help="Required for 10-Q")
    score_p.add_argument("--prior-html", help="Optional prior filing HTML for diffs")

    metrics_p = sub.add_parser("metrics", help="Extract + compute section metrics")
    metrics_p.add_argument("--html", required=True)
    metrics_p.add_argument("--form", required=True)
    metrics_p.add_argument("--prior-html")

    args = parser.parse_args()

    if args.command == "extract":
        html = _load_html(args.html)
        sections = extract_sections_from_html(html, args.form)
        out = [
            {
                "section_name": s.section_name,
                "word_count": s.word_count,
                "extraction_confidence": s.extraction_confidence,
                "parser_version": s.parser_version,
            }
            for s in sections
        ]
        print(json.dumps(out, indent=2))
        return

    if args.command == "metrics":
        html = _load_html(args.html)
        sections = extract_sections_from_html(html, args.form)
        prior_sections = None
        if args.prior_html:
            prior_html = _load_html(args.prior_html)
            prior_sections = extract_sections_from_html(prior_html, args.form)
        metrics = compute_section_metrics(sections, prior_sections)
        print(json.dumps(metrics.__dict__, indent=2, default=str))
        return

    if args.command == "score":
        if args.ticker:
            if args.fiscal_year is None:
                score_p.error("--fiscal-year is required with --ticker")
            result = score_filing_ticker(
                args.ticker,
                args.fiscal_year,
                form_type=args.form,
                quarter=args.quarter,
            )
        else:
            html = _load_html(args.html)
            prior_html = _load_html(args.prior_html) if args.prior_html else None
            result = score_filing_html(html, args.form, prior_html=prior_html)
        print(json.dumps(result.to_dict(), indent=2, default=str))
        return



if __name__ == "__main__":
    main()