| Current File : //lib/python2.7/site-packages/cloudinit/simpletable.py |
# Copyright (C) 2017 Amazon.com, Inc. or its affiliates
#
# Author: Ethan Faust <[email protected]>
# Author: Andrew Jorgensen <[email protected]>
#
# This file is part of cloud-init. See LICENSE file for license information.
class SimpleTable(object):
"""A minimal implementation of PrettyTable
for distribution with cloud-init.
"""
def __init__(self, fields):
self.fields = fields
self.rows = []
# initialize list of 0s the same length
# as the number of fields
self.column_widths = [0] * len(self.fields)
self.update_column_widths(fields)
def update_column_widths(self, values):
for i, value in enumerate(values):
self.column_widths[i] = max(
len(value),
self.column_widths[i])
def add_row(self, values):
if len(values) > len(self.fields):
raise TypeError('too many values')
values = [str(value) for value in values]
self.rows.append(values)
self.update_column_widths(values)
def _hdiv(self):
"""Returns a horizontal divider for the table."""
return '+' + '+'.join(
['-' * (w + 2) for w in self.column_widths]) + '+'
def _row(self, row):
"""Returns a formatted row."""
return '|' + '|'.join(
[col.center(self.column_widths[i] + 2)
for i, col in enumerate(row)]) + '|'
def __str__(self):
"""Returns a string representation of the table with lines around.
+-----+-----+
| one | two |
+-----+-----+
| 1 | 2 |
| 01 | 10 |
+-----+-----+
"""
lines = [self._hdiv(), self._row(self.fields), self._hdiv()]
lines += [self._row(r) for r in self.rows] + [self._hdiv()]
return '\n'.join(lines)
def get_string(self):
return self.__str__()