# MQTT live data — Phase 5: all functions now backed by Postgres
from database.pg_mqtt import (
    init_db,
    close_db,
    purge_loop,
    purge_old_data,
    insert_log,
    insert_heartbeat,
    insert_command,
    update_command_response,
    get_logs,
    get_heartbeats,
    get_commands,
    get_latest_heartbeats,
    get_pending_command,
    upsert_alert,
    delete_alert,
    get_alerts,
    partition_manager_loop,
    ensure_current_partitions,
)

# SQLite connection — still used by melodies, builder, manufacturing, and crm
# modules that have not yet been cut over to Postgres.
from database.core import get_db

__all__ = [
    "init_db",
    "close_db",
    "get_db",
    "purge_loop",
    "purge_old_data",
    "insert_log",
    "insert_heartbeat",
    "insert_command",
    "update_command_response",
    "get_logs",
    "get_heartbeats",
    "get_commands",
    "get_latest_heartbeats",
    "get_pending_command",
    "upsert_alert",
    "delete_alert",
    "get_alerts",
    "partition_manager_loop",
    "ensure_current_partitions",
]
