"""direct_delivery database connection setup.""" import os from src.configs.config import secrets_manager_client DD_DB_CREDENTIALS = { 'user': os.environ.get('DD_MYSQL_USER'), 'password': secrets_manager_client.get_cred('DD_MYSQL_PASSWORD'), 'host': os.environ.get('DD_MYSQL_HOST'), 'port': os.environ.get('DD_MYSQL_PORT', '3306'), 'database': os.environ.get('DD_MYSQL_DATABASE', 'direct_delivery'), } DD_DB_CONNECTION_STRING = ( 'mysql+pymysql://{user}:{password}@{host}:{port}/{database}'.format( user=DD_DB_CREDENTIALS.get('user'), password=DD_DB_CREDENTIALS.get('password'), host=DD_DB_CREDENTIALS.get('host'), port=DD_DB_CREDENTIALS.get('port'), database=DD_DB_CREDENTIALS.get('database') ) ) DD_DB_CONNECT_ARGS = { 'charset': 'utf8', 'connect_timeout': int(os.environ.get('DD_MYSQL_CONNECT_TIMEOUT') or 15), }