ctabus/print2d.py

import datetime
from pydoc import pager


def str_coerce(s, **kwargs):
    if isinstance(s, datetime.datetime):
        return s.strftime(kwargs['datetime_format'])
    else:
        return str(s)


def print2d(list_param,
            datetime_format="%A, %B %e, %Y %H:%M:%S",
            seperator=' | ',
            spacer=' ',
            bottom='=',
            l_end='|', r_end='|',
            interactive=False
            ):
    list_param = [[str_coerce(s, datetime_format=datetime_format)
                   for s in row] for row in list_param]

    max_col = []
    for row in list_param:
        for i, col in enumerate(row):
            try:
                max_col[i] = max(max_col[i], len(col))
            except IndexError:
                max_col.append(len(col))

    fmt_row = '{content}'
    if l_end:
        fmt_row = '{} {}'.format(l_end, fmt_row)
    if r_end:
        fmt_row = '{} {}'.format(fmt_row, r_end)

    done = []
    for row in list_param:
        content = seperator.join(col.ljust(max_col[i], spacer if i < len(
            row)-1 or r_end else ' ') for i, col in enumerate(row))
        done.append(fmt_row.format(content=content))

    if bottom:
        bottom = bottom*len(done[0])
        row_sep = ('\n'+bottom+'\n')
    else:
        row_sep = '\n'
    final = row_sep.join(done)
    if bottom:
        final = '\n'.join((bottom, final, bottom))
    if interactive:
        if not bottom:
            final += '\n'
        pager(final)
    else:
        return final