dbt-selly/dbt-env/lib/python3.8/site-packages/agate/table/print_structure.py

28 lines
721 B
Python
Raw Normal View History

2022-03-22 15:13:27 +00:00
#!/usr/bin/env python
# pylint: disable=W0212
import sys
from agate.data_types import Text
def print_structure(self, output=sys.stdout, max_rows=None):
"""
Print this table's column names and types as a plain-text table.
:param output:
The output to print to.
"""
from agate.table import Table
name_column = [n for n in self._column_names]
type_column = [t.__class__.__name__ for t in self._column_types]
rows = zip(name_column, type_column)
column_names = ['column', 'data_type']
text = Text()
column_types = [text, text]
table = Table(rows, column_names, column_types)
return table.print_table(output=output, max_column_width=None, max_rows=max_rows)