feat: add source

Author: yanglbme

docs-en/lc/1.md

@@ -4,6 +4,8 @@ tags:
   - Array
   - Hash Table
 edit_url: 'https://github.com/doocs/leetcode/edit/main/solution/0000-0099/0001.Two%20Sum/README_EN.md'
+source: Weekly Contest 1 Q1
+rating: 1234
 ---
 
 

docs/lc/1.md

@@ -6,6 +6,7 @@ tags:
     - 数组
     - 哈希表
 rating: 1234
+source: 第 1 场周赛 Q1
 ---
 
 # [1. 两数之和](https://leetcode.cn/problems/two-sum)

hooks/ext_info.py

@@ -12,20 +12,26 @@ def on_page_markdown(markdown, page, config, files):
     rating = page.meta.get("rating")
     if not difficulty and not rating:
         return markdown
+    source = page.meta.get("source")
     is_cn_problem_page = (page.edit_url or "").endswith("README.md")
     images = []
+    if source:
+        title = "来源" if is_cn_problem_page else "Source"
+        images.append(
+            f'<img src="https://img.shields.io/badge/{title}-{source}-4051B5?style=flat-square">'
+        )
     if difficulty:
         title = "难度" if is_cn_problem_page else "Difficulty"
         images.append(
             f'<img src="https://img.shields.io/badge/{title}-{difficulty}-4051B5?style=flat-square">'
         )
     if rating:
-        title = "难度分" if is_cn_problem_page else "Rating"
+        title = "分数" if is_cn_problem_page else "Rating"
         images.append(
             f'<img src="https://img.shields.io/badge/{title}-{rating}-4051B5?style=flat-square">'
         )
     # 将 images 放到 <p> 标签中
-    images_html = '<p>' + " ".join(images) + "</p>"
+    images_html = "<p>" + " ".join(images) + "</p>"
     heading_pattern = re.compile(r"(^# .+?$)", re.MULTILINE)
     match = heading_pattern.search(markdown)
     if match: