better handling of an empty result-set

Author: swilly22

redisgraph/client.py

@@ -53,7 +53,6 @@ def __str__(self):
 
         return res
 
-
 class Edge(object):
     """
     An edge connecting two nodes.
@@ -144,12 +143,9 @@ def query(self, q):
         result_set = None
         response = self.redis_con.execute_command("GRAPH.QUERY", self.name, q)
 
-        if len(response) == 1:
-            statistics = response[0]
-        else:
-            data = response[0]
-            statistics = response[1]
-            result_set = [res.decode().split(',') for res in data]
+        data = response[0]
+        statistics = response[1]
+        result_set = [res.decode().split(',') for res in data]
 
         return QueryResult(result_set, statistics)
 

redisgraph/query_result.py

@@ -24,7 +24,7 @@ def __init__(self, result_set, statistics):
        3. Prints the statistics of the query.
     """
     def pretty_print(self):
-        if self.result_set is not None:
+        if not self.is_empty():
             tbl = PrettyTable(self.result_set[0])
             for row in self.result_set[1:]:
                 tbl.add_row(row)
@@ -37,6 +37,9 @@ def pretty_print(self):
         for stat in self.statistics:
             print(stat)
 
+    def is_empty(self):
+        return len(self.result_set) == 0
+
     def _retrieve_data_from_statistics(self, statistics):
         return {
             self.LABELS_ADDED: self._get_value(self.LABELS_ADDED, statistics),

setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 setup(
     name='redisgraph',
-    version='1.2',
+    version='1.3',
 
     description='RedisGraph Python Client',
     url='https://github.com/swilly22/redisgraph-py',