From 72e36f7869815fdc7ae32443eb2ac64897d1996d Mon Sep 17 00:00:00 2001 From: Simon Oehrl <oehrl@vr.rwth-aachen.de> Date: Fri, 6 Mar 2020 16:26:23 +0100 Subject: [PATCH] Fix errors --- access_node/__main__.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/access_node/__main__.py b/access_node/__main__.py index a3f8b53..d9f6485 100644 --- a/access_node/__main__.py +++ b/access_node/__main__.py @@ -12,12 +12,12 @@ import time import requests import psycopg2 -def ConnectToDatabase(): +def ConnectToDatabase(postgres_username, postgres_password, port): return psycopg2.connect(database="postgres", user=postgres_username, password=postgres_password, host="database", port=str(port)) def SetupNestTables(postgres_username, postgres_password, port): - con = ConnectToDatabase() + con = ConnectToDatabase(postgres_username, postgres_password, port) print("Database connection opened successfully!") cur = con.cursor() @@ -54,7 +54,7 @@ def SetupNestTables(postgres_username, postgres_password, port): print("Nest tables created successfully!\n") def SetupArborTables(postgres_username, postgres_password, port): - con = ConnectToDatabase() + con = ConnectToDatabase(postgres_username, postgres_password, port) print("Database connection opened successfully!") cur = con.cursor() cur.execute("DROP TABLE IF EXISTS PROBES CASCADE") @@ -107,11 +107,11 @@ def main(): #SetupArborTables('postgres', 'docker', 5432) # Wait for simulation nodes to post to database - time.sleep(10) + time.sleep(5) # get simulation nodes - con = ConnectToDatabase() + con = ConnectToDatabase('postgres', 'docker', 5432) cur = con.cursor() # NEST cur.execute("SELECT ADDRESS FROM SIMULATION_NODES") -- GitLab