From fa83c9ea7335d64e20eb01848d0cd3f9426f9fca Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan=20M=C3=BCller?= <j.mueller@vr.rwth-aachen.de> Date: Thu, 5 Mar 2020 13:24:07 +0100 Subject: [PATCH] Add initial arbor table design --- access_node/__main__.py | 39 ++++++++++++++++++++++++++++++++++++++- 1 file changed, 38 insertions(+), 1 deletion(-) diff --git a/access_node/__main__.py b/access_node/__main__.py index 755db66..29b726c 100644 --- a/access_node/__main__.py +++ b/access_node/__main__.py @@ -47,9 +47,46 @@ def SetupNestTables(postgres_username, postgres_password, port): FOREIGN KEY (MULTIMETER_ID) REFERENCES MULTIMETERS (MULTIMETER_ID));''') con.commit() - print("Tables created successfully!\n") + con.close() + print("Nest tables created successfully!\n") +def SetupArborTables(postgres_username, postgres_password, port): + con = psycopg2.connect(database="postgres", user=postgres_username, + password=postgres_password, host="database", port=str(port)) + print("Database connection opened successfully!") + cur = con.cursor() + cur.execute("DROP TABLE IF EXISTS PROBES CASCADE") + cur.execute("DROP TABLE IF EXISTS CELLS CASCADE") + cur.execute("DROP TABLE IF EXISTS SIMULATION_NODES CASCADE") + cur.execute("DROP TABLE IF EXISTS ATTRIBUTES CASCADE") + cur.execute('''CREATE TABLE SIMULATION_NODES ( + NODE_ID INT PRIMARY KEY NOT NULL UNIQUE, + ADDRESS VARCHAR(25), + CURRENT_SIM_TIME FLOAT);''') + cur.execute('''CREATE TABLE CELLS ( + CELL_ID INT PRIMARY KEY NOT NULL UNIQUE + );''') + cur.execute('''CREATE TABLE CELL_PROPERTIES ( + CELL_ID INT NOT NULL, + PROPERTY VARCHAR(50), + PRIMARY KEY (CELL_ID, PROPERTY), + FOREIGN KEY (CELL_ID) REFERENCES CELLS (CELL_ID), + );''') + cur.execute('''CREATE TABLE PROBES ( + PROBE_ID INT PRIMARY KEY NOT NULL UNIQUE, + CELL_ID INT, + SEGMENT_ID INT, + POSITION FLOAT, + NODE_ID INT, + FOREIGN KEY (NODE_ID) REFERENCES SIMULATION_NODES (NODE_ID), + FOREIGN KEY (CELL_ID) REFERENCES CELLS (CELL_ID));''') + cur.execute('''CREATE TABLE ATTRIBUTES ( + ATTRIBUTE_ID INT PRIMARY KEY NOT NULL UNIQUE, + NAME INT NOT NULL, + );''') + con.commit() con.close() + print("Arbor tables created successfully!\n") def main(): -- GitLab