utils_os.py

Go to the documentation of this file.

import os
import json
import numpy as np
import shutil
 
def search_file(folder, fileEx):
    """
    폴더에서 특정 확장자의 파일을 검색하는 기능
    하위 폴더까지 모두 검색
    args:
        folder (str): 검색하려는 폴더
        fileEx (str): 파일 확장자
    return:
        file_name_list (list[str]): 파일 이름 리스트
        file_path_list (list[str]): 파일 경로 리스트
    """
    
    file_path_list = []
    for path, dirs, files in os.walk(folder):
        if path.split('/')[-1].startswith('.'):
            continue
        file_path_list.extend([ os.path.join(path, file) for file in files if file.endswith(fileEx)])
 
    file_path_list.sort()
    
    file_name_list = []
    for path in file_path_list:            
        file_name_list.append(os.path.basename(path))
    
    return file_name_list, file_path_list
 
def search_folder(folder):
    """
    폴더내부에 있는 폴더를 검색하는 기능
    args:
        folder (str): 검색하려는 폴더
    return:
        sub_folder_name_list (list[str]): 폴더명 리스트
        sub_folder_path_list (list[str]): 폴더 경로 리스트
    """
    sub_folder_path_list = []
    sub_folder_name_list = []
    for item in os.listdir(folder): # 해당 폴더 내 모든 파일 및 폴더 추출
        sub_folder = os.path.join(folder, item)
 
        if os.path.isdir(sub_folder): # 폴더 여부 확인
            sub_folder_path_list.append(sub_folder)
            sub_folder_name_list.append(item)
            
    return sub_folder_name_list, sub_folder_path_list
 
 
def create_directory(directory):
    """
    디렉토리를 생성하는 기능
    하위 디렉토리를 모두 생성함
    args:
        directory (str): 생성하려는 디렉토리
    """
    try:
        if not os.path.exists(directory):
            os.makedirs(directory)
        else:
            pass
    except OSError:
        print("Error: Failed to create the directory.")
 
 
 
 
class JsonEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.integer):
            return int(obj)
        elif isinstance(obj, np.floating):
            return float(obj)
        elif isinstance(obj, np.ndarray):
            return obj.tolist()
        else:
            return super(JsonEncoder, self).default(obj)  
 
def save_json(data, path):
    """
    json 파일을 저장하는 기능
    args:
        data (dict): 데이터
        path (str): 저장 경로
    """
    try:
        with open(path, "w") as json_file:
            json.dump(data, json_file)
    except Exception  as e:
        print(e)
 
def load_json(path):
    """
    json 파일을 로드하는 기능
    args:
        path (str): json 파일 경로
    return (dict): json 파일의 데이터
    """
    try:
        with open(path, "r") as json_file:
            data = json.load(json_file)
    except Exception  as e:
        data = None
 
    return data
 
def copy_file(src_path, dst_path):
    """
    파일을 복사하는 기능
    args:
        src_path (str): 복사하려는 파일 경로
        dst_path (str): 복사한 파일을 저장하는 경로
    """
    try:
        shutil.copy(src_path, dst_path)
    except Exception as e:
        print(f"파일 복사 중 오류 발생: {src_path} -> {dst_path}")