return DB_result_set when iterating over DB_op results

1 files changed, 11 insertions, 0 deletions

diff --git a/src-py/neo4j_util.py b/src-py/neo4j_util.py
index 6b78dc18..7c228dea 100644
--- a/src-py/neo4j_util.py
+++ b/src-py/neo4j_util.py
@@ -7,6 +7,17 @@ import urllib2
 import model
 import string
 
+class DB_result_set(object):
+    def __init__(self, data):
+        self.data = data
+
+    def __iter__(self):
+        for db_row_dict in self.data['data']:
+            # example: dict: {u'row': [{u'title': u'foo'}]}
+            assert None != db_row_dict['row'][0]
+
+            yield db_row_dict['row'][0]
+
 class Cypher_String_Formatter(string.Formatter):
     """
     Despite parameter support in Cypher, we sometimes do engage in query string building