fix(datasheet): markdown headings break the datasheet

close #23
This commit is contained in:
Uri Shaked
2023-07-30 22:21:00 +03:00
parent 6034820170
commit 7842384b6a
3 changed files with 22 additions and 0 deletions
+17
View File
@@ -0,0 +1,17 @@
import mistune
from mistune.renderers.markdown import MarkdownRenderer
class HeadingsRenderer(MarkdownRenderer):
def __init__(self, min_level: int):
super().__init__()
self.min_level = min_level
def heading(self, token, state):
token["attrs"]["level"] += self.min_level - 1
return super().heading(token, state)
def limit_markdown_headings(source: str, min_level: int) -> str:
markdown = mistune.create_markdown(renderer=HeadingsRenderer(min_level))
return markdown(source)