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