Updated to new secret manager

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