Refactor update.py for Python 3 compatibility

Author: Ori-Gui

update.py

@@ -1,13 +1,17 @@
-#!/usr/bin/env python
+# update.py
+#!/usr/bin/env python3
 
 import os
 from urllib import parse
 
-HEADER = """
-# 🎯 백준, 프로그래머스, SWEA 문제 풀이 목록
+README_PATH = "README.md"
+START_MARKER = "<!-- AA-README-START -->"
+END_MARKER = "<!-- AA-README-END -->"
+
+GENERATED_HEADER = """
+## 📚 Problem Solving
 """
 
-# 백준 등급 이모지 매핑 (예시)
 BOJ_TIER_ORDER = {
     "Bronze": "🥉 Bronze",
     "Silver": "🥈 Silver",
@@ -17,7 +21,6 @@
     "Ruby": "❤️ Ruby"
 }
 
-# 프로그래머스 레벨 이모지 매핑 (예시)
 PROGRAMMERS_LEVEL = {
     "0": "🍼 Lv.0",
     "1": "🐣 Lv.1",
@@ -27,112 +30,93 @@
     "5": "🦅 Lv.5"
 }
 
-# SWEA 단계 이모지 매핑 (예시)
-def swea_label(name):
+
+def swea_label(name: str) -> str:
     return f"⭐ {name.upper()}"
 
+
 def parse_problem_folder(folder_name: str) -> str:
-    """
-    '1000. A+B' 같은 폴더 이름을 '1000 - A+B' 로 변환.
-    폴더명에 '.'이 없으면 그대로 반환.
-    """
     if "." in folder_name:
         parts = folder_name.split(".", 1)
         prob_num = parts[0].strip()
         prob_title = parts[1].strip()
         return f"{prob_num} - {prob_title}"
-    else:
-        return folder_name
-
-def extract_problem_number(folder_name: str) -> int:
-    """
-    폴더 이름에서 문제 번호를 추출합니다.
-    예: "1152. A+B" -> 1152
-    폴더 이름에 '.'가 없으면 정렬 시 뒤로 밀리도록 큰 값을 반환.
-    """
+    return folder_name
+
+
+def extract_problem_number(folder_name: str):
     if "." in folder_name:
         try:
-            num_part = folder_name.split(".")[0].strip()
-            return int(num_part)
+            return int(folder_name.split(".")[0].strip())
         except ValueError:
             return float("inf")
-    else:
-        return float("inf")
+    return float("inf")
+
+
+def build_generated_content() -> str:
+    content = GENERATED_HEADER + "\n"
 
-def main():
-    content = HEADER + "\n"
-    
-    # 메인 카테고리 이름들
     main_categories = ["백준", "프로그래머스", "SWEA"]
-    # 데이터 구조:
-    # data[(메인카테고리, 서브카테고리)] = { 문제폴더이름: [파일경로, ...] }
     data = {}
-    
+
     for root, dirs, files in os.walk("."):
-        # .git, .github, images 폴더는 건너뛰기
         dirs[:] = [d for d in dirs if d not in (".git", ".github", "images")]
-        if root == ".":  # 루트 디렉토리는 건너뜁니다.
+
+        if root == ".":
             continue
-        
-        # 경로(root)에서 메인 카테고리가 어느 것에 속하는지 체크
+
         main_cat = None
         for cat in main_categories:
             if os.sep + cat + os.sep in root or root.endswith(os.sep + cat) or root == "./" + cat:
                 main_cat = cat
                 break
+
         if main_cat is None:
             continue
-        
-        # 메인 카테고리 바로 아래 폴더(또는 그 아래 하위 경로)를 서브 카테고리(sub_cat)로 간주합니다.
+
         main_path = os.path.join(".", main_cat)
-        rel_path = os.path.relpath(root, main_path)  # 예: 'Bronze/1000. A+B' 등
+        rel_path = os.path.relpath(root, main_path)
         parts = rel_path.split(os.sep)
-        sub_cat = parts[0]  # 예: Bronze 또는 "." (파일이 바로 메인 폴더에 있을 때)
-        
-        # 문제 폴더가 있다면 (예: "1000. A+B")
+        sub_cat = parts[0]
+
         problem_folder = None
         if len(parts) > 1:
             problem_folder = parts[1]
-        
-        # data 초기화
+
         if (main_cat, sub_cat) not in data:
             data[(main_cat, sub_cat)] = {}
-        
+
         if problem_folder is not None:
             if problem_folder not in data[(main_cat, sub_cat)]:
                 data[(main_cat, sub_cat)][problem_folder] = []
-            
+
             for file in files:
-                if file.lower() == "readme.md":  # 문제 설명 파일 제외
+                if file.lower() == "readme.md":
                     continue
                 full_path = os.path.join(root, file)
                 data[(main_cat, sub_cat)][problem_folder].append(full_path)
-        else:
-            # sub_cat 자체가 문제 폴더인 경우(메인 카테고리 폴더에 바로 파일이 있는 경우)
-            pass
-    
-    # data의 내용을 바탕으로 content 구성 - 각 서브 카테고리(레벨)별로 하나의 표 생성
+
     for main_cat in main_categories:
-        # 해당 메인 카테고리에 해당하는 (main_cat, sub_cat) 키 목록
         keys = [(k, v) for k, v in data.items() if k[0] == main_cat]
         if not keys:
             continue
-        
-        # 메인 카테고리 헤더
+
         content += "---\n"
         content += f"## 📚 {main_cat}\n"
-        
-        # 서브 카테고리(예: Bronze, Silver 등) 정렬
+
         if main_cat == "백준":
             order = ["Bronze", "Silver", "Gold", "Platinum", "Diamond", "Ruby"]
-            keys_sorted = sorted(keys, key=lambda item: order.index(item[0][1]) if item[0][1] in order else 999)
+            keys_sorted = sorted(
+                keys,
+                key=lambda item: order.index(item[0][1]) if item[0][1] in order else 999
+            )
         else:
             keys_sorted = sorted(keys, key=lambda x: x[0][1])
-        
+
         for (mc, sub_cat), problem_map in keys_sorted:
             if sub_cat == ".":
                 continue
-            
+
             if mc == "백준":
                 tier_title = BOJ_TIER_ORDER.get(sub_cat, f"✅ {sub_cat}")
             elif mc == "프로그래머스":
@@ -145,17 +129,42 @@ def main():
             content += f"### {tier_title}\n"
             content += "| 문제 | 링크 |\n"
             content += "| ----- | ---- |\n"
-            
-            # 문제 폴더를 문제 번호 기준(숫자)으로 정렬합니다.
-            for pfolder, file_list in sorted(problem_map.items(), key=lambda item: extract_problem_number(item[0])):
+
+            for pfolder, _ in sorted(problem_map.items(), key=lambda item: extract_problem_number(item[0])):
                 parsed_name = parse_problem_folder(pfolder)
                 folder_path = os.path.join(".", mc, sub_cat, pfolder)
                 content += f"| {parsed_name} | [링크]({parse.quote(folder_path)}) |\n"
-            
+
             content += "\n"
-    
-    with open("README.md", "w", encoding="utf-8") as fd:
-        fd.write(content)
+
+    return content.strip() + "\n"
+
+
+def update_readme():
+    generated = build_generated_content()
+
+    if os.path.exists(README_PATH):
+        with open(README_PATH, "r", encoding="utf-8") as f:
+            readme = f.read()
+    else:
+        readme = ""
+
+    block = f"{START_MARKER}\n{generated}{END_MARKER}"
+
+    if START_MARKER in readme and END_MARKER in readme:
+        start = readme.index(START_MARKER)
+        end = readme.index(END_MARKER) + len(END_MARKER)
+        updated = readme[:start] + block + readme[end:]
+    else:
+        prefix = readme.strip()
+        if prefix:
+            updated = prefix + "\n\n---\n\n" + block + "\n"
+        else:
+            updated = block + "\n"
+
+    with open(README_PATH, "w", encoding="utf-8") as f:
+        f.write(updated)
+
 
 if __name__ == "__main__":
-    main()
+    update_readme()