diff --git a/alembic.ini b/alembic.ini index 370f462..4796f76 100644 --- a/alembic.ini +++ b/alembic.ini @@ -60,7 +60,7 @@ version_path_separator = os # Use os.pathsep. Default configuration used for ne # are written from script.py.mako # output_encoding = utf-8 -sqlalchemy.url = mysql+pymysql://admin:PWM_556rtX@192.168.0.113:3306/league_db +sqlalchemy.url = mysql+pymysql://admin:PWM_556rtX@192.168.0.113:3306/league_prod [post_write_hooks] diff --git a/data/database.py b/data/database.py index 5f0daa1..bac929c 100644 --- a/data/database.py +++ b/data/database.py @@ -1,4 +1,3 @@ - import mariadb import os from dotenv import load_dotenv @@ -6,25 +5,20 @@ from dotenv import load_dotenv load_dotenv() def db_connection(): + # Wir wählen die Konfiguration basierend auf der URL + is_prod = os.getenv("APP_URL") == "https://liga.au-fab.eu" + + # Prefix für die Umgebungsvariablen wählen + prefix = "DB_" if is_prod else "DB_TEST_" + try: connection = mariadb.connect( host=os.getenv("DB_HOST", "localhost"), port=int(os.getenv("DB_PORT", 3306)), - user=os.getenv("DB_USER"), + user=os.getenv(f"{prefix}USER"), password=os.getenv("DB_PASSWORD"), - database=os.getenv("DB_NAME") + database=os.getenv(f"{prefix}NAME") ) return connection except mariadb.Error as e: - raise RuntimeError(f"Fehler beim Verbinden mit DB: {e}") - - - - - - - - - - - + raise RuntimeError(f"Fehler beim Verbinden mit {'Prod' if is_prod else 'Test'}-DB: {e}")