import os
import shutil
import unicodedata

"""
def copy_dic_subdirs(source_root, target_root_b, target_root_c):
    for root, dirs, files in os.walk(source_root):
        for dir_name in dirs:
        
            parent_dir = os.path.basename(os.path.dirname(os.path.join(root, dir_name)))
            target_root = None
            parent_dir = unicodedata.normalize('NFC', parent_dir)

            # 부모 디렉토리가 '2교시'인지, '3교시'인지 확인
            if parent_dir == '1교시':
                target_root = target_root_b
            elif parent_dir == '2교시':
                target_root = target_root_c
            
            if target_root:
                source_dic_path = os.path.join(root, dir_name)
                target_dic_path = os.path.join(target_root, dir_name)
                
                # DIC 하위 디렉토리와 파일 복사
                shutil.copytree(source_dic_path, target_dic_path, dirs_exist_ok=True)
                print(f"Copied {source_dic_path} to {target_dic_path}")
            
            else:
                print(f"Skipping {dir_name} under {parent_dir}, as it doesn't match '2교시' or '3교시'.")
"""

def copy_ent_files(source_root, target_root):
    # 대상 디렉토리가 없으면 생성
    os.makedirs(target_root, exist_ok=True)
    
    for root, dirs, files in os.walk(source_root):
        for file in files:
            if file.endswith('.ent'):
                space_remove_file = file.replace(" ","")
                
                source_file_path = os.path.join(root, file)
                target_file_path = os.path.join(target_root, space_remove_file)
                
                # 파일 복사
                shutil.copy2(source_file_path, target_file_path)
                print(f"Copied {source_file_path} to {target_file_path}")

# 사용법

# source_directory = r"D:\project\data\CAS_제2510회 정기\제2510회 코딩활용능력 2급 정기 답안파일"  # 원본 디렉토리 경로
source_directory = r"D:\project\data\CAT_제2511회 정기\답안파일"  # 원본 디렉토리 경로
target_directory = r".\ent\2511_CAT_3_A"
# target_directory_a = r"./output/A"  # '1교시'의 타겟 경로
# target_directory_b = r"./output/B"  # '2교시'의 타겟 경로
# target_directory_c = r"./output/C"  # '3교시'의 타겟 경로

copy_ent_files(source_directory, target_directory)