ggx-docs/pdf_handler.py at main · corridor/ggx-docs · GitHub

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
"""
Based on mkdocs-with-pdf, create a theme handler that can add the appropriate CSS and
Add the PDF to our documentation correctly.
Ref: https://github.com/orzih/mkdocs-with-pdf/blob/v0.8.0/mkdocs_with_pdf/themes/generic.py
"""

from bs4 import BeautifulSoup
from mkdocs_with_pdf.themes import material


def get_stylesheet(debug_html: bool) -> str:  # noqa: FBT001 -- Needs to be as per mkdoc-with-pdf argspec
    return material.get_stylesheet(debug_html)


def get_script_sources() -> list:
    return material.get_script_sources()


def inject_link(html: str, href: str) -> str:
    """Adding PDF View button on navigation bar(using material theme)"""

    soup = BeautifulSoup(html, "html.parser")
    nav = soup.find("nav", class_="md-header__inner")
    if nav:
        a = soup.new_tag("a", href=href, title="Download as PDF", **{"class": "md-header-nav__button md-icon"})
        a.append(
            BeautifulSoup(
                """
                <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 512 512">
                <path d="M128,0c-17.6,0-32,14.4-32,32v448c0,17.6,14.4,32,32,32h320c17.6,0,32-14.4,32-32V128L352,0H128z" fill="#E2E5E7"/>
                <path d="m384 128h96l-128-128v96c0 17.6 14.4 32 32 32z" fill="#B0B7BD"/>
                <polygon points="480 224 384 128 480 128" fill="#CAD1D8"/>
                <path d="M416,416c0,8.8-7.2,16-16,16H48c-8.8,0-16-7.2-16-16V256c0-8.8,7.2-16,16-16h352c8.8,0,16,7.2,16,16  V416z" fill="#F15642"/>
                <g fill="#fff">
                <path d="m101.74 303.15c0-4.224 3.328-8.832 8.688-8.832h29.552c16.64 0 31.616 11.136 31.616 32.48 0 20.224-14.976 31.488-31.616 31.488h-21.36v16.896c0 5.632-3.584 8.816-8.192 8.816-4.224 0-8.688-3.184-8.688-8.816v-72.032zm16.88 7.28v31.872h21.36c8.576 0 15.36-7.568 15.36-15.504 0-8.944-6.784-16.368-15.36-16.368h-21.36z"/>
                <path d="m196.66 384c-4.224 0-8.832-2.304-8.832-7.92v-72.672c0-4.592 4.608-7.936 8.832-7.936h29.296c58.464 0 57.184 88.528 1.152 88.528h-30.448zm8.064-72.912v57.312h21.232c34.544 0 36.08-57.312 0-57.312h-21.232z"/>
                <path d="m303.87 312.11v20.336h32.624c4.608 0 9.216 4.608 9.216 9.072 0 4.224-4.608 7.68-9.216 7.68h-32.624v26.864c0 4.48-3.184 7.92-7.664 7.92-5.632 0-9.072-3.44-9.072-7.92v-72.672c0-4.592 3.456-7.936 9.072-7.936h44.912c5.632 0 8.96 3.344 8.96 7.936 0 4.096-3.328 8.704-8.96 8.704h-37.248v0.016z"/>
                </g>
                <path d="m400 432h-304v16h304c8.8 0 16-7.2 16-16v-16c0 8.8-7.2 16-16 16z" fill="#CAD1D8"/>
                </svg>
                """,  # noqa: E501
                "html.parser",
            )
        )
        nav.append(a)
        return str(soup)

    return html