dnd-item-generator/dnd_item/cli.py

import logging
import os

from pathlib import Path

import typer

from rich.logging import RichHandler
from rich.console import Console

from dnd_item.types import WeaponGenerator, MagicWeaponGenerator
from dnd_item import five_e

app = typer.Typer()
app_state = {}


@app.callback()
def main(
    cr: int = typer.Option(default=None, help='The Challenge Rating to use when determining rarity.'),
):
    debug = os.getenv("FANITEM_DEBUG", None)
    logging.basicConfig(
        format="%(name)s %(message)s",
        level=logging.DEBUG if debug else logging.INFO,
        handlers=[RichHandler(rich_tracebacks=True, tracebacks_suppress=[typer])],
    )
    logging.getLogger('markdown_it').setLevel(logging.ERROR)

    app_state['cr'] = cr or 0
    app_state['data'] = Path(__file__).parent / Path("sources")


@app.command()
def weapon(count: int = typer.Option(1, help="The number of weapons to generate.")):
    console = Console()
    for weapon in WeaponGenerator().random(count=count, challenge_rating=app_state['cr']):
        console.print(weapon.details)


@app.command()
def magic_weapon(count: int = typer.Option(1, help="The number of weapons to generate.")):
    console = Console()
    for weapon in MagicWeaponGenerator().random(count=count, challenge_rating=app_state['cr']):
        console.print(weapon.details)


@app.command()
def convert():
    src = five_e.weapons()
    print(src.as_yaml)