diff options
Diffstat (limited to '')
| -rw-r--r-- | pysite/rst/roles.py | 37 | 
1 files changed, 37 insertions, 0 deletions
| diff --git a/pysite/rst/roles.py b/pysite/rst/roles.py index bff00533..09c4d373 100644 --- a/pysite/rst/roles.py +++ b/pysite/rst/roles.py @@ -2,6 +2,7 @@  from docutils import nodes  from docutils.parsers.rst.roles import set_classes  from docutils.parsers.rst.states import Inliner +from flask import url_for  from jinja2 import escape @@ -44,3 +45,39 @@ def icon_role(_role: str, rawtext: str, text: str, lineno: int, inliner: Inliner          node = nodes.raw(html, html, format="html", **options)          return [node], [] + + +def url_for_role(_role: str, rawtext: str, text: str, lineno: int, inliner: Inliner, +              options: dict = None, _content: dict = None): +    if options is None: +        options = {} + +    set_classes(options) + +    if "/" in text: +        parts = [escape(x) for x in text.split("/")] +    else: +        msg = inliner.reporter.error("URL specification must be in the form <page.name>/<text>", line=lineno) +        prb = inliner.problematic(text, rawtext, msg) + +        return [prb], [msg] + +    if len(parts) != 2: +        msg = inliner.reporter.error("URL specification must be in the form <page.name>/<text>", line=lineno) +        prb = inliner.problematic(text, rawtext, msg) + +        return [prb], [msg] +    else: +        try: +            url = url_for(parts[0]) +            name = parts[1] + +            html = f"""<a href="{url}">{name}</a>""" + +            node = nodes.raw(html, html, format="html", **options) +            return [node], [] +        except Exception as e: +            msg = inliner.reporter.error(str(e), line=lineno) +            prb = inliner.problematic(text, rawtext, msg) + +            return [prb], [msg] | 
