From 4ce94e49774eb4fdc56a59241d515f17b6ae1c03 Mon Sep 17 00:00:00 2001
From: s-prechtl <stefan@tague.at>
Date: Thu, 12 Sep 2024 21:37:07 +0200
Subject: [PATCH] feat: additional option to provide other host for upgrader

---
 updater.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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'),