66 lines
1.9 KiB
Python
Executable File
66 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
import os
|
||
import re
|
||
import sys
|
||
|
||
|
||
def add_space(origin: str):
|
||
modified = "{} {}".format(origin[0], origin[1])
|
||
print(repr(origin), "->", modified)
|
||
return modified
|
||
|
||
|
||
def change_dot(origin: str):
|
||
modified = "{}．".format(origin)
|
||
print(repr(origin), "->", modified)
|
||
return modified
|
||
|
||
|
||
def remove_space(origin: str):
|
||
modified = origin.replace(" ", "")
|
||
print(repr(origin), "->", modified)
|
||
return modified
|
||
|
||
|
||
def handle(full_path: str):
|
||
print(full_path)
|
||
with open(full_path, "r", encoding="utf-8") as fp:
|
||
content = fp.read()
|
||
content = re.sub(
|
||
"[0-9a-zA-Z][|\u4e00-\u9fa5]", lambda x: add_space(x.group(0)), content
|
||
)
|
||
content = re.sub(
|
||
"[|\u4e00-\u9fa5][0-9a-zA-Z]", lambda x: add_space(x.group(0)), content
|
||
)
|
||
content = re.sub("[0-9a-zA-Z%][。]", lambda x: change_dot(x.group(0)), content)
|
||
content = re.sub("[，。；：？！”）] ", lambda x: remove_space(x.group(0)), content)
|
||
content = re.sub(" [，。；：？！“（]", lambda x: remove_space(x.group(0)), content)
|
||
content = re.sub("^[ ]+$", "", content)
|
||
content = re.sub(" MtF ", " MtF ", content, flags=re.IGNORECASE)
|
||
content = re.sub(" LGBT ", " LGBT ", content, flags=re.IGNORECASE)
|
||
content = re.sub(" QQ ", " QQ ", content, flags=re.IGNORECASE)
|
||
content = re.sub("\n\n\n", "\n\n", content)
|
||
with open(full_path, "w", encoding="utf-8") as fp:
|
||
fp.write(content)
|
||
|
||
|
||
def walk_all_files(base: str):
|
||
print(base)
|
||
for root, dirs, files in os.walk(base):
|
||
for file in files:
|
||
full_path = os.path.join(root, file)
|
||
full_path: str
|
||
if full_path.endswith(".md"):
|
||
handle(full_path)
|
||
|
||
|
||
def main():
|
||
if len(sys.argv) < 2:
|
||
walk_all_files(os.getcwd())
|
||
else:
|
||
walk_all_files(sys.argv[1])
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|