From b6d337323fa47c46f330da38c0a3354734728687 Mon Sep 17 00:00:00 2001
From: Simon Oehrl <oehrl@vr.rwth-aachen.de>
Date: Sat, 4 Apr 2020 19:21:06 +0200
Subject: [PATCH] Add optional database_host.txt file

---
 access_node/controllers/arbor_controller.py | 5 ++++-
 access_node/controllers/nest_controller.py  | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/access_node/controllers/arbor_controller.py b/access_node/controllers/arbor_controller.py
index 00a4ac4..2bdc3fa 100644
--- a/access_node/controllers/arbor_controller.py
+++ b/access_node/controllers/arbor_controller.py
@@ -15,8 +15,11 @@ import numpy as np
 
 
 def connect_to_database():
+    database_host = 'database'
+    with open('database_host.txt') as database_host_file:
+        database_host = database_host_file.readline()
     return psycopg2.connect(database="postgres", user="postgres",
-                       password="postgres", host="database", port="5432")
+                       password="postgres", host=database_host, port="5432")
 
 
 def arbor_get_attributes():  # noqa: E501
diff --git a/access_node/controllers/nest_controller.py b/access_node/controllers/nest_controller.py
index ffd4aa0..ffa94ea 100644
--- a/access_node/controllers/nest_controller.py
+++ b/access_node/controllers/nest_controller.py
@@ -15,8 +15,11 @@ import numpy as np
 
 
 def connect_to_database():
+    database_host = 'database'
+    with open('database_host.txt') as database_host_file:
+        database_host = database_host_file.readline()
     return psycopg2.connect(database="postgres", user="postgres",
-                       password="postgres", host="database", port="5432")
+                       password="postgres", host=database_host, port="5432")
 
 
 def nest_get_gids():  # noqa: E501
-- 
GitLab