summaryrefslogtreecommitdiff
path: root/markdown_renderer.py
blob: d64fcf21bf927f0768ddb39a5c2301657f37872f (plain)
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
import markdown
import frontmatter
import markdown.treeprocessors
import markdown.extensions


class DefaultClassTreeprocessor(markdown.treeprocessors.Treeprocessor):
    def run(self, root):
        for element in root.iter("a"):
            print(element.attrib)
            if "class" not in element.attrib or "button" not in element.attrib["class"]:
                element.set("class", "link")
            if "href" in element.attrib and element.attrib["href"] == "":
                del element.attrib["href"]
        for element in root.iter("img"):
            element.set("class", "img")


class DefaultClassExtension(markdown.extensions.Extension):
    def extendMarkdown(self, md):
        md.treeprocessors.register(DefaultClassTreeprocessor(md), "link_class", 2)


def md_to_html(raw_content: str) -> str:
    metadata, content = frontmatter.parse(raw_content)
    html_text = markdown.markdown(
        content,
        output_format="html",
        extensions=["attr_list", "fenced_code", "tables", DefaultClassExtension(), "md_in_html"]
    )
    html_text = html_text.replace("<p>+-+-+-</p>", '<section><div class="center-grid"><div class="div-sizing">')
    html_text = html_text.replace("<p>+-+-+-", '<section><div class="center-grid"><div class="div-sizing">')
    html_text = html_text.replace("+-+-+-</p>", '<section><div class="center-grid"><div class="div-sizing">')
    html_text = html_text.replace("-+-+-+</p>", "</div></div></section>")
    html_text = html_text.replace("<p>-+-+-+", "</div></div></section>")
    html_text = html_text.replace("<p>+=+=+=</p>", '<section class="normal-section"><div class="center-grid"><div class="div-sizing">')
    html_text = html_text.replace("<p>+=+=+=", '<section class="normal-section"><div class="center-grid"><div class="div-sizing">')
    html_text = html_text.replace("+=+=+=</p>", '<section class="normal-section"><div class="center-grid"><div class="div-sizing">')
    html_text = html_text.replace("<p>{{{</p>", '<div class="flex-container gap">')
    html_text = html_text.replace("<p>{{{", '<div class="flex-container gap">')
    html_text = html_text.replace("{{{</p>", '<div class="flex-container gap">')
    html_text = html_text.replace("<p>}}}</p>", '</div>')
    html_text = html_text.replace("}}}</p>", '</div>')
    html_text = html_text.replace("<p>}}}", '</div>')

    return (metadata, html_text)