add write queries for the algos

Author: Mark Needham

generate_notebook.py

@@ -49,6 +49,10 @@ def find_tag(file, tag):
 if len(sys.argv) >= 6:
     explanation_tag = sys.argv[5]
 
+write_graph_tag = "write-sample-graph"
+if len(sys.argv) >= 7:
+    write_graph_tag = sys.argv[6]
+
 heading_text = """\
 # {0}
 {1}
@@ -106,6 +110,21 @@ def find_tag(file, tag):
 
 streaming_graph_explanation_text = find_tag(algorithm_file, explanation_tag)
 
+write_graph_text = '''We can also call a version of the algorithm that will store the result as a property on a
+node. This is useful if we want to run future queries that use the result.'''
+
+write_query_content = find_tag(cypher_file, write_graph_tag)
+
+write_graph = '''\
+write_query = """\
+
+%s
+"""
+
+with driver.session() as session:
+    session.write_transaction(lambda tx: tx.run(write_query))''' % write_query_content
+
+
 viz_intro_text = '''## Graph Visualisation
 
 Sometimes a picture can tell more than a table of results and this is often the case with graph algorithms. 
@@ -222,6 +241,8 @@ def find_tag(file, tag):
                nbf.v4.new_markdown_cell(streaming_graph_text),
                nbf.v4.new_code_cell(run_algorithm),
                nbf.v4.new_markdown_cell(streaming_graph_explanation_text),
+               nbf.v4.new_markdown_cell(write_graph_text),
+               nbf.v4.new_code_cell(write_graph),
                nbf.v4.new_markdown_cell(viz_intro_text),
                nbf.v4.new_code_cell(neo_vis_div_cell),
                nbf.v4.new_markdown_cell(python_to_js_text),

generate_notebooks.sh

@@ -6,16 +6,16 @@ python generate_notebook.py \
     "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/pagerank.adoc" \
     "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/scripts/pagerank.cypher"
 
-python generate_notebook.py \
-    "Louvain" \
-    "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/louvain.adoc" \
-    "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/scripts/louvain.cypher"
-
 python generate_notebook.py \
     "Betweenness Centrality" \
     "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/betweenness-centrality.adoc" \
     "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/scripts/betweenness-centrality.cypher"
 
+python generate_notebook.py \
+    "Louvain" \
+    "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/louvain.adoc" \
+    "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/scripts/louvain.cypher"
+
 python generate_notebook.py \
     "Closeness Centrality" \
     "https://github.com/neo4j-contrib/neo4j-graph-algorithms/raw/3.2/doc/asciidoc/closeness-centrality.adoc" \