-
-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathmarkdown_code_blocks.py
46 lines (35 loc) · 1.32 KB
/
markdown_code_blocks.py
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
from __future__ import annotations
import argparse
from collections.abc import Sequence
import mistune
import pygments.formatters
import pygments.lexers
import pygments.util
class CodeRenderer(mistune.HTMLRenderer):
def block_code(self, code: str, info: str | None = None) -> str:
try:
lexer = pygments.lexers.get_lexer_by_name(info, stripnl=False)
cssclass = f'highlight {info}'
except pygments.util.ClassNotFound:
lexer = pygments.lexers.get_lexer_by_name('text', stripnl=False)
cssclass = 'highlight'
formatter = pygments.formatters.HtmlFormatter(cssclass=cssclass)
return pygments.highlight(code, lexer=lexer, formatter=formatter)
def highlight(
doc: str,
Renderer: type[mistune.HTMLRenderer] = CodeRenderer,
) -> str:
return mistune.create_markdown(
renderer=Renderer(),
plugins=['strikethrough', 'footnotes', 'table'],
)(doc)
def main(argv: Sequence[str] | None = None) -> int:
parser = argparse.ArgumentParser()
parser.add_argument('filename', default='/dev/stdin')
args = parser.parse_args(argv)
with open(args.filename) as f:
hl = highlight(f.read())
print(f'<!doctype html><html><body>{hl}</body></html>')
return 0
if __name__ == '__main__':
raise SystemExit(main())