file_handler.py

"""
파일 입출력 관련 모듈
"""
import os
import pandas as pd
from typing import Dict, Any
import config as cfg


def load_erp_form_template(erp_form_file: str) -> pd.DataFrame:
    """
    ERP 양식 파일 로드
    
    Args:
        erp_form_file: ERP 양식 파일 경로
        
    Returns:
        ERP 양식 데이터프레임
    """
    try:
        erp_form = pd.read_csv(erp_form_file, encoding=cfg.DEFAULT_ENCODING)
        print(f"ERP 양식 파일 '{erp_form_file}'을 성공적으로 로드했습니다.")
        return erp_form
    except Exception as e:
        print(f"ERP 양식 파일 로드 실패: {e}")
        print("기본 양식 없이 진행합니다.")
        return None


def save_to_files(result_df: pd.DataFrame, output_csv: str, output_excel: str, erp_data_count: int) -> None:
    """
    결과를 CSV 및 Excel 파일로 저장
    
    Args:
        result_df: 결과 데이터프레임
        output_csv: CSV 출력 파일 경로
        output_excel: Excel 출력 파일 경로
        erp_data_count: ERP 데이터 행 수
        
    Returns:
        None
    """
    # CSV 파일 저장
    print(f"'{output_csv}'로 CSV 저장 중...")
    try:
        result_df.to_csv(output_csv, index=False, encoding=cfg.CSV_OUTPUT_ENCODING)
        print(f"처리 완료: {erp_data_count}개 행이 '{output_csv}'에 저장됨 ({cfg.CSV_OUTPUT_ENCODING} 인코딩)")
        print(f"데이터는 {cfg.ERP_DATA_ROW_START}행부터 시작합니다.")
    except Exception as e:
        print(f"CSV 파일 저장 중 오류 발생: {e}")
    
    # Excel 파일 저장
    print(f"\n'{output_excel}'로 엑셀 파일 저장 중...")
    try:
        result_df.to_excel(output_excel, index=False, engine='openpyxl')
        print(f"처리 완료: {erp_data_count}개 행이 '{output_excel}'에 저장됨")
        print(f"엑셀 파일이 성공적으로 생성되었습니다: {os.path.abspath(output_excel)}")
    except Exception as e:
        print(f"엑셀 파일 저장 중 오류 발생: {e}")
        print("CSV 파일은 정상적으로 저장되었습니다.")
        print("CSV 파일을 열 때는 Excel의 '데이터' 탭에서 '텍스트/CSV에서' 기능을 사용하시기 바랍니다.")


def prepare_file_with_template(erp_df: pd.DataFrame, erp_form: pd.DataFrame) -> pd.DataFrame:
    """
    ERP 양식을 적용하여 파일 준비
    
    Args:
        erp_df: ERP 데이터프레임
        erp_form: ERP 양식 데이터프레임
        
    Returns:
        결과 데이터프레임
    """
    if erp_form is not None:
        # 양식 파일의 컬럼 순서 사용
        form_columns = erp_form.columns.tolist()
        
        # 결과 데이터프레임을 양식 컬럼 순서에 맞게 재정렬
        for col in form_columns:
            if col not in erp_df.columns:
                erp_df[col] = ""
        
        erp_df = erp_df[form_columns]
        
        # erp_form 복사 (양식 파일의 처음 4행만 사용)
        result_df = erp_form.copy()
        
        # 양식 파일이 4행보다 많으면 4행만 유지
        if len(result_df) > cfg.ERP_DATA_ROW_START - 1:
            result_df = result_df.iloc[:(cfg.ERP_DATA_ROW_START - 1)]
        
        # 빈 행 추가 (필요한 경우)
        current_rows = len(result_df)
        target_rows = cfg.ERP_DATA_ROW_START - 1  # 시작행 - 1 (인덱스는 0부터 시작하므로)
        
        # 현재 행 수가 타겟 행 수보다 적으면 빈 행 추가
        if current_rows < target_rows:
            empty_rows_needed = target_rows - current_rows
            empty_df = pd.DataFrame([[""] * len(form_columns) for _ in range(empty_rows_needed)], columns=form_columns)
            result_df = pd.concat([result_df, empty_df], ignore_index=True)
        
        # 처리된 데이터 추가 (ERP_DATA_ROW_START행부터 시작)
        result_df = pd.concat([result_df, erp_df], ignore_index=True)
        return result_df
    else:
        # 양식 파일이 없는 경우 빈 데이터프레임 생성 후 데이터 추가
        # 필요한 빈 행 생성 (ERP_DATA_ROW_START-1개의 빈 행)
        empty_rows = cfg.ERP_DATA_ROW_START - 1
        empty_df = pd.DataFrame([[""] * len(erp_df.columns) for _ in range(empty_rows)], columns=erp_df.columns)
        result_df = pd.concat([empty_df, erp_df], ignore_index=True)
        return result_df