from djagitit.db import SnowflakeDB, ReportingDB
from loguru import logger


_snw_instance: dict[str, SnowflakeDB] = {}
_rdb_instance: ReportingDB | None = None


def get_snw(warehouse: str) -> SnowflakeDB:
    if warehouse not in _snw_instance:
        logger.debug(f"Creating SnowflakeDB connection with warehouse {warehouse}")
        _snw_instance[warehouse] = SnowflakeDB(account="delphi.us-east-1", warehouse=warehouse)
    return _snw_instance[warehouse]


def get_rdb() -> ReportingDB:
    global _rdb_instance
    if _rdb_instance is None:
        logger.debug("Creating ReportingDB connection")
        _rdb_instance = ReportingDB()
    return _rdb_instance