WeChatReport_OS/replace.py
PeterZhong a4cca18a42 2024
2025-01-01 01:23:29 +08:00

25 lines
1.1 KiB
Python

import os
from bs4 import BeautifulSoup
if __name__ == '__main__':
file_list = os.listdir("./mdbook_test/book")
for item in file_list:
full_path = os.path.join("./mdbook_test/book", item)
if os.path.isfile(full_path) and item.endswith(".html"):
with open(full_path, encoding="utf-8", mode="r") as file:
content = file.read()
soup = BeautifulSoup(content)
if not soup.find(id="sidebar") == None:
soup.find(id="sidebar").decompose()
while soup.find("nav") is not None:
soup.find("nav").decompose()
while soup.find(class_="fa-search") is not None:
soup.find(class_="fa-search").decompose()
if not soup.find(id="search-wrapper") == None:
soup.find(id="search-wrapper").decompose()
with open(full_path, encoding="utf-8", mode="w+") as file:
origin_str = soup.prettify()
correct_str = origin_str.replace("http://localhost:8888/nbextensions/assets", ".")
content = file.write(correct_str)
print("已完成"+item)