small fixes & typing.

Author: dsevilla

neo4j/sesion7.ipynb

@@ -141,20 +141,20 @@
    },
    "outputs": [],
    "source": [
-    "from neo4j import GraphDatabase, basic_auth\n",
+    "from neo4j import Driver, GraphDatabase, Session, Result, basic_auth\n",
     "\n",
-    "driver = GraphDatabase.driver(\n",
+    "driver: Driver = GraphDatabase.driver(\n",
     "    BOLT_URL,\n",
     "    auth=basic_auth(\"neo4j\", \"\"))\n",
-    "session = driver.session()\n",
+    "session: Session = driver.session()\n",
     "\n",
     "cypher_query = '''\n",
     "MATCH (n)\n",
     "RETURN n AS id\n",
     "LIMIT 10\n",
     "'''\n",
     "\n",
-    "results = session.run(cypher_query,  parameters={})\n",
+    "results: Result = session.run(cypher_query,  parameters={})\n",
     "\n",
     "for record in results:\n",
     "  print(record['id'])"
@@ -177,14 +177,16 @@
    },
    "outputs": [],
    "source": [
+    "from neo4j import Result\n",
+    "\n",
     "query = '''\n",
     "MATCH (n)\n",
     "RETURN n\n",
     "LIMIT 10\n",
     "'''\n",
     "\n",
     "with driver.session() as session:\n",
-    "  results = session.run(query)\n",
+    "  results: Result = session.run(query)\n",
     "  for record in results:\n",
     "    print(record)"
    ]
@@ -199,10 +201,10 @@
    "source": [
     "from collections.abc import Generator\n",
     "from contextlib import contextmanager\n",
-    "from typing import Any\n",
+    "from neo4j import Result\n",
     "\n",
     "@contextmanager\n",
-    "def run_query(query) -> Generator[Any, Any, None]:\n",
+    "def run_query(query) -> Generator[Result, None, None]:\n",
     "  with driver.session() as session:\n",
     "    yield session.run(query)"
    ]
@@ -566,7 +568,7 @@
     "query=\"MATCH q=(r)-[:ANSWERS]->(p) RETURN p.Id,r.Id;\"\n",
     "\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  df = pd.DataFrame([r.values() for r in res], columns=res.keys())\n",
     "  df['r.Id'] = pd.to_numeric(df['r.Id'],downcast='unsigned')\n",
     "  df['p.Id'] = pd.to_numeric(df['p.Id'],downcast='unsigned')\n",
@@ -690,12 +692,12 @@
     "query = \"MATCH p=shortestPath( (u1:User {Id: '24'})-[*]-(u2:User {Id:'25'}) ) RETURN p\"\n",
     "\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  for r in res:\n",
     "    print(r['p'])\n",
     "\n",
     "with driver.session() as session:\n",
-    "  result = session.run(query)\n",
+    "  result: Result = session.run(query)\n",
     "  visualize_result(result.graph(), nodes_text_properties)"
    ]
   },
@@ -719,12 +721,12 @@
     "query=\"MATCH p=shortestPath( (u1:User {Id: '324'})-[*]-(u2:User {Id:'25'}) ) RETURN p\"\n",
     "\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  for r in res:\n",
     "    print(r['p'])\n",
     "\n",
     "with driver.session() as session:\n",
-    "  result = session.run(query)\n",
+    "  result: Result = session.run(query)\n",
     "  visualize_result(result.graph(), nodes_text_properties)"
    ]
   },
@@ -748,7 +750,7 @@
     "query=\"MATCH p=allShortestPaths( (u1:User {Id: '24'})-[*]-(u2:User {Id:'25'}) ) RETURN p\"\n",
     "\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  for r in res:\n",
     "    print(r['p'])"
    ]
@@ -779,7 +781,7 @@
     "'''\n",
     "\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  for r in res:\n",
     "    print(r)"
    ]
@@ -804,7 +806,7 @@
     "query=\"MATCH (t:Tag)-->(:Question)<--(u:User) RETURN t.name,collect(distinct u.Id) ORDER BY t.name;\"\n",
     "\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  for r in res:\n",
     "    print(r)"
    ]
@@ -829,7 +831,7 @@
    "source": [
     "query=\"MATCH (t:Tag)-->(:Question)<--(u:User) RETURN u.Id, collect(distinct t.name) ORDER BY toInteger(u.Id);\"\n",
     "with driver.session() as session:\n",
-    "  res = session.run(query)\n",
+    "  res: Result = session.run(query)\n",
     "  for r in res:\n",
     "    print(r)"
    ]