Update README auto updates

Author: NinePiece2

.github/workflows/build-deploy-site.yml

@@ -6,7 +6,7 @@ on:
       - main
     paths:
       - "Solutions/**"
-      - "Scripts/**"
+      - "Scripts/update_readmes.py"
       - "mkdocs.yml"
       - "Style/**"
       - Site_README.md

.github/workflows/update-leetcode-grid.yml

@@ -4,6 +4,7 @@ on:
   push:
     paths:
       - "Solutions/**"
+      - "Scripts/update_project_readme.py"
   workflow_dispatch:
 
 jobs:

Scripts/update_project_readme.py

@@ -1,7 +1,7 @@
 import os
 import re
-from pathlib import Path
 import requests
+from pathlib import Path
 
 SRC_DIR = "Solutions"
 SITE_README_PATH = Path("Site_README.md")
@@ -10,7 +10,7 @@
 LINK_ICON = "🔗"  # Unicode link icon
 
 def fetch_problem_data(slug: str):
-    """Fetch difficulty, tags, and real problem number from LeetCode GraphQL API."""
+    """Fetch problem number, difficulty, and tags from LeetCode GraphQL API."""
     url = "https://leetcode.com/graphql"
     query = {
         "query": """
@@ -24,28 +24,31 @@ def fetch_problem_data(slug: str):
         """,
         "variables": {"titleSlug": slug},
     }
+    headers = {
+        "Content-Type": "application/json",
+        "User-Agent": "Mozilla/5.0"
+    }
     try:
-        resp = requests.post(url, json=query, timeout=10)
+        resp = requests.post(url, json=query, headers=headers, timeout=10)
         resp.raise_for_status()
-        data = resp.json()
-        question = data.get("data", {}).get("question", {})
-        number = question.get("frontendQuestionId", "")
-        difficulty = question.get("difficulty", "")
-        tags = [t['name'] for t in question.get("topicTags", [])]
+        data = resp.json()["data"]["question"]
+        number = data.get("frontendQuestionId", "")
+        difficulty = data.get("difficulty", "")
+        tags = [t["name"] for t in data.get("topicTags", [])]
         return number, difficulty, tags
     except Exception as e:
         print(f"Failed to fetch {slug}: {e}")
         return "", "", []
 
-# MkDocs badge (HTML)
+# MkDocs HTML badge
 def difficulty_badge_site(difficulty: str) -> str:
     colors = {"Easy": "#46c6c2", "Medium": "#fac31d", "Hard": "#f8615c"}
-    color = colors.get(difficulty, "#9E9E9E")
     if not difficulty:
         return ""
+    color = colors.get(difficulty, "#9E9E9E")
     return f'<span style="background-color:#ffffff1a; color:{color}; padding:2px 6px; border-radius:6px;">{difficulty}</span>'
 
-# GitHub badge (Shields.io)
+# GitHub Shields.io badge
 def difficulty_badge_github(difficulty: str) -> str:
     colors = {"Easy": "4c1", "Medium": "f9c851", "Hard": "e05d44"}
     if not difficulty:
@@ -54,16 +57,13 @@ def difficulty_badge_github(difficulty: str) -> str:
     badge_url = f"https://img.shields.io/badge/Difficulty-{difficulty}-{color}.svg"
     return f"![{difficulty}]({badge_url})"
 
-def generate_table_rows(cached_data, badge_func):
+def generate_table_rows(problems, badge_func):
     rows = []
-    for slug, real_number, title_slug in cached_data:
-        difficulty, tags = cached_data[(slug, real_number)]
-        title = f"{real_number}. " + ' '.join([w.capitalize() for w in title_slug.split('-')])
+    for slug, number, title, difficulty, tags in problems:
         diff_badge = badge_func(difficulty)
-        tags_str = ', '.join(tags)
-        solution_link = f"https://leetcode.romitsagu.com/solutions/leetcode{real_number}#/"
+        tags_str = ", ".join(tags)
+        solution_link = f"https://leetcode.romitsagu.com/solutions/leetcode{number}#/"
         icon_md = f"[{LINK_ICON}]({solution_link})"
-
         rows.append(f"| {title} | {icon_md} | {diff_badge} | {tags_str} |")
     return "\n".join(rows)
 
@@ -93,32 +93,30 @@ def update_readme(readme_path, rows_str):
     print(f"{readme_path.name} updated successfully.")
 
 def main():
-    cached_data = []
-    api_cache = {}
+    problems = []
 
-    # Fetch real number, difficulty, and tags for each problem
+    # Step 1: Read all problem folders and fetch data
     for folder in sorted(os.listdir(SRC_DIR)):
         folder_path = Path(SRC_DIR) / folder
         if not folder_path.is_dir():
             continue
+
         match = re.match(r"(\d+)-(.+)", folder)
         if not match:
             continue
-        _, slug = match.groups()
-
-        if slug not in api_cache:
-            real_number, difficulty, tags = fetch_problem_data(slug)
-            api_cache[slug] = (difficulty, tags)
-        else:
-            real_number, _, _ = fetch_problem_data(slug)
+        folder_number, slug = match.groups()
+        folder_number = str(int(folder_number))
+        title = f"{folder_number}. " + " ".join([w.capitalize() for w in slug.split("-")])
 
-        cached_data.append((slug, real_number, slug))
+        # Fetch API data once
+        number, difficulty, tags = fetch_problem_data(slug)
+        problems.append((slug, number, title, difficulty, tags))
 
-    # Generate rows for both READMEs
-    site_rows = generate_table_rows(api_cache, difficulty_badge_site)
-    github_rows = generate_table_rows(api_cache, difficulty_badge_github)
+    # Step 2: Generate table rows
+    site_rows = generate_table_rows(problems, difficulty_badge_site)
+    github_rows = generate_table_rows(problems, difficulty_badge_github)
 
-    # Update both READMEs
+    # Step 3: Update both READMEs
     update_readme(SITE_README_PATH, site_rows)
     update_readme(GITHUB_README_PATH, github_rows)