mirror of
https://github.com/microsoft/ai-edu.git
synced 2026-08-29 02:27:20 +08:00
Nathan Bu
004bc74d6c
* Keep updating Chapter 16 of ASE * update * update * update script and add README * update * update * update * update * update * update * update * add .DS_Store --------- Co-authored-by: shbu <shaopeng@microsoft.com>
80 lines
4.4 KiB
Python
80 lines
4.4 KiB
Python
import os
|
|
import shutil
|
|
import re
|
|
import subprocess
|
|
|
|
# Define a function named merge_markdown_files that takes a folder path and a file name as parameters and calls copy_img_folders and merge_markdown_content functions
|
|
def merge_markdown_files(folder_path, file_name,output_folder="output"):
|
|
current_root_dir = os.path.dirname(os.getcwd())
|
|
# Use the os.walk() function to iterate over all subdirectories in the folder and store the result in a variable
|
|
walk_result = list(os.walk(folder_path))
|
|
if os.path.exists(output_folder):
|
|
shutil.rmtree(output_folder)
|
|
# Loop through each tuple in the list
|
|
for dirpath, dirnames, filenames in walk_result:
|
|
# Loop through each directory name in the current subdirectory
|
|
for dir in dirnames:
|
|
# Check if the directory name is "img"
|
|
if dir == "img":
|
|
# Get the source path of the img folder by joining it with the directory path
|
|
src_path = os.path.join(dirpath, dir)
|
|
relative_path = os.path.relpath( os.path.abspath(dirpath),current_root_dir)
|
|
# print(f'relative_path {relative_path}')
|
|
# Get the destination path of the img folder by joining it with the file name and removing the extension
|
|
dst_path = os.path.join(output_folder,relative_path, dir).replace(" ", "")
|
|
# Copy the img folder to the destination path using shutil.copytree()
|
|
shutil.copytree(src_path, dst_path)
|
|
# Use the os.walk() function to iterate over all subdirectories in the folder and store the result in a variable
|
|
# Sort the list of tuples by the first element (the directory path)
|
|
walk_result.sort(key=lambda x: x[0])
|
|
|
|
# Define the regular expression pattern to match <img> tags
|
|
image_tag_pattern = r'<img src="(.*?)"/>'
|
|
# Define the replacement pattern for Markdown image syntax
|
|
replacement = r''
|
|
# Open a new markdown file in write mode using the with syntax
|
|
with open(os.path.join(output_folder, file_name), "w", encoding="utf-8") as output_file:
|
|
# Loop through each tuple in the sorted list
|
|
for dirpath, dirnames, filenames in walk_result:
|
|
# Sort the list of directory names in alphabetical order
|
|
dirnames.sort()
|
|
# Sort the list of file names in alphabetical order
|
|
filenames.sort()
|
|
has_section_name = False
|
|
# Loop through each file name in the current subdirectory
|
|
for file in filenames:
|
|
# Check if the file name ends with ".md"
|
|
if file.endswith(".md"):
|
|
if not has_section_name:
|
|
# Get the folder relative path by using the os.path.relpath() function
|
|
section_name = os.path.relpath(dirpath, folder_path)
|
|
if section_name == ".": section_name = "目录"
|
|
# Write the section name as a header, followed by a newline character
|
|
output_file.write("# " + section_name.replace("\\","\n\n# ") + "\n\n")
|
|
has_section_name = True
|
|
# Open the markdown file in read mode using the with syntax
|
|
with open(os.path.join(dirpath, file), "r", encoding="utf-8") as file1:
|
|
# Read the content of the markdown file
|
|
content = file1.read()
|
|
chapter_title = file.replace(".md","")
|
|
if chapter_title == "README":
|
|
chapter_title = "前言"
|
|
chapter_title = "## " + chapter_title
|
|
if chapter_title not in content:
|
|
# Write the file name as the title of the content, followed by a newline character
|
|
output_file.write(chapter_title + "\n\n")
|
|
relative_path = os.path.relpath(os.path.abspath(dirpath),current_root_dir).replace(" ", "")
|
|
# print(f'relative_path {relative_path}')
|
|
# Replace any image path text in the content with the new location using re.sub()
|
|
content = re.sub(r"img\/(\w+\.(JPG|SVG|jpg|svg|png))", f"{repr(relative_path)[1:-1]}/img/\g<1>", content)
|
|
converted_string = re.sub(image_tag_pattern, replacement, content)
|
|
# Write the converted_string to the new file
|
|
output_file.write(converted_string+ "\n")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Call the function with the folder path and the file name as arguments
|
|
book_name = "A5-现代软件工程（更新中）"
|
|
merge_markdown_files(f"../基础教程/{book_name}", f"{book_name}.md")
|
|
pandoc_cmd_list = ["pandoc.exe", f"output/{book_name}.md", '-o', f"output/{book_name}.docx", '--reference-doc', 'customref.docx', '--resource-path', 'output']
|
|
subprocess.run(pandoc_cmd_list, stdout=subprocess.PIPE) |