diff --git a/database/db.py b/database/db.py index d86cd41..7a55afc 100644 --- a/database/db.py +++ b/database/db.py @@ -2,13 +2,16 @@ from sqlmodel import SQLModel, create_engine from utilities import config +from utilities.logger import agon_logger from . import models # noqa: F401 - Need this import for creating the models. if config.ENVIRONMENT == "prod": url = f"mysql+pymysql://{config.SQL_USER}:{config.SQL_PASS}@{config.SQL_HOST}/{config.SQL_DB}" + agon_logger.info("Connected to prod database.") else: url = "sqlite:///database.db" + agon_logger.info("Connected to local dev database.") engine = create_engine(url) # Don't need to create all when connected to prod.