mirror of
https://github.com/krahets/hello-algo.git
synced 2024-12-27 03:06:29 +08:00
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
|
"""
|
||
|
|
File: build_markdown_docs.py
|
||
|
|
Created Time: 2023-02-06
|
||
|
|
Author: Krahets (krahets@163.com)
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sys, os.path as osp
|
||
|
|
sys.path.append(osp.dirname(osp.dirname(osp.dirname(osp.abspath(__file__)))))
|
||
|
|
|
||
|
|
import re
|
||
|
|
import glob
|
||
|
|
import shutil
|
||
|
|
from docs.utils.extract_code_python import ExtractCodeBlocksPython
|
||
|
|
|
||
|
|
def build_markdown(md_path):
|
||
|
|
with open(md_path, "r") as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
|
||
|
|
code_blocks_dict = {}
|
||
|
|
file_pattern = re.compile(r'\s*```(\w+)\s+title="(.+)"')
|
||
|
|
src_pattern = re.compile(r'\s*\[class\]\{(.*?)\}-\[func\]\{(.*?)\}')
|
||
|
|
for i in range(len(lines)):
|
||
|
|
# Find the line target to the source codes
|
||
|
|
src_match = src_pattern.match(lines[i])
|
||
|
|
if src_match is None:
|
||
|
|
continue
|
||
|
|
for j in range(i - 1, -1 ,-1):
|
||
|
|
file_match = file_pattern.match(lines[j])
|
||
|
|
if file_match is not None:
|
||
|
|
break
|
||
|
|
|
||
|
|
# Get code blocks
|
||
|
|
lang = file_match[1]
|
||
|
|
file_name = file_match[2]
|
||
|
|
if file_name not in code_blocks_dict:
|
||
|
|
code_blocks_dict[file_name] = ExtractCodeBlocksPython(
|
||
|
|
file_path=osp.dirname(md_path).replace("docs/", f"codes/{lang}/") + f"/{file_name}")
|
||
|
|
|
||
|
|
header_line = i
|
||
|
|
class_label = src_match[1]
|
||
|
|
func_label = src_match[2]
|
||
|
|
code_blocks = code_blocks_dict[file_name]
|
||
|
|
src_info = {
|
||
|
|
"line_number": i,
|
||
|
|
"class_label": src_match[1],
|
||
|
|
"func_label": src_match[2],
|
||
|
|
"code_blocks": code_blocks_dict[file_name]
|
||
|
|
}
|
||
|
|
|
||
|
|
# Add the class to the doc
|
||
|
|
if not func_label and class_label:
|
||
|
|
if class_label in code_blocks.classes:
|
||
|
|
lines.pop(header_line)
|
||
|
|
class_block = code_blocks.classes[class_label]["block"]
|
||
|
|
for code_line in class_block[::-1]:
|
||
|
|
ind = " " * 4 if code_line != "\n" else ""
|
||
|
|
lines.insert(header_line, ind + code_line)
|
||
|
|
# Add the function to the doc
|
||
|
|
elif func_label and not class_label:
|
||
|
|
if func_label in code_blocks.functions:
|
||
|
|
lines.pop(header_line)
|
||
|
|
func_block = code_blocks.functions[func_label]
|
||
|
|
for code_line in func_block["block"][::-1]:
|
||
|
|
ind = " " * 4 if code_line != "\n" else ""
|
||
|
|
lines.insert(header_line, ind + code_line)
|
||
|
|
# Add the class method to the doc
|
||
|
|
elif func_label and class_label:
|
||
|
|
if class_label in code_blocks.classes:
|
||
|
|
class_dict = code_blocks.classes[class_label]
|
||
|
|
if func_label in class_dict["functions"]:
|
||
|
|
lines.pop(header_line)
|
||
|
|
func_block = class_dict["functions"][func_label]
|
||
|
|
for code_line in func_block["block"][::-1]:
|
||
|
|
lines.insert(header_line, code_line)
|
||
|
|
|
||
|
|
with open(md_path.replace("docs/", "build/"), "w") as f:
|
||
|
|
f.writelines(lines)
|
||
|
|
print(f"Built {md_path}")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# Copy files to the build dir
|
||
|
|
shutil.copytree("docs", "build", dirs_exist_ok=True)
|
||
|
|
shutil.rmtree("build/utils")
|
||
|
|
# Build docs
|
||
|
|
for md_path in glob.glob("docs/chapter_*/*.md"):
|
||
|
|
build_markdown(md_path)
|