1 files changed, 5 insertions, 2 deletions
diff --git a/database.py b/database.py
index 23e06f0..846da4d 100644
--- a/database.py
+++ b/database.py
@@ -4,7 +4,7 @@ import os
 
 @dataclass
 class PowerDatabase:
-    host: str = "db"
+    host: str = None
     user: str = "root"
     passwd: str = None
     db: str = "power"
@@ -13,6 +13,9 @@ class PowerDatabase:
     def __enter__(self):
         if self.passwd is None:
             self.passwd = os.environ["MYSQL_ROOT_PASSWORD"]
+
+        if self.host is None:
+            self.host = os.environ["MYSQL_HOST"]
             
         try:
             self.__connection = self.__get_connection()
@@ -112,7 +115,7 @@ if __name__ == "__main__":
         dotenv.load_dotenv(dotenv_path = "db.env")
         host = "srv.athome"
     else:
-        host = "db"
+        host = None
 
     with PowerDatabase(host = host) as db:
         print(db.get_tasmota_devices())