FIX LOGIC

Author: gh-develop

.github/scripts/update_readme.py

@@ -51,20 +51,54 @@ def parse_readme(lines):
 
 
 # -----------------------------
-# 백준 정보 수집
+# 백준 정보 수집 (정상 동작)
 # -----------------------------
 def fetch_baekjoon_info(number):
     url = f"https://www.acmicpc.net/problem/{number}"
-    res = requests.get(url, timeout=10)
+    headers = {
+        "User-Agent": "Mozilla/5.0"
+    }
+
+    res = requests.get(url, headers=headers, timeout=10)
+    if res.status_code != 200:
+        return {
+            "title": "UNKNOWN",
+            "level": "UNKNOWN",
+            "link": url,
+        }
 
     soup = BeautifulSoup(res.text, "html.parser")
 
-    title = soup.select_one("#problem_title")
-    level = soup.select_one("#problem-info tbody tr td")
+    # 문제 제목
+    title_tag = soup.select_one("#problem_title")
+    title = title_tag.text.strip() if title_tag else "UNKNOWN"
+
+    # solved.ac tier 이미지에서 난이도 추출
+    tier_img = soup.select_one("img.solvedac-tier")
+    level = "UNKNOWN"
+
+    if tier_img and "src" in tier_img.attrs:
+        src = tier_img["src"]
+        # 예: silver5.svg
+        tier = src.split("/")[-1].replace(".svg", "")
+
+        tier_map = {
+            "bronze": "Bronze",
+            "silver": "Silver",
+            "gold": "Gold",
+            "platinum": "Platinum",
+            "diamond": "Diamond",
+            "ruby": "Ruby",
+        }
+
+        for k, v in tier_map.items():
+            if tier.startswith(k):
+                level = f"{v} {tier[len(k):].upper()}"
+                break
 
     return {
-        "title": title.text.strip() if title else "UNKNOWN",
-        "level": level.text.strip() if level else "UNKNOWN",
+        "title": title,
+        "level": level,
         "link": url,
     }
 
@@ -93,7 +127,7 @@ def make_row(index, status, platform, number, info):
 
 
 # -----------------------------
-# 메인 처리
+# 디렉토리 처리
 # -----------------------------
 def process_directory(dir_path):
     readme_path = os.path.join(dir_path, "README.md")
@@ -141,6 +175,9 @@ def process_directory(dir_path):
     return updated
 
 
+# -----------------------------
+# 메인
+# -----------------------------
 def main():
     any_updated = False