major fixes

Author: matheusmota

.ipython/extensions/neo4j_cypher_query.py

@@ -1,21 +1,47 @@
+import pandas as pd
 from yfiles_jupyter_graphs_for_neo4j import Neo4jGraphWidget
 from neo4j import GraphDatabase
 from yfiles_jupyter_graphs import GraphWidget
 from IPython.core.magic import (register_line_magic,
                                 register_cell_magic)
 
+
+
 driver = GraphDatabase.driver(uri = "neo4j://localhost")
 
-def query_neo4j(cypher_query: str):
-    query_graph_result = driver.session().run(cypher_query).graph()
-    graph_widget = GraphWidget(overview_enabled=False, context_start_with='Data', graph = query_graph_result)
+def print_counters(counters):
+    for attr in dir(counters):
+        # Filter out private methods and attributes
+        if not attr.startswith('_'):
+            value = getattr(counters, attr)
+            if isinstance(value, int):
+                if value > 0:
+                    print(f"{attr.replace('_', ' ').capitalize()}: {value}")
+
+def query_neo4j_output_graph(cypher_query: str):
+    query_graph_result = driver.session().run(cypher_query)
+    graph_widget = GraphWidget(overview_enabled=False, context_start_with='Data', graph = query_graph_result.graph())
     graph_widget.set_sidebar(enabled=False, start_with='Data')
+    print_counters(query_graph_result.consume().counters)
     display(graph_widget)
     return graph_widget
 
+def query_neo4j_output_table(cypher_query: str):
+    query_graph_result = driver.session().run(cypher_query)
+    df = pd.DataFrame(query_graph_result.data())
+    pd.set_option('display.max_colwidth', None)
+    print_counters(query_graph_result.consume().counters)
+    display(df)
+    return df
+
+
+
 def cypher(line, cell_content):
-    print(line)
-    query_neo4j(cell_content)
+    
+    if line == 'graph':
+        query_neo4j_output_graph(cell_content)
+    else:
+        query_neo4j_output_table(cell_content)
 
 def load_ipython_extension(ipython):
     """This function is called when the extension is
@@ -24,4 +50,4 @@ def load_ipython_extension(ipython):
     `register_magic_function` method of the shell
     instance."""
     ipython.register_magic_function(cypher, 'cell')
-    
+    

labs/pt-br/1.Cypher Basics-Copy1.ipynb

@@ -16,7 +16,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "metadata": {
     "tags": []
    },
@@ -26,30 +26,6 @@
     "from datasci4health.graphdrawer.visgraph import draw"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "<ipython-input-3-548a69b4459e>:5: ExperimentalWarning: The configuration may change in the future.\n",
-      "  driver.verify_connectivity()\n"
-     ]
-    }
-   ],
-   "source": [
-    "from neo4j import GraphDatabase\n",
-    "\n",
-    "URI = \"neo4j://localhost\"\n",
-    "with GraphDatabase.driver(URI) as driver:\n",
-    "    driver.verify_connectivity()"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 4,
@@ -86,17 +62,6 @@
     "https://ipython-cypher.readthedocs.io/en/latest/introduction.html\n"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "!python -m graph_notebook.ipython_profile.configure_ipython_profile"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 6,