import os
import pandas as pd
from openpyxl import load_workbook

import os
import pandas as pd
from processors import process_excel

INPUT_DIR = "input"
OUTPUT_DIR = "output"

def ensure_output_folder():
    if not os.path.exists(OUTPUT_DIR):
        os.makedirs(OUTPUT_DIR)

def main():
    ensure_output_folder()

    for filename in os.listdir(INPUT_DIR):
        if filename.endswith((".xlsx", ".xls")):
            path = os.path.join(INPUT_DIR, filename)
            print(f"正在處理: {filename}")

            df = pd.read_excel(path)

            # 調(diào)用自定義處理邏輯
            df = process_excel(df)

            # 輸出結(jié)果
            output_path = os.path.join(OUTPUT_DIR, filename)
            df.to_excel(output_path, index=False)
            print(f"已保存到: {output_path}")

if __name__ == "__main__":
    main()

def process_excel(df):
    # 1 刪除全空行
    df = df.dropna(how="all")

    # 2 統(tǒng)一列名格式
    df.columns = [str(c).strip() for c in df.columns]

    # 3 添加一個時間戳列
    import datetime
    df["處理時間"] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")

    return df

def process_excel(df):
    if "產(chǎn)品名稱" in df.columns:
        df["產(chǎn)品名稱"] = df["產(chǎn)品名稱"].replace("舊名", "新名")
    return df

def process_excel(df):
    return df  # 不做處理，原樣返回

all_data = []

for filename in os.listdir(INPUT_DIR):
    if filename.endswith((".xls", ".xlsx")):
        df = pd.read_excel(os.path.join(INPUT_DIR, filename))
        df["來源文件"] = filename
        all_data.append(df)

final_df = pd.concat(all_data, ignore_index=True)
final_df.to_excel("output/合并結(jié)果.xlsx", index=False)

def process_excel(df):
    for name, group in df.groupby("分類"):
        group.to_excel(f"output/{name}.xlsx", index=False)
    return None