dnd/deadsands/site_tools/extensions.py

import textwrap
from pathlib import Path

from jinja2_simple_tags import StandaloneTag
from rolltable.tables import RollTable as _RT


class RollTable(StandaloneTag):
    tags = {"rolltable"}

    def render(self, sources, frequency="default", die=8, indent=0, **kwargs):
        rt = _RT([Path(f"sources/{s}.yaml").read_text() for s in sources], frequency=frequency, die=die)
        return textwrap.indent(rt.as_yaml(), " " * indent)