From 44feb45b3fffe2d86597f355aff586c6b8991b19 Mon Sep 17 00:00:00 2001 From: Guillaume Schurck <g.schurck@gmail.com> Date: Sun, 20 Nov 2022 16:40:06 +0100 Subject: [PATCH] update db requests --- sg-backend/db.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/sg-backend/db.py b/sg-backend/db.py index 06dbcd1..fde0325 100644 --- a/sg-backend/db.py +++ b/sg-backend/db.py @@ -54,12 +54,13 @@ class Neo4j: def get_relations(self): with self.driver.session() as session: - results = session.run("MATCH p = (a)-[r]->(b) RETURN a.name as source,b.name as target, r as relation_type") + results = session.run( + "MATCH p = (a:Software)-[r]->(b:Software) RETURN a.name as source,b.name as target, r as relation_type") return results.data() def get_nodes(self): with self.driver.session() as session: - results = session.run("MATCH (a) return Id(a) as key, a.name as name") + results = session.run("MATCH (a:Software) return Id(a) as key, a.name as name") return results.data() def insert_software(self, insert_data): @@ -84,4 +85,4 @@ class Neo4j: results = session.run( "MATCH (a:Software), (b:Software) WHERE a.name = '%s' AND b.name = '%s' CREATE (a)-[r:%s]->(b) " "RETURN type(r)" % (insert_data.software1, insert_data.software2, insert_data.relationType)) - return results.data() + return results.data() \ No newline at end of file -- GitLab