import pandas as pd
import os

def process(file_number):
    # 讀入Excel內容
    bh = pd.read_excel(f'Bump Report_{file_number}.xls')
    cop = pd.read_excel(f'planarity Report_{file_number}.xls')

    # 選取需要的資料範圍
    bh2 = bh.iloc[4:,5]  
    bh4 = bh.iloc[4:,9]
    cop2 = cop.iloc[4:, 10]

    # 將選取的資料合併成一個新的DataFrame
    data = {
        'Lot': bh2,
        'Wafer': bh4,
        '': cop2
    }
    new_df = pd.DataFrame(data)

    # 建立新的DataFrame和額外的資料
    data = {
        'Lot': [bh.iloc[2,2]],
        'Wafer': [bh.iloc[2,3]],
        '': ['']
    }
    df_result = pd.DataFrame(data)

    # 將資料全部合併成一個DataFrame
    # Pandas 1.x寫法：df_result = df_result.append(new_df, ignore_index=True)
    df_result = pd.concat([df_result, new_df])

    # 顯示結果
    print(df_result)

    # 存為Excel
    df_result.to_excel(f'final-{file_number}.xlsx')

# 取得目錄下所有檔案
all_files = os.listdir()

for f in all_files:
    
    if('Bump Report_' in f):
        # 讀出Bump Report_後面的編號
        file_number = f.replace('Bump Report_', '').replace('.xls', '')

        # 將編號傳給process函式，函式會用途一個編號去抓兩個excel檔案來合併
        process(file_number)