Updated to new secret manager

db_module/__init__.py

@@ -1,8 +1,16 @@
-from .db import get_db, create_connection_pool, close_connection_pool, start_healthcheck_thread
+import logging
 
+from .db import (
+	close_connection_pool,
+	create_connection_pool,
+	get_db,
+	start_healthcheck_thread,
+)
+
+logging.getLogger(__name__).addHandler(logging.NullHandler())
 __all__ = [
-	"get_db",
-	"create_connection_pool",
 	"close_connection_pool",
+	"create_connection_pool",
+	"get_db",
 	"start_healthcheck_thread",
 ]

db_module/db.py

@@ -1,16 +1,17 @@
-from mysql.connector import pooling, Error
-from mysql.connector.pooling import MySQLConnectionPool, MySQLConnection
-import threading
-from secret_manager import return_credentials
+import logging
 import os
+import threading
 import time
-from simple_logger_handler import setup_logger
-from typing import Generator
+from collections.abc import Generator
 
-db_username = return_credentials("/etc/secrets/db_username")
-db_password = return_credentials("/etc/secrets/db_password")
+from mysql.connector import Error, pooling
+from mysql.connector.pooling import MySQLConnection, MySQLConnectionPool
+from secret_manager import get_secret
 
-logger = setup_logger(__name__)
+db_username = get_secret("/etc/secrets/db_username")
+db_password = get_secret("/etc/secrets/db_password")
+
+logger = logging.getLogger(__name__)
 
 MAX_RETRIES = 10
 RETRY_DELAY = 5