76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
"""Factory for creating the appropriate report store backend.
|
|
|
|
Returns a :class:`MongoReportStore` when a MongoDB connection string is
|
|
configured, otherwise falls back to the filesystem :class:`ReportStore`.
|
|
|
|
Usage::
|
|
|
|
from tradingagents.portfolio.store_factory import create_report_store
|
|
|
|
store = create_report_store(run_id="a1b2c3d4")
|
|
store.save_scan("2026-03-20", {...})
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
from typing import Union
|
|
|
|
from tradingagents.portfolio.report_store import ReportStore
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def create_report_store(
|
|
run_id: str | None = None,
|
|
*,
|
|
base_dir: str | None = None,
|
|
mongo_uri: str | None = None,
|
|
mongo_db: str | None = None,
|
|
) -> Union[ReportStore, "MongoReportStore"]: # noqa: F821
|
|
"""Create and return the appropriate report store.
|
|
|
|
Resolution order for the backend:
|
|
|
|
1. If *mongo_uri* is passed explicitly, use MongoDB.
|
|
2. If ``TRADINGAGENTS_MONGO_URI`` env var is set, use MongoDB.
|
|
3. Fall back to the filesystem :class:`ReportStore`.
|
|
|
|
Args:
|
|
run_id: Short identifier for the current run.
|
|
base_dir: Override for the filesystem store's base directory.
|
|
mongo_uri: MongoDB connection string (overrides env var).
|
|
mongo_db: MongoDB database name (default ``"tradingagents"``).
|
|
|
|
Returns:
|
|
A store instance (either ``ReportStore`` or ``MongoReportStore``).
|
|
"""
|
|
uri = mongo_uri or os.getenv("TRADINGAGENTS_MONGO_URI", "")
|
|
db = mongo_db or os.getenv("TRADINGAGENTS_MONGO_DB", "tradingagents")
|
|
|
|
if uri:
|
|
try:
|
|
from tradingagents.portfolio.mongo_report_store import MongoReportStore
|
|
|
|
store = MongoReportStore(
|
|
connection_string=uri,
|
|
db_name=db,
|
|
run_id=run_id,
|
|
)
|
|
# ensure_indexes() is called automatically in __init__
|
|
logger.info("Using MongoDB report store (db=%s, run_id=%s)", db, run_id)
|
|
return store
|
|
except Exception:
|
|
logger.warning(
|
|
"MongoDB connection failed — falling back to filesystem store",
|
|
exc_info=True,
|
|
)
|
|
|
|
# Filesystem fallback
|
|
_base = base_dir or os.getenv("PORTFOLIO_DATA_DIR") or os.getenv(
|
|
"TRADINGAGENTS_REPORTS_DIR", "reports"
|
|
)
|
|
logger.info("Using filesystem report store (base=%s, run_id=%s)", _base, run_id)
|
|
return ReportStore(base_dir=_base, run_id=run_id)
|