python3 support (#35)

* python3 support

* set decode_responses=True

Author: gkorland

redisgraph/__init__.py

@@ -1,3 +1,3 @@
-from node import Node
-from edge import Edge
-from graph import Graph
+from .node import Node
+from .edge import Edge
+from .graph import Graph

redisgraph/edge.py

@@ -1,4 +1,4 @@
-from util import *
+from .util import *
 
 class Edge(object):
     """

redisgraph/graph.py

@@ -1,5 +1,5 @@
-from util import *
-from query_result import QueryResult
+from .util import *
+from .query_result import QueryResult
 class Graph(object):
     """
     Graph, collection of nodes and edges.

redisgraph/node.py

@@ -1,4 +1,4 @@
-from util import *
+from .util import *
 
 class Node(object):
     """

redisgraph/query_result.py

@@ -1,5 +1,5 @@
-from node import Node
-from edge import Edge
+from .node import Node
+from .edge import Edge
 from prettytable import PrettyTable
 
 class ResultSetColumnTypes(object):
@@ -72,7 +72,7 @@ def parse_records(self, raw_result_set):
                 elif self.header[idx][0] == ResultSetColumnTypes.COLUMN_RELATION:
                     record.append(self.parse_edge(cell))
                 else:
-                    print "Unknown column type.\n"
+                    print("Unknown column type.\n")
             records.append(record)
 
         return records
@@ -132,13 +132,13 @@ def parse_scalar(self, cell):
             elif value == "false":
                 scalar = False
             else:
-                print "Unknown boolean type\n"
+                print("Unknown boolean type\n")
 
         elif scalar_type == ResultSetScalarTypes.PROPERTY_DOUBLE:
             scalar = float(value)
 
         elif scalar_type == ResultSetScalarTypes.PROPERTY_UNKNOWN:
-            print "Unknown scalar type\n"
+            print("Unknown scalar type\n")
 
         return scalar
 
@@ -170,7 +170,7 @@ def pretty_print(self):
             print(str(tbl) + '\n')
 
         for stat in self.statistics:
-            print "%s %s" %(stat, self.statistics[stat])
+            print("%s %s" %(stat, self.statistics[stat]))
 
     def is_empty(self):
         return len(self.result_set) == 0

test.py

@@ -2,39 +2,39 @@
 import unittest
 from redisgraph import *
 
+
 class TestStringMethods(unittest.TestCase):
     def setUp(self):
-        self.r = redis.Redis(host='localhost', port=6379)
+        self.r = redis.Redis(host='localhost', port=6379, decode_responses=True)
 
     def test_graph_creation(self):
-		redis_graph = Graph('social', self.r)
-
-		john = Node(label='person', properties={'name': 'John Doe', 'age': 33, 'gender': 'male', 'status': 'single'})
-		redis_graph.add_node(john)
-
-		japan = Node(label='country', properties={'name': 'Japan'})
-		redis_graph.add_node(japan)
-
-		edge = Edge(john, 'visited', japan, properties={'purpose': 'pleasure'})
-		redis_graph.add_edge(edge)
-
-		redis_graph.commit()
-
-		query = """MATCH (p:person)-[v:visited {purpose:"pleasure"}]->(c:country)
+        redis_graph = Graph('social', self.r)
+        
+        john = Node(label='person', properties={'name': 'John Doe', 'age': 33, 'gender': 'male', 'status': 'single'})
+        redis_graph.add_node(john)
+        japan = Node(label='country', properties={'name': 'Japan'})
+        
+        redis_graph.add_node(japan)
+        edge = Edge(john, 'visited', japan, properties={'purpose': 'pleasure'})
+        redis_graph.add_edge(edge)
+        
+        redis_graph.commit()
+        
+        query = """MATCH (p:person)-[v:visited {purpose:"pleasure"}]->(c:country)
 				   RETURN p, v, c"""
-
-		result = redis_graph.query(query)
-
-		person = result.result_set[0][0]
-		visit = result.result_set[0][1]
-		country = result.result_set[0][2]
-
-		self.assertEqual(person, john)
-		self.assertEqual(visit.properties, edge.properties)		
-		self.assertEqual(country, japan)
+        
+        result = redis_graph.query(query)
+        
+        person = result.result_set[0][0]
+        visit = result.result_set[0][1]
+        country = result.result_set[0][2]
+        
+        self.assertEqual(person, john)
+        self.assertEqual(visit.properties, edge.properties)
+        self.assertEqual(country, japan)
 
 		# All done, remove graph.
-		redis_graph.delete()
+        redis_graph.delete()
 
 if __name__ == '__main__':
 	unittest.main()