#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os
import sys

# 1) 여기에 실제 베이스 경로를 넣어주세요.
BASE_DIRS = [
    "/data/ai-hub/ppe-dataset/105.공사현장_안전장비_인식_데이터/01.데이터/1.Training/라벨링데이터_241008_add",
    "/data/ai-hub/ppe-dataset/105.공사현장_안전장비_인식_데이터/01.데이터/2.Validation/라벨링데이터(zip)_241008_add"
]

# 2) 변경할 폴더명 매핑: old_name -> new_name
RENAME_MAP = {
    "1.怨듬룞二쇳깮": "1.공동주택",
    "2.以묒옣鍮꾨쭔": "2.중장비만",
    "3.怨듭옣":      "3.공장",
    "9.二쇱긽蹂듯빀": "9.주상복합"
}

def rename_dirs(base_path):
    if not os.path.isdir(base_path):
        print(f"[ERROR] 경로를 찾을 수 없습니다: {base_path}", file=sys.stderr)
        return
    # 하위 디렉토리부터 올라오면서 처리
    for root, dirs, _ in os.walk(base_path, topdown=False):
        for d in dirs:
            if d in RENAME_MAP:
                old_path = os.path.join(root, d)
                new_name = RENAME_MAP[d]
                new_path = os.path.join(root, new_name)
                if os.path.exists(new_path):
                    print(f"[SKIP] 이미 존재: {new_path}")
                else:
                    os.rename(old_path, new_path)
                    print(f"[RENAMED] {old_path} → {new_path}")

if __name__ == "__main__":
    for base in BASE_DIRS:
        print(f"\n>>> 처리 시작: {base}")
        rename_dirs(base)
    print("\n모든 처리 완료!")