49 lines
1.3 KiB
Python
Executable File
49 lines
1.3 KiB
Python
Executable File
from typing import List
|
|
|
|
HTML_REPLACEMENTS = [
|
|
(r'&', r'&'),
|
|
(r'"', r'"'),
|
|
]
|
|
|
|
|
|
def parse_markdown(text: str, tab_length: int = 2) -> str:
|
|
lines = text.split("\n")
|
|
output: List[str] = []
|
|
list_level = 0
|
|
|
|
def end_outstanding_lists(level: int, end_level: int) -> int:
|
|
while level > end_level:
|
|
level -= 1
|
|
output.append("</ul>")
|
|
if level > 0:
|
|
output.append("</li>")
|
|
return end_level
|
|
|
|
for i, line in enumerate(lines):
|
|
if i + 1 < len(lines) and lines[i + 1].startswith("==="): # heading
|
|
output.append(f"<h1>{line}</h1>")
|
|
elif line.startswith("==="):
|
|
pass
|
|
elif line.lstrip().startswith("* "): # list
|
|
line_level = 1 + line.count(" " * tab_length, 0, line.index("*"))
|
|
if list_level >= line_level:
|
|
list_level = end_outstanding_lists(list_level, line_level)
|
|
else:
|
|
list_level += 1
|
|
if list_level > 1:
|
|
output[-1] = output[-1].replace("</li>", "")
|
|
output.append("<ul>")
|
|
output.append(f"<li>{line.replace('*', '', 1).lstrip()}</li>")
|
|
else:
|
|
list_level = end_outstanding_lists(list_level, 0)
|
|
if len(line) > 0:
|
|
output.append(line)
|
|
|
|
end_outstanding_lists(list_level, 0)
|
|
output_str = "\n".join(output) + "\n"
|
|
|
|
for (fr, to) in HTML_REPLACEMENTS:
|
|
output_str = output_str.replace(fr, to)
|
|
|
|
return output_str
|