Add route to load graph

api/graph.py

+from neo4j import GraphDatabase
+import time
+
+URL = 'jdbc:postgresql://db/discussion?user=postgres&password=FooBar'
+
+
+class Graph(object):
+    def __init__(self, uri, user, password):
+        self._driver = GraphDatabase.driver(uri, auth=(user, password))
+
+    def close(self):
+        self._driver.close()
+
+    def load(self):
+        t0 = self.run(self._delete_everything)
+        t1 = self.run(self._create_index_on_statement_uid)
+        t2 = self.run(self._create_statement_nodes)
+        t3 = self.run(self._fill_statements_with_content)
+        t4 = self.run(self._create_user_nodes)
+        t5 = self.run(self._create_relations_between_users_and_statements)
+        t6 = self.run(self._create_issue_nodes)
+        t7 = self.run(self._connect_issues_with_statements)
+        t8 = self.run(self._every_user_likes_his_position)
+        t9 = self.run(self._create_random_ratings)
+        t10 = self.run(self._delete_zero_ratings)
+
+        return {
+            "delete_time": t0,
+            "index_time": t1,
+            "statement_node_time": t2,
+            "statement_content_time": t3,
+            "user_node_time": t4,
+            "statement_user_relation_time": t5,
+            "issue_node_time": t6,
+            "statement_to_issue_time": t7,
+            "user_rates_own_positions_time": t8,
+            "random_ratings_time": t9,
+            "delete_zero_ratings_time": t10
+        }
+
+    def run(self, func):
+        t1 = time.time()
+        with self._driver.session() as session:
+            session.write_transaction(func)
+        return time.time() - t1
+
+    @staticmethod
+    def _create_index_on_statement_uid(tx):
+        result = tx.run(
+            "CREATE INDEX ON :Statement(uid) "
+        )
+        return result
+
+    @staticmethod
+    def _create_index_on_user_uid(tx):
+        result = tx.run(
+            "CREATE INDEX ON :Statement(uid) "
+        )
+        return result
+
+    @staticmethod
+    def _create_statement_nodes(tx):
+        result = tx.run(
+            "CALL apoc.load.jdbc($url, 'statements') "
+            "YIELD row "
+            "MERGE (statement:Statement{uid:row.uid, is_position:row.is_position, is_disabled:row.is_disabled})"
+            , url=URL
+        )
+        return result
+
+    @staticmethod
+    def _fill_statements_with_content(tx):
+        result = tx.run(
+            "CALL apoc.load.jdbc($url, 'textversions') "
+            "YIELD row "
+            "MATCH (statement:Statement{uid:row.statement_uid}) "
+            "WHERE NOT EXISTS(statement.content) "
+            "SET statement += {content:row.content}"
+            , url=URL
+        )
+        return result
+
+    @staticmethod
+    def _create_user_nodes(tx):
+        result = tx.run(
+            "CALL apoc.load.jdbc($url, 'users') "
+            "YIELD row "
+            "MERGE (user:User{uid:row.uid, public_nickname:row.public_nickname})"
+            , url=URL
+        )
+        return result
+
+    @staticmethod
+    def _create_relations_between_users_and_statements(tx):
+        result = tx.run(
+            "CALL apoc.load.jdbc($url, 'textversions') "
+            "YIELD row "
+            "MATCH (user:User), (statement:Statement) "
+            "WHERE user.uid = row.author_uid AND statement.uid = row.statement_uid "
+            "MERGE (user)-[:HAS_WRITTEN]->(statement)"
+            , url=URL
+        )
+        return result
+
+    @staticmethod
+    def _create_issue_nodes(tx):
+        result = tx.run(
+            "CALL apoc.load.jdbc($url, 'issues') "
+            "YIELD row "
+            "MERGE (issue:Issue{uid:row.uid, title:row.title})"
+            , url=URL
+        )
+        return result
+
+    @staticmethod
+    def _connect_issues_with_statements(tx):
+        result = tx.run(
+            "CALL apoc.load.jdbc($url, 'statement_to_issue') "
+            "YIELD row "
+            "MATCH (statement:Statement{uid:row.statement_uid}), (issue:Issue{uid:row.issue_uid}) "
+            "MERGE (statement)-[:WRITTEN_IN]->(issue)"
+            , url=URL
+        )
+        return result
+
+    @staticmethod
+    def _every_user_likes_his_position(tx):
+        result = tx.run(
+            "MATCH(user:User)-[:HAS_WRITTEN]->(statement:Statement{is_position:True}) "
+            "MERGE(user)-[:LIKES{rating: 5}]->(statement)"
+        )
+        return result
+
+    @staticmethod
+    def _create_random_ratings(tx):
+        result = tx.run(
+            "MATCH (user:User) "
+            "WHERE NOT EXISTS((user)-[:LIKES]->()) "
+            "MATCH (statement:Statement{is_position: True}) "
+            "MERGE (user)-[:LIKES{rating:round(rand()*5)}]->(statement)"
+        )
+        return result
+
+    @staticmethod
+    def _delete_zero_ratings(tx):
+        result = tx.run(
+            "MATCH ()-[r:LIKES{rating:0.0}]->() "
+            "DETACH DELETE r"
+        )
+        return result
+
+    @staticmethod
+    def _delete_everything(tx):
+        result = tx.run(
+            "MATCH (a) "
+            "DETACH DELETE a"
+        )
+        return result

api/requirements.txt

 Flask==0.10.1
 flask-cors==3.0.7
+neo4j==1.7.1
\ No newline at end of file

api/server.py

-from flask import Flask, render_template
+from flask import Flask, render_template, jsonify
 from flask_cors import CORS
 
+from api.graph import Graph
+
 app = Flask(__name__, template_folder='.')
 CORS(app)
 
@@ -10,5 +12,10 @@ def root(file):
     return render_template('html/' + file)
 
 
+@app.route('/load')
+def test():
+    return jsonify(Graph(uri='bolt://neo:7687', user='neo4j', password='neo4j').load())
+
+
 if __name__ == '__main__':
     app.run(debug=True, host="0.0.0.0", port=5000)