working connection

analytics/src/analytics/cli.py

@@ -8,12 +8,16 @@
 
 from analytics.datasets.deliverable_tasks import DeliverableTasks
 from analytics.datasets.sprint_board import SprintBoard
-from analytics.integrations import github, slack
+from analytics.integrations import github, slack, db
 from analytics.metrics.base import BaseMetric, Unit
 from analytics.metrics.burndown import SprintBurndown
 from analytics.metrics.burnup import SprintBurnup
 from analytics.metrics.percent_complete import DeliverablePercentComplete
 
+import logging
+import sqlalchemy
+logger = logging.getLogger(__name__)
+
 # fmt: off
 # Instantiate typer options with help text for the commands below
 SPRINT_FILE_ARG = typer.Option(help="Path to file with exported sprint data")
@@ -121,6 +125,17 @@ def calculate_sprint_burnup(
         output_dir=output_dir,
     )
 
+@export_app.command(name="test_connection")
+def test_connection(
+) -> None:
+    """Test function that ensures the DB connection works."""
+    engine = db.get_db()
+    # connection method from sqlalchemy
+    engine.connect() 
+    logger.warning("Connected to postgres db")
+    print(engine)
+    print(engine.connect())
+
 
 @metrics_app.command(name="deliverable_percent_complete")
 def calculate_deliverable_percent_complete(

analytics/src/analytics/integrations/db.py

@@ -18,5 +18,5 @@ def get_db() -> Engine:
     sqlalchemy.engine.Engine
     A SQLAlchemy engine object representing the connection to the database.
     """
-    # return create_engine( , pool_pre_ping=True)
+
     return create_engine(settings.database_url, pool_pre_ping=True)