print_dump

print_dump.py

def pp(json_docs):
    """preety print for json docs"""
    print json.dumps(json_docs, indent=4)


def get_md5(key):
    """return md5 hex for key"""
    md5 = hashlib.md5()
    md5.update(key)
    return md5.hexdigest()


def print_table(lod, args):
    """
    Input: list of dictionary or list of list.
    Args: list of headers name which is the key in dictionary
    if preetyTable is not install then the output will be json format.
    """
    try:
        from prettytable import PrettyTable
        pt = PrettyTable(border=True, horizontal_char='-',
                         field_names=[x.title() for x in args])

        [pt.align.__setitem__(x.title(), "l") for x in args]

        if isinstance(lod[0], dict):
            [pt.add_row([x[item] for item in args]) for x in lod]
        else:
            [pt.add_row([x[i] for i in range(len(lod[0]))]) for x in lod]
        return pt
    except Exception:
        pp(lod)