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
|
import argparse
import sys
from markdown_it.renderer import RendererHTML
from myst_parser.config.main import MdParserConfig
from myst_parser.parsers.mdit import create_md_parser
def print_anchors(args=None):
""" """
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument(
"input",
nargs="?",
type=argparse.FileType("r", encoding="utf8"),
default=sys.stdin,
help="Input file (default stdin)",
)
arg_parser.add_argument(
"-o",
"--output",
type=argparse.FileType("w", encoding="utf8"),
default=sys.stdout,
help="Output file (default stdout)",
)
arg_parser.add_argument(
"-l", "--level", type=int, default=2, help="Maximum heading level."
)
args = arg_parser.parse_args(args)
parser = create_md_parser(MdParserConfig(heading_anchors=args.level), RendererHTML)
def _filter_plugin(state):
state.tokens = [
t
for t in state.tokens
if t.type.startswith("heading_") and int(t.tag[1]) <= args.level
]
parser.use(lambda p: p.core.ruler.push("filter", _filter_plugin))
text = parser.render(args.input.read())
args.output.write(text)
|
