mantis-ib 0.2.0__tar.gz

mantis_ib-0.2.0/.gitignore

@@ -0,0 +1,15 @@
+# IDE files
+.vscode/
+ 
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+.pytest_cache/
+.python-version
+
+# Virtual environments
+.venv

mantis_ib-0.2.0/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.5
+Name: mantis-ib
+Version: 0.2.0
+Summary: Interactive Brokers source provider for MANTIS
+Author-email: Louis Zhuang <yzhuang@outlook.com>
+Requires-Python: >=3.14
+Requires-Dist: adbc-driver-postgresql>=1.12.0
+Requires-Dist: asof-store[sql-postgres,sql-sqlite]
+Requires-Dist: mantis-imf
+Requires-Dist: polars[adbc,pyarrow]>=2.0.0
+Requires-Dist: pydantic>=2.0
+Requires-Dist: service-registrar>=0.2.1
+Requires-Dist: zapros>=0.20.0
+Description-Content-Type: text/markdown
+
+# MANTIS Interactive Brokers provider
+
+This package provides the IB instrument source for MANTIS IMF through a
+`service-registrar` entry point and the provider-specific command line.
+
+Install `mantis-ib` alongside `mantis-imf` to enable the default IB resolver
+source and the `ib` CLI.
+
+## Commands
+
+```powershell
+ib index
+ib scrape-instruments
+ib scrape-reconcile
+ib scrape-funds
+ib scrape-exchanges
+```
+
+The `ib index` command requires PostgreSQL and adds trigram indexes to the
+instrument and template tables. The instrument scraping commands accept
+repeated `--product-type` and `--product-country` filters. Use `--dry-run` to
+print the remote product summary without opening a store. Pass `--sql-uri URI`
+to override the configured SQL URI for a command. Run `ib <command> --help` to
+see command-specific options.
+
+Configure the default SQL URI in the repository's `service-registrar.toml`:
+
+```toml
+["asof-store.sql"]
+sql_uri = "sqlite:///ib-instruments.db"
+```

mantis_ib-0.2.0/README.md

@@ -0,0 +1,31 @@
+# MANTIS Interactive Brokers provider
+
+This package provides the IB instrument source for MANTIS IMF through a
+`service-registrar` entry point and the provider-specific command line.
+
+Install `mantis-ib` alongside `mantis-imf` to enable the default IB resolver
+source and the `ib` CLI.
+
+## Commands
+
+```powershell
+ib index
+ib scrape-instruments
+ib scrape-reconcile
+ib scrape-funds
+ib scrape-exchanges
+```
+
+The `ib index` command requires PostgreSQL and adds trigram indexes to the
+instrument and template tables. The instrument scraping commands accept
+repeated `--product-type` and `--product-country` filters. Use `--dry-run` to
+print the remote product summary without opening a store. Pass `--sql-uri URI`
+to override the configured SQL URI for a command. Run `ib <command> --help` to
+see command-specific options.
+
+Configure the default SQL URI in the repository's `service-registrar.toml`:
+
+```toml
+["asof-store.sql"]
+sql_uri = "sqlite:///ib-instruments.db"
+```

mantis_ib-0.2.0/pyproject.toml

@@ -0,0 +1,32 @@
+[project]
+name = "mantis-ib"
+dynamic = ["version"]
+description = "Interactive Brokers source provider for MANTIS"
+readme = "README.md"
+authors = [{ name = "Louis Zhuang", email = "yzhuang@outlook.com" }]
+requires-python = ">=3.14"
+dependencies = [
+    "adbc-driver-postgresql>=1.12.0",
+    "asof-store[sql-postgres,sql-sqlite]",
+    "mantis-imf",
+    "polars[adbc,pyarrow]>=2.0.0",
+    "pydantic>=2.0",
+    "service-registrar>=0.2.1",
+    "zapros>=0.20.0",
+]
+
+[project.scripts]
+ib = "mantis_ib.cli:main"
+
+[project.entry-points."service_registrar.providers"]
+ib = "mantis_ib.plugin:IB_INSTRUMENT_SOURCE"
+
+[build-system]
+requires = ["hatchling", "uv-dynamic-versioning"]
+build-backend = "hatchling.build"
+
+[tool.hatch.version]
+source = "uv-dynamic-versioning"
+
+[tool.uv.sources]
+mantis-imf = { workspace = true }

mantis_ib-0.2.0/src/mantis_ib/__init__.py

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

mantis_ib-0.2.0/src/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])