From 324c548bebf35ed0cb69a1e0be2abd7d6591ac5f Mon Sep 17 00:00:00 2001
From: Simon Oehrl <simon.oehrl@rwth-aachen.de>
Date: Sat, 4 Apr 2020 23:28:01 +0200
Subject: [PATCH] Add try-catch blocks

---
 access_node/__main__.py                     | 7 +++++--
 access_node/controllers/arbor_controller.py | 7 +++++--
 access_node/controllers/nest_controller.py  | 7 +++++--
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/access_node/__main__.py b/access_node/__main__.py
index 5516d73..a791940 100644
--- a/access_node/__main__.py
+++ b/access_node/__main__.py
@@ -14,8 +14,11 @@ import psycopg2
 
 def ConnectToDatabase(postgres_username, postgres_password):
     database_host = 'database'
-    with open('database_host.txt') as database_host_file:
-        database_host = database_host_file.readline().rstrip('\n')
+    try:
+        with open('database_host.txt') as database_host_file:
+            database_host = database_host_file.readline().rstrip('\n')
+    except:
+        pass
     return psycopg2.connect(database="postgres", user="postgres",
                        password="postgres", host=database_host, port="5432")
 
diff --git a/access_node/controllers/arbor_controller.py b/access_node/controllers/arbor_controller.py
index 9a6ff23..5ce7f77 100644
--- a/access_node/controllers/arbor_controller.py
+++ b/access_node/controllers/arbor_controller.py
@@ -16,8 +16,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().rstrip('\n')
+    try:
+        with open('database_host.txt') as database_host_file:
+            database_host = database_host_file.readline().rstrip('\n')
+    except:
+        pass
     return psycopg2.connect(database="postgres", user="postgres",
                        password="postgres", host=database_host, port="5432")
 
diff --git a/access_node/controllers/nest_controller.py b/access_node/controllers/nest_controller.py
index c2bdab3..14a3d52 100644
--- a/access_node/controllers/nest_controller.py
+++ b/access_node/controllers/nest_controller.py
@@ -16,8 +16,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().rstrip('\n')
+    try:
+        with open('database_host.txt') as database_host_file:
+            database_host = database_host_file.readline().rstrip('\n')
+    except:
+        pass
     return psycopg2.connect(database="postgres", user="postgres",
                        password="postgres", host=database_host, port="5432")
 
-- 
GitLab