From 8e20a72f5cd0464f673a374e5a3741a58932e770 Mon Sep 17 00:00:00 2001 From: LV-426 Date: Sun, 12 Oct 2014 15:17:17 +0200 Subject: return DB_result_set when iterating over DB_op results --- src-py/neo4j_util.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'src-py/neo4j_util.py') 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 -- cgit v1.3.1