feat: format contest file

yanglbme · yanglbme · commit 83aedbc07068 · 2024-03-11T10:20:57.000+08:00
diff --git a/main.py b/main.py
@@ -3,19 +3,10 @@
 from collections import defaultdict
 
 
-def format_contest_md(content: str) -> str:
-    content = content.replace("[English Version](/solution/CONTEST_README_EN.md)", "")
-    content = content.replace("[中文文档](/solution/CONTEST_README.md)", "")
-    res = re.findall(r"\[(.*?)\]\((.*?)\)", content)
-    for _, link in res:
-        num = link.split("/")[-2].split(".")[0]
-        num = int(num)
-        content = content.replace(link, f"./lc/{num}.md")
-    content = f"---\ncomments: true\n---\n\n" + content
-    return content
+for contest_file in ["docs/contest.md", "docs-en/contest.md"]:
+    with open(contest_file, "r", encoding="utf-8") as f:
+        content = f.read()
 
-
-def format_contest_md_en(content: str) -> str:
     content = content.replace("[English Version](/solution/CONTEST_README_EN.md)", "")
     content = content.replace("[中文文档](/solution/CONTEST_README.md)", "")
     res = re.findall(r"\[(.*?)\]\((.*?)\)", content)
@@ -24,20 +15,10 @@ def format_contest_md_en(content: str) -> str:
         num = int(num)
         content = content.replace(link, f"./lc/{num}.md")
     content = f"---\ncomments: true\n---\n\n" + content
-    return content
-
 
-with open("docs/contest.md", "r", encoding="utf-8") as f:
-    contest = f.read()
-    contest = format_contest_md(contest)
-with open("docs/contest.md", "w", encoding="utf-8") as f:
-    f.write(contest)
+    with open(contest_file, "w", encoding="utf-8") as f:
+        f.write(content)
 
-with open("docs-en/contest.md", "r", encoding="utf-8") as f:
-    contest_en = f.read()
-    contest_en = format_contest_md_en(contest_en)
-with open("docs-en/contest.md", "w", encoding="utf-8") as f:
-    f.write(contest_en)
 
 code_dict = {
     "py": ("Python3", "python"),
