Upload 5 files

utils/__init__.py

@@ -0,0 +1,4 @@
+from utils.file_utils import ensure_directory_exists
+from utils.excel_utils import save_to_files, save_to_csv, save_to_excel
+from utils.template_utils import load_erp_form_template, prepare_file_with_template
+from utils.reporting_utils import print_data_summary, generate_report_file

utils/excel_utils.py

@@ -0,0 +1,68 @@
+"""
+Excel 파일 처리 유틸리티
+"""
+import os
+import pandas as pd
+from typing import Dict, Any
+from core import config as cfg
+from pyexcel_xls import save_data
+from collections import OrderedDict
+
+def save_to_csv(df: pd.DataFrame, output_path: str, data_count: int = 0) -> bool:
+    try:
+        df.to_csv(output_path, index=False, encoding=cfg.CSV_OUTPUT_ENCODING)
+        print(f"처리 완료: {data_count}개 행이 '{output_path}'에 저장됨 ({cfg.CSV_OUTPUT_ENCODING} 인코딩)")
+        print(f"데이터는 {cfg.ERP_DATA_ROW_START}행부터 시작합니다.")
+        return True
+    except Exception as e:
+        print(f"CSV 파일 저장 중 오류 발생: {e}")
+        return False
+
+def save_to_excel(df: pd.DataFrame, output_path: str, data_count: int = 0) -> bool:
+    try:
+        # Excel 97-2003 형식(.xls)으로 저장
+        xls_path = output_path.replace('.xlsx', '.xls')
+        
+        # 데이터프레임을 리스트로 변환
+        headers = df.columns.tolist()
+        data = [headers]  # 헤더를 첫 번째 행으로 추가
+        
+        # 데이터프레임의 각 행을 리스트로 변환하여 data에 추가
+        for _, row in df.iterrows():
+            data.append(row.tolist())
+        
+        # OrderedDict 생성 (Sheet1이라는 이름의 시트에 데이터 저장)
+        data_dict = OrderedDict()
+        data_dict["Sheet1"] = data
+        
+        # xls 파일로 저장
+        save_data(xls_path, data_dict)
+        
+        print(f"처리 완료: {data_count}개 행이 '{xls_path}'에 저장됨")
+        print(f"엑셀 파일이 성공적으로 생성되었습니다: {os.path.abspath(xls_path)}")
+        return True
+    except Exception as e:
+        print(f"엑셀 파일 저장 중 오류 발생: {e}")
+        
+        # 대체 저장 시도 (일반 Excel 형식)
+        try:
+            backup_path = output_path
+            df.to_excel(backup_path, index=False, engine='openpyxl')
+            print(f"대체 형식(.xlsx)으로 파일 저장 완료: {backup_path}")
+            return True
+        except Exception as ex:
+            print(f"대체 저장도 실패: {ex}")
+            return False
+
+def save_to_files(result_df: pd.DataFrame, output_csv: str, output_excel: str, erp_data_count: int) -> None:
+    # CSV 파일 저장
+    print(f"'{output_csv}'로 CSV 저장 중...")
+    csv_saved = save_to_csv(result_df, output_csv, erp_data_count)
+    
+    # Excel 파일 저장
+    print(f"\n'{output_excel}'로 엑셀 파일 저장 중...")
+    excel_saved = save_to_excel(result_df, output_excel, erp_data_count)
+    
+    if csv_saved and not excel_saved:
+        print("CSV 파일은 정상적으로 저장되었습니다.")
+        print("CSV 파일을 열 때는 Excel의 '데이터' 탭에서 '텍스트/CSV에서' 기능을 사용하시기 바랍니다.")

utils/file_utils.py

@@ -0,0 +1,16 @@
+"""
+파일 기본 처리 유틸리티
+"""
+import os
+from typing import Dict, Any
+
+def ensure_directory_exists(dir_path: str) -> None:
+    """
+    디렉토리가 없으면 생성
+    
+    Args:
+        dir_path: 생성할 디렉토리 경로
+    """
+    if not os.path.exists(dir_path):
+        os.makedirs(dir_path)
+        print(f"디렉토리 생성: {dir_path}")

utils/template_utils.py

@@ -0,0 +1,75 @@
+"""
+ERP 양식 및 템플릿 관련 유틸리티
+"""
+import pandas as pd
+from typing import Dict, Any, Optional
+from core import config as cfg
+
+def load_erp_form_template(erp_form_file: str) -> Optional[pd.DataFrame]:
+    """
+    ERP 양식 파일 로드
+    
+    Args:
+        erp_form_file: ERP 양식 파일 경로
+        
+    Returns:
+        ERP 양식 데이터프레임 또는 None
+    """
+    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 prepare_file_with_template(erp_df: pd.DataFrame, erp_form: Optional[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 복사 (양식 파일의 처음 n행만 사용)
+        result_df = erp_form.copy()
+        
+        # 양식 파일이 지정된 시작행보다 많으면 필요한 만큼만 유지
+        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