42 Zeilen
1.6 KiB
Python
42 Zeilen
1.6 KiB
Python
import os
|
|
import re
|
|
import codecs
|
|
import cmarkgfm # pip3 install cmarkgfm
|
|
from .builder import Builder
|
|
|
|
|
|
class HTMLBuilder(Builder):
|
|
"""Traintastic HTML manual builder base class"""
|
|
|
|
def _file_to_html(self, page):
|
|
with codecs.open(os.path.join(self._language_dir, page['markdown']), 'r', 'utf-8') as md:
|
|
html = cmarkgfm.github_flavored_markdown_to_html(md.read())
|
|
|
|
# parse id
|
|
html = re.sub(r'<h([1-6])([^>]*)>(.*) {#([a-z0-9-]+)}</h\1>', r'<h\1\2 id="\4">\3</h\1>', html)
|
|
|
|
# replace -> by
|
|
html = re.sub(r'->', '\u2794', html)
|
|
|
|
# set target="_blank" for external links:
|
|
html = re.sub(r'<a([^>]+href="http(s|)://)', r'<a target="_blank"\1', html)
|
|
|
|
if 'code' in page and page['code'] == 'lua':
|
|
html = re.sub(r'<code>(.+)</code>', self._highlight_lua, html)
|
|
|
|
# change img title attribute to figcaption
|
|
html = re.sub(r'(<img[^>]+)title="([^">]*)"([^>]*>)',
|
|
lambda m:
|
|
'<figure>' +
|
|
m.group(1) + m.group(3) +
|
|
'<figcaption>' + m.group(2) + '</figcaption></figure>',
|
|
html)
|
|
|
|
return html
|
|
|
|
def _highlight_lua(self, m):
|
|
code = m.group(1)
|
|
code = re.sub(r'\b([A-Z_][A-Z0-9_]*)\b', r'<span class="const">\1</span>', code) # CONSTANTS
|
|
code = re.sub(r'\b(and|break|do|else|elseif|end|false|for|function|goto|if|in|local|nil|not|or|repeat|return|then|true|until|while)\b', r'<span class="keyword">\1</span>', code) # keywords
|
|
return '<code class="lua">' + code + '</code>'
|