diff --git a/models/connection.py b/models/connection.py index fb1a9e5..4cfe799 100644 --- a/models/connection.py +++ b/models/connection.py @@ -9,11 +9,11 @@ class DBConnection: def __init__(self, pool_size=5, max_overflow=10): - db_host = decouple_config("DB_HOST") - db_user = decouple_config("DB_USER") - db_pass = quote_plus(decouple_config("DB_PASS")) - db_name = decouple_config("DB_NAME") - db_port = decouple_config("DB_PORT") + db_host = decouple_config("DATABASE_HOST") + db_user = decouple_config("DATABASE_USER") + db_pass = quote_plus(decouple_config("DATABASE_PASSWORD")) + db_name = decouple_config("DATABASE_NAME") + db_port = decouple_config("DATABASE_PORT") self.db_engine = create_engine( f'mysql+pymysql://{db_user}:{db_pass}@{db_host}:{db_port}/{db_name}', poolclass=QueuePool,