not_prettytable.py

not_prettytable.py

import tabulate


class NotPrettyTable(object):
    def __init__(self, headers, tablefmt='grid'):
        self._headers = tuple(headers)
        self._rows = []
        self._tablefmt = tablefmt

    def add_row(self, row):
        if not isinstance(row, tuple):
            row = tuple(row)
        if len(row) != len(self._headers):
            raise ValueError("New row to add must have %s length"
                             % len(self._headers))
        self._rows.append(row)

    def get_string(self):
        return str(self)

    def __str__(self):
        return tabulate.tabulate(self._rows, self._headers,
                                 tablefmt=self._tablefmt)