aboutsummaryrefslogtreecommitdiffstats
path: root/pysite/rst/roles.py
diff options
context:
space:
mode:
Diffstat (limited to 'pysite/rst/roles.py')
-rw-r--r--pysite/rst/roles.py37
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]