utils/helpers.py

# utils/helpers.py
"""유틸리티 함수들"""
import os
from typing import Optional


def ensure_dir(directory: str):
    """
    디렉토리가 없으면 생성
    
    Args:
        directory: 디렉토리 경로
    """
    if not os.path.exists(directory):
        os.makedirs(directory)
        print(f"📁 디렉토리 생성: {directory}")


def format_file_size(size_bytes: int) -> str:
    """
    파일 크기를 읽기 쉬운 형식으로 변환
    
    Args:
        size_bytes: 바이트 단위 크기
    
    Returns:
        str: 포맷된 크기 (예: "1.5 MB")
    """
    for unit in ['B', 'KB', 'MB', 'GB']:
        if size_bytes < 1024.0:
            return f"{size_bytes:.1f} {unit}"
        size_bytes /= 1024.0
    return f"{size_bytes:.1f} TB"


def truncate_text(text: str, max_length: int = 100, suffix: str = "...") -> str:
    """
    텍스트를 지정된 길이로 자르기
    
    Args:
        text: 원본 텍스트
        max_length: 최대 길이
        suffix: 잘린 경우 추가할 접미사
    
    Returns:
        str: 잘린 텍스트
    """
    if len(text) <= max_length:
        return text
    return text[:max_length - len(suffix)] + suffix


def safe_get(dictionary: dict, key: str, default: Optional[any] = None):
    """
    안전하게 딕셔너리 값 가져오기
    
    Args:
        dictionary: 딕셔너리
        key: 키
        default: 기본값
    
    Returns:
        값 또는 기본값
    """
    try:
        return dictionary.get(key, default)
    except:
        return default