Merge pull request #43 from svetlyak40wt/develop

Render table in Markdown format on unicode(dataset). Closes #41.
This commit is contained in:
Kenneth Reitz
2011-12-24 23:05:03 -08:00
2 changed files with 33 additions and 0 deletions
+17
View File
@@ -223,6 +223,23 @@ class Dataset(object):
except AttributeError:
return '<dataset object>'
def __unicode__(self):
result = [self.__headers]
result.extend(map(unicode, row) for row in self._data)
# here, we calculate max width for each column
lens = (map(len, row) for row in result)
field_lens = map(max, zip(*lens))
# delimiter between header and data
result.insert(1, [u'-' * length for length in field_lens])
format_string = u'|'.join(u'{%s:%s}' % item for item in enumerate(field_lens))
return u'\n'.join(format_string.format(*row) for row in result)
# ---------
# Internals
+16
View File
@@ -649,6 +649,22 @@ class TablibTestCase(unittest.TestCase):
self.assertEquals(self.founders[orig_target_header], data[target_header])
def test_unicode_renders_markdown_table(self):
# add another entry to test right field width for
# integer
self.founders.append(('Old', 'Man', 100500))
self.assertEquals(
u"""
first_name|last_name |gpa
----------|----------|------
John |Adams |90
George |Washington|67
Thomas |Jefferson |50
Old |Man |100500
""".strip(),
unicode(self.founders)
)
if __name__ == '__main__':