adding yaml output

This commit is contained in:
evilchili 2022-08-06 13:32:34 -07:00
parent 06791a0758
commit 25de2c8c3c
2 changed files with 18 additions and 0 deletions

View File

@ -163,6 +163,19 @@ class RollTable:
self._rows.append(self._column_filter([f'd{face+1}'] + row))
return self._rows
@property
def as_markdown(self) -> str:
return ''
@property
def as_yaml(self) -> dict:
struct = [{'headers': self.rows[0]}]
for row in self.rows[1:]:
struct.append({
row[0]: row[1:]
})
return yaml.dump(struct)
def _config(self):
"""
Parse data sources, generate headers, and create the column filters

View File

@ -122,3 +122,8 @@ def test_no_descriptions():
t = tables.RollTable([fixture_no_descriptions], die=1)
assert 'd1' in str(t)
assert 'option 1' in str(t)
def test_yaml():
t = tables.RollTable([fixture_metadata + fixture_source])
print(t.as_yaml)