fix logic

Author: gh-develop

.github/scripts/update_readme.py

@@ -2,14 +2,14 @@
 import re
 import requests
 from bs4 import BeautifulSoup
-import certifi  # SSL 검증용
+import certifi
 
 ROOT_DIR = "."
 VALID_EXT = (".py", ".cpp", ".java", ".kt", ".js")
 
-# -------------------------------
+# -----------------------------
 # 난이도 정렬 기준 (백준)
-# -------------------------------
+# -----------------------------
 BOJ_LEVEL_RANK = {
     "Bronze V": 1, "Bronze IV": 2, "Bronze III": 3, "Bronze II": 4, "Bronze I": 5,
     "Silver V": 6, "Silver IV": 7, "Silver III": 8, "Silver II": 9, "Silver I": 10,
@@ -19,37 +19,32 @@
     "Ruby V": 26, "Ruby IV": 27, "Ruby III": 28, "Ruby II": 29, "Ruby I": 30,
 }
 
-# -------------------------------
-# 백준 API
-# -------------------------------
+# -----------------------------
+# 백준 정보 가져오기
+# -----------------------------
 def fetch_boj_info(number):
     url = f"https://solved.ac/api/v3/problem/show?problemId={number}"
     try:
         r = requests.get(url, timeout=10, verify=certifi.where())
         if r.status_code != 200:
             return None
-    except requests.exceptions.RequestException:
+        data = r.json()
+        level_name = [k for k, v in BOJ_LEVEL_RANK.items() if v == data.get("level", 0)]
+        level_name = level_name[0] if level_name else "UNKNOWN"
+        return {"title": data.get("titleKo", "UNKNOWN"), "level": level_name, "link": f"https://www.acmicpc.net/problem/{number}"}
+    except Exception:
         return None
 
-    data = r.json()
-    level = data.get("level", 0)
-    title = data.get("titleKo", "UNKNOWN")
-
-    level_name = [k for k, v in BOJ_LEVEL_RANK.items() if v == level]
-    level_name = level_name[0] if level_name else "UNKNOWN"
-
-    return {"title": title, "level": level_name, "link": f"https://www.acmicpc.net/problem/{number}"}
-
-# -------------------------------
-# 프로그래머스
-# -------------------------------
+# -----------------------------
+# 프로그래머스 정보 가져오기
+# -----------------------------
 def fetch_programmers_info(pid):
     url = f"https://school.programmers.co.kr/learn/courses/30/lessons/{pid}"
     try:
         r = requests.get(url, timeout=10, verify=certifi.where())
         if r.status_code != 200:
             return None
-    except requests.exceptions.RequestException:
+    except Exception:
         return None
 
     soup = BeautifulSoup(r.text, "html.parser")
@@ -61,9 +56,9 @@ def fetch_programmers_info(pid):
 
     return {"title": title, "level": level, "link": url}
 
-# -------------------------------
+# -----------------------------
 # 정렬 기준
-# -------------------------------
+# -----------------------------
 def platform_rank(platform):
     return 0 if platform == "백준" else 1
 
@@ -75,9 +70,9 @@ def level_rank(platform, level):
         return int(m.group(1)) if m else 999
     return 999
 
-# -------------------------------
-# README 업데이트
-# -------------------------------
+# -----------------------------
+# 디렉토리 처리
+# -----------------------------
 def process_directory(dir_path):
     readme_path = os.path.join(dir_path, "README.md")
     if not os.path.isfile(readme_path):
@@ -105,27 +100,45 @@ def process_directory(dir_path):
             status = "❌" if name.endswith("_FAIL") else "✅"
             problems[f"pg_{num}"] = ("프로그래머스", num, status)
 
+    # 기존 README 읽기
+    existing_numbers = set()
+    with open(readme_path, "r", encoding="utf-8") as f:
+        lines = f.readlines()
+        for line in lines:
+            m = re.search(r"\|\s*(?:⬜|✅|❌)\s*\|\s*(백준|프로그래머스)\s*\|\s*(\d+)", line)
+            if m:
+                existing_numbers.add(m.group(2))
+
+    # 기존 문제 + 새 문제 모두 rows로 준비
     rows = []
 
     for _, (platform, num, status) in problems.items():
-        info = fetch_boj_info(num) if platform == "백준" else fetch_programmers_info(num)
-        if not info:
-            continue
+        if platform == "백준":
+            info = fetch_boj_info(num)
+        else:
+            info = fetch_programmers_info(num)
+
+        title = info["title"] if info else "UNKNOWN"
+        level = info["level"] if info else "UNKNOWN"
+        link = info["link"] if info else "#"
+
         rows.append({
             "platform": platform,
             "number": num,
             "status": status,
-            "title": info["title"],
-            "level": info["level"],
-            "link": info["link"]
+            "title": title,
+            "level": level,
+            "link": link
         })
 
+    # 정렬
     rows.sort(key=lambda r: (
         platform_rank(r["platform"]),
         level_rank(r["platform"], r["level"]),
         int(r["number"])
     ))
 
+    # README 작성
     new_lines = [
         "| 순번 | 풀이 여부 | 플랫폼 | 문제 번호 | 문제 이름 | 난이도 | 풀이 링크 |\n",
         "|----|----|----|----|----|----|----|\n"
@@ -143,14 +156,20 @@ def process_directory(dir_path):
 
     return True
 
-# -------------------------------
+# -----------------------------
 # main
-# -------------------------------
+# -----------------------------
 def main():
     for root, dirs, files in os.walk(ROOT_DIR):
         if root.startswith("./.git") or root.startswith("./.github"):
             continue
+
+        # ✅ 루트 README 제외
+        if root == ".":
+            continue
+
         process_directory(root)
 
+
 if __name__ == "__main__":
     main()