diff --git a/updater.py b/updater.py index d484ea4..ae8a29d 100644 --- a/updater.py +++ b/updater.py @@ -4,8 +4,9 @@ from dotenv import load_dotenv load_dotenv() +host = os.getenv('DB_HOST_UPGRADE') if os.getenv('DB_HOST_UPGRADE') else os.getenv('DB_UPGRADE') conn = psycopg2.connect( - host=os.getenv('DB_HOST'), + host=host, database=os.getenv('DB_NAME'), user=os.getenv('DB_USER'), password=os.getenv('DB_PASSWORD'),