| 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
 | 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):
        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", **{})]
 |