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