cat/01_copy_files.py

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급 정기 답안파일"  # 원본 디렉토리 경로

exam_name = "2602_CAS_2_A"
# exam_name = "2602_6_CAS_2_B"

# source_directory = r"D:\project\data\CAT_제2511회 정기\답안파일"  # 원본 디렉토리 경로
source_directory = r"D:\project\data\제2602회 수시3_CAS(대구)\답안파일"  # 원본 디렉토리 경로
target_directory = rf".\ent\{exam_name}"  # 타겟 디렉토리 경로

copy_ent_files(source_directory, target_directory)