hariqueen commited on
Commit
37c3c08
Β·
verified Β·
1 Parent(s): d169b4b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -14
main.py CHANGED
@@ -52,16 +52,6 @@ def process_rental_company(company_name: str):
52
 
53
 
54
  def process_rental_company_with_voucher(uploaded_file_path, voucher_number):
55
- """
56
- μ‚¬μš©μžκ°€ μ—…λ‘œλ“œν•œ 파일과 μ „ν‘œλ²ˆν˜Έλ₯Ό λ°›μ•„ ERP μžλ™ μ „ν‘œνŒŒμΌμ„ μƒμ„±ν•˜λŠ” ν•¨μˆ˜
57
-
58
- Args:
59
- uploaded_file_path (str): μ—…λ‘œλ“œλœ 파일 경둜 (CSV λ³€ν™˜λœ 경둜)
60
- voucher_number (str): μ‚¬μš©μž μž…λ ₯ μ „ν‘œλ²ˆν˜Έ
61
-
62
- Returns:
63
- output_path (str): μ΅œμ’… μ €μž₯된 파일 경둜 (xlsx)
64
- """
65
  company_name = "ν•œκ΅­λ Œνƒˆ"
66
  company_config = cfg.RENTAL_COMPANIES[company_name]
67
 
@@ -74,23 +64,26 @@ def process_rental_company_with_voucher(uploaded_file_path, voucher_number):
74
  erp_df = erp_generator.prepare_erp_columns(erp_df)
75
  erp_df = erp_generator.set_management_items(erp_df, df_filtered, company_config)
76
 
77
- # μ „ν‘œλ²ˆν˜Έ μ±„μš°κΈ°
78
  if 'ROW_ID' in erp_df.columns:
79
  erp_df['ROW_ID'] = voucher_number
80
  if 'NO_DOCU' in erp_df.columns:
81
  erp_df['NO_DOCU'] = voucher_number
82
 
83
- # ⭐ ERP ν…œν”Œλ¦Ώ 양식에 λ§žμΆ°μ„œ μ€€λΉ„
84
  erp_form = file_handler.load_erp_form_template(company_config['erp_form_file'])
 
 
85
  result_df = file_handler.prepare_file_with_template(erp_df, erp_form)
86
 
87
  # μ €μž₯
88
  output_filename = f"μžλ™μ „ν‘œ_μ™„μ„±νŒŒμΌ_{datetime.now().strftime('%Y%m%d')}.xlsx"
89
  output_path = os.path.join(cfg.OUTPUT_DIR, output_filename)
 
90
  os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
91
 
92
- # ⭐ λ°˜λ“œμ‹œ result_df μ €μž₯
93
- result_df.to_excel(output_path, index=False, engine="openpyxl", startrow=cfg.ERP_DATA_ROW_START - 1)
94
 
95
  return output_path
96
 
 
52
 
53
 
54
  def process_rental_company_with_voucher(uploaded_file_path, voucher_number):
 
 
 
 
 
 
 
 
 
 
55
  company_name = "ν•œκ΅­λ Œνƒˆ"
56
  company_config = cfg.RENTAL_COMPANIES[company_name]
57
 
 
64
  erp_df = erp_generator.prepare_erp_columns(erp_df)
65
  erp_df = erp_generator.set_management_items(erp_df, df_filtered, company_config)
66
 
67
+ # μ „ν‘œλ²ˆν˜Έ μ±„μ›Œλ„£κΈ°
68
  if 'ROW_ID' in erp_df.columns:
69
  erp_df['ROW_ID'] = voucher_number
70
  if 'NO_DOCU' in erp_df.columns:
71
  erp_df['NO_DOCU'] = voucher_number
72
 
73
+ # ERP 양식 λ‘œλ“œ
74
  erp_form = file_handler.load_erp_form_template(company_config['erp_form_file'])
75
+
76
+ # ERP 양식에 λ§žμΆ°μ„œ 데이터 μ€€λΉ„
77
  result_df = file_handler.prepare_file_with_template(erp_df, erp_form)
78
 
79
  # μ €μž₯
80
  output_filename = f"μžλ™μ „ν‘œ_μ™„μ„±νŒŒμΌ_{datetime.now().strftime('%Y%m%d')}.xlsx"
81
  output_path = os.path.join(cfg.OUTPUT_DIR, output_filename)
82
+
83
  os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
84
 
85
+ # 이제 μ€€λΉ„λœ result_dfλ₯Ό μ €μž₯
86
+ result_df.to_excel(output_path, index=False, engine='openpyxl')
87
 
88
  return output_path
89