Idiomaticised examples

Author: technige

test/examples/autocommit_transaction_example.py

@@ -18,17 +18,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# tag::autocommit-transaction-import[]
-from neo4j.v1 import Session;
 from test.examples.base_application import BaseApplication
+
+# tag::autocommit-transaction-import[]
 # end::autocommit-transaction-import[]
 
+
 class AutocommitTransactionExample(BaseApplication):
     def __init__(self, uri, user, password):
         super(AutocommitTransactionExample, self).__init__(uri, user, password)
 
     # tag::autocommit-transaction[]
     def add_person(self, name):
-        session = self._driver.session()
-        session.run( "CREATE (a:Person {name: $name})", {"name": name} )
+        with self._driver.session() as session:
+            session.run("CREATE (a:Person {name: $name})", name=name)
     # end::autocommit-transaction[]

test/examples/base_application.py

@@ -23,7 +23,7 @@
 
 class BaseApplication(object):
     def __init__(self, uri, user, password):
-        self._driver = GraphDatabase.driver(uri, auth=( user, password ), max_retry_time=0)
+        self._driver = GraphDatabase.driver(uri, auth=(user, password), max_retry_time=0)
 
     def close(self):
         self._driver.close()

test/examples/basic_auth_example.py

@@ -33,5 +33,5 @@ def close(self):
         self._driver.close()
 
     def can_connect(self):
-        record_list = list(self._driver.session().run("RETURN 1"))
-        return int(record_list[0][0]) == 1
+        result = self._driver.session().run("RETURN 1")
+        return result.single()[0] == 1

test/examples/config_connection_timeout_example.py

@@ -22,12 +22,12 @@
 from neo4j.v1 import GraphDatabase
 # end::config-connection-timeout-import[]
 
+
 class ConfigConnectionTimeoutExample:
     # tag::config-connection-timeout[]
     def __init__(self, uri, user, password):
-        self._driver = GraphDatabase.driver( uri, auth=( user, password ),
-                Config.build().withConnectionTimeout( 15, SECONDS ).toConfig() )
+        self._driver = GraphDatabase.driver(uri, auth=(user, password), connection_timeout=15)
     # end::config-connection-timeout[]
 
     def close(self):
-        self._driver.close();
+        self._driver.close()

test/examples/config_max_retry_time_example.py

@@ -30,4 +30,4 @@ def __init__(self, uri, user, password):
     # end::config-max-retry-time[]
 
     def close(self):
-        self._driver.close();
+        self._driver.close()

test/examples/config_trust_example.py

@@ -30,4 +30,4 @@ def __init__(self, uri, user, password):
     # end::config-trust[]
 
     def close(self):
-        self._driver.close();
+        self._driver.close()

test/examples/custom_auth_example.py

@@ -33,5 +33,6 @@ def close(self):
         self._driver.close()
 
     def can_connect(self):
-        record_list = list(self._driver.session().run("RETURN 1"))
-        return int(record_list[0][0]) == 1
+        with self._driver.session() as session:
+            result = session.run("RETURN 1")
+            return result.single()[0] == 1

test/examples/cypher_error_example.py

@@ -18,25 +18,28 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# tag::cypher-error-import[]
-from neo4j.v1 import GraphDatabase, ClientError
 from test.examples.base_application import BaseApplication
+
+# tag::cypher-error-import[]
+from neo4j.v1 import ClientError
 # end::cypher-error-import[]
 
+
 class CypherErrorExample(BaseApplication):
     def __init__(self, uri, user, password):
         super(CypherErrorExample, self).__init__(uri, user, password)
 
     # tag::cypher-error[]
     def get_employee_number(self, name):
-        session =  self._driver.session()
-        session.read_transaction(lambda tx: self.select_employee(tx, name))
+        with self._driver.session() as session:
+            return session.read_transaction(self.select_employee, name)
 
-    def select_employee(self, tx, name):
+    @staticmethod
+    def select_employee(tx, name):
         try:
-            record_list = list(tx.run("SELECT * FROM Employees WHERE name = $name", {"name": name}))
-            return int(record_list[0]["employee_number"])
-        except ClientError as e:
-            print(e.message)
+            result = tx.run("SELECT * FROM Employees WHERE name = $name", name=name)
+            return result.single()["employee_number"]
+        except ClientError as error:
+            print(error.message)
             return -1
     # end::cypher-error[]

test/examples/driver_lifecycle_example.py

@@ -22,11 +22,12 @@
 from neo4j.v1 import GraphDatabase
 # end::driver-lifecycle-import[]
 
+
 # tag::driver-lifecycle[]
 class DriverLifecycleExample:
     def __init__(self, uri, user, password):
         self._driver = GraphDatabase.driver(uri, auth=(user, password))
 
     def close(self):
-        self._driver.close();
+        self._driver.close()
 # end::driver-lifecycle[]

test/examples/hello_world_example.py

@@ -20,25 +20,29 @@
 
 # tag::hello-world-import[]
 from neo4j.v1 import GraphDatabase
-from test.examples.base_application import BaseApplication
 # end::hello-world-import[]
 
+
 # tag::hello-world[]
-class HelloWorldExample(BaseApplication):
+class HelloWorldExample(object):
+
     def __init__(self, uri, user, password):
-        super(HelloWorldExample, self).__init__(uri, user, password)
-
-    def _create_and_return_greeting(self, tx, message):
-        record_list = list(tx.run("CREATE (a:Greeting) " +
-                                  "SET a.message = $message " +
-                                  "RETURN a.message + ', from node ' + id(a)",
-                                  {"message": message}))
-        return str(record_list[0][0])
-        
+        self._driver = GraphDatabase.driver(uri, auth=(user, password))
+
+    def close(self):
+        self._driver.close()
+
     def print_greeting(self, message):
         with self._driver.session() as session:
-            greeting = session.write_transaction(lambda tx: self._create_and_return_greeting(tx, message))
+            greeting = session.write_transaction(self._create_and_return_greeting, message)
             print(greeting)
+
+    @staticmethod
+    def _create_and_return_greeting(tx, message):
+        result = tx.run("CREATE (a:Greeting) "
+                        "SET a.message = $message "
+                        "RETURN a.message + ', from node ' + id(a)", message=message)
+        return result.single()[0]
 # end::hello-world[]
 
 # tag::hello-world-output[]