Added .dicts() method to results

Author: catherinedevlin

README.rst

@@ -109,6 +109,9 @@ leftmost column serving as key, for unique values.
     In [15]: result['richard2']
     Out[15]: (u'richard2', u'Richard II', u'History of Richard II', 1595, u'h', None, u'Moby', 22411, 628)
 
+Results can also be retrieved as an iterator of dictionaries (``result.dicts()``)
+or a single dictionary with a tuple of scalar values per key (``result.dict()``)
+
 Assignment
 ----------
 
@@ -276,7 +279,7 @@ Credits
 - Andrés Celis for SQL Server bugfix
 - Michael Erasmus for DataFrame truth bugfix
 - Noam Finkelstein for README clarification
-- Xiaochuan Yu for `<<` operator
+- Xiaochuan Yu for `<<` operator, syntax colorization
 
 .. _Distribute: http://pypi.python.org/pypi/distribute
 .. _Buildout: http://www.buildout.org/

src/sql/run.py

@@ -140,14 +140,22 @@ def __getitem__(self, key):
                 raise KeyError('%d results for "%s"' % (len(result), key))
             return result[0]
     def dict(self):
-        "Returns a dict built from the result set, with column names as keys"
+        """Returns a single dict built from the result set
+ 
+        Keys are column names; values are a tuple"""
         return dict(zip(self.keys, zip(*self)))
 
+    def dicts(self):
+        "Iterator yielding a dict for each row"
+        for row in self:
+            yield dict(zip(self.keys, row))
+
     def DataFrame(self):
         "Returns a Pandas DataFrame instance built from the result set."
         import pandas as pd
         frame = pd.DataFrame(self, columns=(self and self.keys) or [])
         return frame
+
     def pie(self, key_word_sep=" ", title=None, **kwargs):
         """Generates a pylab pie chart from the result set.
 

src/tests/test_magic.py

@@ -191,3 +191,23 @@ def test_csv_to_file():
             for row in content.splitlines():
                 assert row.count(',') == 1
             assert len(content.splitlines()) == 3
+
+@with_setup(_setup_writer, _teardown_writer)
+def test_dict():
+    result = ip.run_line_magic('sql',  "sqlite:// SELECT * FROM writer;")
+    result = result.dict()
+    assert isinstance(result, dict)
+    assert 'first_name' in result
+    assert 'last_name' in result
+    assert 'year_of_death' in result
+    assert len(result['last_name']) == 2
+
+@with_setup(_setup_writer, _teardown_writer)
+def test_dicts():
+    result = ip.run_line_magic('sql',  "sqlite:// SELECT * FROM writer;")
+    for row in result.dicts():
+        assert isinstance(row, dict)
+        assert 'first_name' in row
+        assert 'last_name' in row
+        assert 'year_of_death' in row
+