Update main.py

main.py

@@ -10,6 +10,7 @@ import data_processor
 import erp_generator
 import file_handler
 import reporter
+import pandas as pd
 
 
 def process_rental_company(company_name: str):
@@ -82,6 +83,57 @@ def process_uploaded_file(file_obj):
 
     return output_csv
 
+def process_rental_company_with_voucher(uploaded_file_path, voucher_number):
+    """
+    사용자가 업로드한 파일과 전표번호를 받아 ERP 자동 전표파일을 생성하는 함수
+
+    Args:
+        uploaded_file_path (str): 업로드된 파일 경로 (CSV 변환된 경로)
+        voucher_number (str): 사용자 입력 전표번호
+
+    Returns:
+        output_path (str): 최종 저장된 파일 경로 (xlsx)
+    """
+    # 한국렌탈 기준으로 설정 (혹시 다르면 추후 확장 가능)
+    company_name = "한국렌탈"
+    company_config = cfg.RENTAL_COMPANIES[company_name]
+
+    # 매핑 정보 로드
+    mapping_file = company_config['mapping_file']
+    mapping_dict = mapping_utils.load_mapping_file(mapping_file)
+
+    # 데이터 로드 및 전처리
+    df, df_filtered = data_processor.load_and_preprocess_data(uploaded_file_path, company_config, mapping_dict)
+
+    # 데이터 요약 생성
+    summary = data_processor.summarize_data(df_filtered, mapping_dict)
+
+    # ERP 데이터 생성
+    erp_df = erp_generator.generate_erp_data(df_filtered, company_config)
+
+    # ERP 표준 컬럼 구조로 준비
+    erp_df = erp_generator.prepare_erp_columns(erp_df)
+
+    # 관리항목 설정
+    erp_df = erp_generator.set_management_items(erp_df, df_filtered, company_config)
+
+    # **여기서 전표번호 채워넣기**
+    if 'ROW_ID' in erp_df.columns:
+        erp_df['ROW_ID'] = voucher_number
+    if 'NO_DOCU' in erp_df.columns:
+        erp_df['NO_DOCU'] = voucher_number
+
+    # 저장 경로 설정
+    output_filename = f"자동전표_완성파일_{datetime.now().strftime('%Y%m%d')}.xlsx"
+    output_path = os.path.join(cfg.OUTPUT_DIR, output_filename)
+
+    os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
+
+    # 엑셀로 저장
+    erp_df.to_excel(output_path, index=False)
+
+    return output_path
+
 
 def main():
     """