Support both python2 and python3.

Author: srbaker

test/examples/autocommit_transaction_example.py

@@ -25,7 +25,7 @@
 
 class AutocommitTransactionExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(AutocommitTransactionExample, self).__init__(uri, user, password)
 
     # tag::autocommit-transaction[]
     def add_person(self, name):

test/examples/base_application.py

@@ -20,7 +20,7 @@
 
 from neo4j.v1 import GraphDatabase
 
-class BaseApplication:
+class BaseApplication(object):
     def __init__(self, uri, user, password):
         self._driver = GraphDatabase.driver( uri, auth=( user, password ) )
 

test/examples/cypher_error_example.py

@@ -25,7 +25,7 @@
 
 class CypherErrorExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(CypherErrorExample, self).__init__(uri, user, password)
 
     # FIXME: this doesn't work because read_transaction behaves
     # differently than in Java, so this throws a ClientError

test/examples/hello_world_example.py

@@ -26,7 +26,7 @@
 # tag::hello-world[]
 class HelloWorldExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(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) " +

test/examples/read_write_transaction_example.py

@@ -25,7 +25,7 @@
 
 class ReadWriteTransactionExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(ReadWriteTransactionExample, self).__init__(uri, user, password)
 
     # tag::read-write-transaction[]
     def add_person(self, name):

test/examples/result_consume_example.py

@@ -25,7 +25,7 @@
 
 class ResultConsumeExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(ResultConsumeExample, self).__init__(uri, user, password)
 
     # tag::result-consume[]
     def get_people(self):

test/examples/result_retain_example.py

@@ -25,7 +25,7 @@
 
 class ResultRetainExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(ResultRetainExample, self).__init__(uri, user, password)
 
     # tag::result-retain[]
     def add_employees(self, company_name):

test/examples/service_unavailable_example.py

@@ -25,7 +25,7 @@
 
 class ServiceUnavailableExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(ServiceUnavailableExample, self).__init__(uri, user, password)
 
     # tag::service-unavailable[]
     def addItem(self):

test/examples/session_example.py

@@ -24,7 +24,7 @@
 
 class SessionExample(BaseApplication):
     def __init__(self, uri, user, password):
-        super().__init__(uri, user, password)
+        super(SessionExample, self).__init__(uri, user, password)
 
     # tag::session[]
     def do_work(self):

test/examples/test_examples.py

@@ -20,6 +20,27 @@
 
 from test.integration.tools import IntegrationTestCase
 
+
+## Python2 doesn't have contextlib.redirect_stdout()
+#  http://eli.thegreenplace.net/2015/redirecting-all-kinds-of-stdout-in-python/
+import sys
+from contextlib import contextmanager
+@contextmanager
+def stdout_redirector(stream):
+    old_stdout = sys.stdout
+    sys.stdout = stream
+    try:
+        yield
+    finally:
+        sys.stdout = old_stdout
+
+def get_string_io():
+    if sys.version_info[0] < 3:
+        from StringIO import StringIO
+    else:
+        from io import StringIO
+    return StringIO()
+
 class ExamplesTest(IntegrationTestCase):
 
     def setUp(self):
@@ -65,12 +86,10 @@ def test_driver_lifecycle_example(self):
         self.assertIsInstance(example, DriverLifecycleExample)
 
     def test_hello_world_example(self):
-        import io
-        from contextlib import redirect_stdout
         from hello_world_example import HelloWorldExample
 
-        f = io.StringIO()
-        with redirect_stdout(f):
+        f = get_string_io()
+        with stdout_redirector(f):
             example = HelloWorldExample(self.bolt_uri, self.user, self.password)
             example.print_greeting("hello, world")
             example.close()