Files
kubevela/hack/website/clean-md.py
2021-05-07 10:39:30 +08:00

38 lines
1.0 KiB
Python

import re
import os
import sys
def convert_link(md):
data = ""
with open(md, "r") as f:
data = f.read()
url_arr = re.findall(r'(\[(.*?)\]\((.*?)\))', data)
for url in url_arr:
if url[2].startswith("http") or url[2].startswith("https"):
continue
if ".md" in url[2] or ".mdx" in url[2]:
new_path = url[2].replace(".md", "")
new_path = new_path.replace(".mdx", "")
new_url = "[{}]({})".format(url[1], new_path)
data = data.replace(url[0], new_url)
print(f"convert {url[0]} to {new_url}")
with open(md, "w") as f:
f.write(data)
def main(path):
files = os.walk(path)
for path, dir_list,file_list in files:
for file_name in file_list:
file_path = os.path.join(path, file_name)
if file_path[-3:] == ".md" or file_path[-3:] == ".mdx" :
convert_link(file_path)
if __name__ == "__main__":
if len(sys.argv) != 2:
sys.exit(1)
main(sys.argv[1])