adding new fields for link counts and linking back boolean

Author: kronoc

runner.py

@@ -1,5 +1,6 @@
 from bottle import route, run, template
 from neo4j.v1 import GraphDatabase, basic_auth
+from random import *
 
 driver = GraphDatabase.driver('bolt://localhost',auth=basic_auth("neo4j", "password"))
 
@@ -15,11 +16,19 @@ def get_graph(article):
     rels = []
     i = 0
     for record in results:
-        nodes.append({"title": record["link"], "url": generateUrl(record["link"])})
+#        nodes.append({"title": record["link"], "url": generateUrl(record["link"]), "rel_count": get_links_count(record["link"])})
+        nodes.append({"title": record["link"], "url": generateUrl(record["link"]), "rel_count": randint(1, 100), "rev_link": "true"})
         target = i
         i += 1
     return dict(pages=nodes)
 
+def get_links_count(article):
+    db = get_db()
+    query = 'MATCH (p:Page {title:"%s"})<-[:Link]-(o:Page) RETURN count(o)' % article.replace("_"," ")
+    result = db.run(query)
+    return result
+
+
 def generateUrl(title):
     url_root = "https://en.wikipedia.org/wiki/"
     return url_root+title.replace(" ","_")