1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
import pprint
from docutils import nodes
from docutils.parsers.rst import Directive
from docutils.parsers.rst.directives import unchanged, unchanged_required
from flask import url_for
from jinja2 import escape
BUTTON_TYPES = ("default", "primary", "secondary", "danger", "darkish", "darker")
ICON_WEIGHT_TABLE = {
"light": "fal",
"regular": "far",
"solid": "fas",
"branding": "fab"
}
ICON_WEIGHTS = tuple(ICON_WEIGHT_TABLE.values())
class ButtonDirective(Directive):
has_content = True
option_spec = {
"icon": unchanged_required,
"text": unchanged_required,
"type": unchanged,
"url": unchanged,
}
def run(self):
pprint.pprint(self.__dict__) # DEBUG
icon = self.options.get("icon", "")
button_type = self.options.get("type", "primary")
text = self.options["text"]
url = self.options["url"]
if icon:
parts = [escape(x) for x in icon.split("/")]
if len(parts) != 2:
raise self.error("Icon specification must be in the form <type>/<name>")
elif parts:
weight = parts[0]
if weight not in ICON_WEIGHTS:
weight = ICON_WEIGHT_TABLE.get(weight)
if not weight:
raise self.error(
"Icon type must be one of light, regular, solid or "
"branding, or a font-awesome weight class"
)
icon_html = f"""<i class="uk-icon fa-fw {weight} fa-{parts[1]}"></i>"""
else:
icon_html = ""
if button_type not in BUTTON_TYPES:
self.error(f"Button type must be one of {', '.join(BUTTON_TYPES[:-1])} or {[-1]}")
if url.startswith("flask://"):
url = url_for(url.split("://", 1)[1])
elif url.startswith("wiki://"):
url = url_for("wiki.page", page=url.split("://", 1)[1])
html = f"""<a class="uk-button uk-button-{button_type}" href=\"{url}\">{icon_html} {text}</a>"""
return [nodes.raw(html, html, format="html", **{})]
|
