40 lines
990 B
Python
40 lines
990 B
Python
|
from rolltable import tables
|
||
|
|
import typer
|
||
|
|
from rich import print
|
||
|
|
from rich.table import Table
|
||
|
|
|
||
|
|
|
||
|
|
app = typer.Typer()
|
||
|
|
|
||
|
|
|
||
|
|
@app.command("roll-table")
|
||
|
|
def create(
|
||
|
|
source: str = typer.Argument(
|
||
|
|
...,
|
||
|
|
help="Path to the yaml-formatted source file."),
|
||
|
|
frequency: str = typer.Option(
|
||
|
|
'default',
|
||
|
|
help='use the specified frequency from the source file'),
|
||
|
|
die: int = typer.Option(
|
||
|
|
20,
|
||
|
|
help='The size of the die for which to create a table'),
|
||
|
|
collapse: bool = typer.Option(
|
||
|
|
True,
|
||
|
|
help='If True, collapse multiple die values with the same option.')
|
||
|
|
):
|
||
|
|
"""
|
||
|
|
CLI for creating roll tables.
|
||
|
|
"""
|
||
|
|
with open(source, 'r') as src:
|
||
|
|
rt = tables.RollTable(source=src, frequency=frequency, die=die,
|
||
|
|
collapsed=collapse)
|
||
|
|
rt.load_source()
|
||
|
|
table = Table(*rt.rows[0])
|
||
|
|
for row in rt.rows[1:]:
|
||
|
|
table.add_row(*row)
|
||
|
|
print(table)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
app()
|