Viewing file:  simpletable.py (1.93 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# 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:
"""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__()
|