90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
|
import re, jinja2, mistune, shutil, os, pathlib, tarfile
|
||
|
|
from pygments.lexers import get_lexer_by_name
|
||
|
|
from pygments.formatters import HtmlFormatter
|
||
|
|
from pygments import highlight
|
||
|
|
|
||
|
|
|
||
|
|
class PygmentsHTMLRenderer(mistune.HTMLRenderer):
|
||
|
|
def block_code(self, code: str, info = None):
|
||
|
|
if not info:
|
||
|
|
return '\n<pre><code>%s</code></pre>\n' % mistune.escape(code)
|
||
|
|
lexer = get_lexer_by_name(info, stripall=True)
|
||
|
|
formatter = HtmlFormatter(lineseparator='<br>')
|
||
|
|
return highlight(code, lexer, formatter)
|
||
|
|
|
||
|
|
|
||
|
|
def convert_markdown(page: str):
|
||
|
|
plugins = ['footnotes', 'table', 'strikethrough', 'url']
|
||
|
|
renderer = PygmentsHTMLRenderer(escape=False)
|
||
|
|
return mistune.create_markdown(plugins=plugins, renderer=renderer)(page)
|
||
|
|
|
||
|
|
|
||
|
|
def extract_title(page: str):
|
||
|
|
matches = re.match('<h1>(.*?)</h1>', page)
|
||
|
|
if matches:
|
||
|
|
return matches.group(1)
|
||
|
|
return 'BlankHex'
|
||
|
|
|
||
|
|
|
||
|
|
def handle_file(path: str, input_dir: str, output_dir: str, template_name: str):
|
||
|
|
# Calculate input and output paths
|
||
|
|
relpath = os.path.relpath(path, input_dir)
|
||
|
|
input_path = path
|
||
|
|
output_path = os.path.join(output_dir, relpath)
|
||
|
|
if input_path.endswith('.md'):
|
||
|
|
output_path = output_path.replace('.md', '.html')
|
||
|
|
|
||
|
|
# Don't convert if output path exists
|
||
|
|
if os.path.exists(output_path):
|
||
|
|
return
|
||
|
|
|
||
|
|
# Run conversion
|
||
|
|
pathlib.Path(os.path.dirname(output_path)).mkdir(parents=True, exist_ok=True)
|
||
|
|
if input_path.endswith('.md'):
|
||
|
|
# Read Markdown document
|
||
|
|
with open(input_path, 'r') as handle:
|
||
|
|
markdown_page = handle.read()
|
||
|
|
|
||
|
|
# Get Pygments styles for light and dark themes
|
||
|
|
light_style = HtmlFormatter(style='default').get_style_defs()
|
||
|
|
dark_style = HtmlFormatter(style='monokai').get_style_defs()
|
||
|
|
|
||
|
|
# Convert Markdown document to HTML document
|
||
|
|
html_page = convert_markdown(markdown_page)
|
||
|
|
html_header = extract_title(html_page)
|
||
|
|
environment = jinja2.Environment(loader=jinja2.FileSystemLoader('template/'))
|
||
|
|
template = environment.get_template(template_name)
|
||
|
|
output_page = template.render(title=html_header,
|
||
|
|
body=html_page,
|
||
|
|
light_style=light_style,
|
||
|
|
dark_style=dark_style)
|
||
|
|
|
||
|
|
# Write HTML document
|
||
|
|
with open(output_path, 'w') as handle:
|
||
|
|
handle.write(output_page)
|
||
|
|
else:
|
||
|
|
# Copy file as is
|
||
|
|
shutil.copy(path, output_path)
|
||
|
|
|
||
|
|
|
||
|
|
def convert_dir(input_dir: str, output_dir: str, template_name: str):
|
||
|
|
# Convert or copy every file from the input directory to the output directory
|
||
|
|
for subdir, dirs, files in os.walk(input_dir):
|
||
|
|
for file in files:
|
||
|
|
handle_file(os.path.join(subdir, file), input_dir, output_dir, template_name)
|
||
|
|
|
||
|
|
|
||
|
|
# Remove output from previous run
|
||
|
|
if os.path.isdir('public'):
|
||
|
|
shutil.rmtree('public')
|
||
|
|
if os.path.isfile('public.tgz'):
|
||
|
|
os.remove('public.tgz')
|
||
|
|
|
||
|
|
# Run conversion for english, russian and 'common' folder
|
||
|
|
convert_dir('content/en', 'public/en', 'en.html')
|
||
|
|
convert_dir('content/ru', 'public/ru', 'ru.html')
|
||
|
|
convert_dir('content/common', 'public', 'none.html')
|
||
|
|
with tarfile.open('public.tgz', 'w:gz') as tar:
|
||
|
|
for file in os.listdir('public'):
|
||
|
|
tar.add(os.path.join('public', file), file)
|