mantis-ib 0.2.0__py3-none-any.whl

mantis_ib/__init__.py

@@ -0,0 +1 @@
+"""Interactive Brokers provider for MANTIS IMF."""

mantis_ib/cli.py

@@ -0,0 +1,55 @@
+"""Interactive Brokers command line entry point."""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from collections.abc import Sequence
+
+from .scraper import main as scraper_main
+from .search import create_search_indexes
+
+_SCRAPER_COMMANDS = (
+    "scrape-instruments",
+    "scrape-reconcile",
+    "scrape-funds",
+    "scrape-exchanges",
+)
+
+
+def _build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="ib",
+        description="Search, index, and scrape Interactive Brokers instruments.",
+    )
+    parser.add_argument("command", nargs="?", choices=("index", *_SCRAPER_COMMANDS))
+    parser.add_argument("arguments", nargs=argparse.REMAINDER)
+    return parser
+
+
+def _run_index(argv: Sequence[str]) -> int:
+    parser = argparse.ArgumentParser(prog="ib index")
+    parser.add_argument("--sql-uri")
+    parser.add_argument("--table-name", default="ib_instruments")
+    parser.add_argument(
+        "--missing-primary-key-table-name",
+        default="ib_instrument_templates",
+    )
+    args = parser.parse_args(argv)
+    return create_search_indexes(
+        sql_uri=args.sql_uri,
+        table_name=args.table_name,
+        missing_primary_key_table_name=args.missing_primary_key_table_name,
+    )
+
+
+def main(argv: Sequence[str] | None = None) -> int:
+    command_line = list(sys.argv[1:] if argv is None else argv)
+    parser = _build_parser()
+    args = parser.parse_args(command_line)
+    if args.command is None:
+        parser.print_help()
+        return 0
+    if args.command == "index":
+        return _run_index(args.arguments)
+    return scraper_main([args.command, *args.arguments])