support for embedded fields

This commit is contained in:
Wilson Júnior
2011-12-13 07:46:49 -02:00
parent 4a269eb2c4
commit ed5fba6b0f
2 changed files with 53 additions and 2 deletions

View File

@@ -320,12 +320,22 @@ class SelectResult(object):
"""
def __init__(self, cursor, fields):
self._cursor = cursor
self._fields = fields
self._fields = [f.split('.') for f in fields]
def _get_value(self, keys, data):
for key in keys:
if data:
data = data.get(key)
else:
break
return data
def next(self):
try:
data = self._cursor.next()
return [data.get(f) for f in self._fields]
return [self._get_value(f, data)
for f in self._fields]
except StopIteration, e:
self.rewind()
raise e