import xlrd
import xlsxwriter
import glob
biao_tou = []
# 獲取要合并的所有exce表格
def get_exce():
    all_exce = glob.glob("*.xls*")
    print("該目錄下有" + str(len(all_exce)) + "個(gè)exce文件：")
    if (len(all_exce) == 0):
        return 0
    else:
        for i in range(len(all_exce)):
            print(all_exce[i])
        return all_exce
# 打開Exce文件
def open_exce(name):
    fh = xlrd.open_workbook(name)
    return fh
# 獲取exce文件下的所有sheet
def get_sheet(fh):
    sheets = fh.sheets()
    return sheets
# 獲取sheet下有多少行數(shù)據(jù)
def get_sheetrow_num(sheet):
    return sheet.nrows
# 獲取sheet下的數(shù)據(jù)
def get_sheet_data(sheet, row,biao_tou_num):
    for i in range(row):
        if (i <biao_tou_num):
            global biao_tou
            values= sheet.row_values(i)
            biao_tou.append(values)
            continue
        values = sheet.row_values(i)
        all_data1.append(values)
    return all_data1
#獲取表頭數(shù)量
def get_biao_tou_num(exce1,exce2):
    fh = open_exce(exce1)
    fhx= open_exce(exce2)
    sheet_1 = fh.sheet_by_index(0)
    sheet_2=fhx.sheet_by_index(0)
    row_sum_1 = sheet_1.nrows
    row_sum_2 = sheet_2.nrows
    # 獲取第一張sheet表對(duì)象有效行數(shù)
    # 獲取sheet表某一行所有數(shù)據(jù)類型及值
    for i in range(row_sum_1):
        # 獲取sheet表對(duì)象某一行數(shù)據(jù)值
        if (i+1== row_sum_2):
            return i
        #row_0_value = sheet_1.row_values(0)
        row_content_1 = sheet_1.row_values(i)
        row_content_2=  sheet_2.row_values(i)
        if(row_content_1==row_content_2):
            continue
        else:
            return i
        #防止越界
if __name__ == '__main__':
    print("使用本程序只需要把程序放到需要合并表格同目錄下")
    print("如發(fā)現(xiàn)bug，QQ聯(lián)系作者：1916624867")
    print("作者:吃點(diǎn)飯飯\n\n\n")
    all_exce = get_exce()
    # 得到要合并的所有exce表格數(shù)據(jù)
    if (all_exce == 0):
        print("該目錄下無.xlsx文件！請(qǐng)把程序移動(dòng)到要合并的表格同目錄下！")
        end = input("按Enter結(jié)束程序")
        exit()
    if (len(all_exce) ==1 ):
        print("該目錄下只有一個(gè).xlsx文件！無需合并")
        end = input("按Enter結(jié)束程序")
        exit()
    # 表頭數(shù)
    print("自動(dòng)檢測(cè)表頭中......")
    biao_tou_num=get_biao_tou_num(all_exce[0],all_exce[1])
    print("表頭數(shù)為:",biao_tou_num,)
    guess=input("y/n?")
    if(guess=="n"):
        biao_tou_num=input("請(qǐng)輸入表頭數(shù):")
        biao_tou_num=int(biao_tou_num)
    all_data1 = []
    # 用于保存合并的所有行的數(shù)據(jù)
    # 下面開始文件數(shù)據(jù)的獲取
    for exce in all_exce:
        fh = open_exce(exce)
        # 打開文件
        sheets = get_sheet(fh)
        # 獲取文件下的sheet數(shù)量
        for sheet in range(len(sheets)):
            row = get_sheetrow_num(sheets[sheet])
            # 獲取一個(gè)sheet下的所有的數(shù)據(jù)的行數(shù)
            all_data2 = get_sheet_data(sheets[sheet], row,biao_tou_num)
            # 獲取一個(gè)sheet下的所有行的數(shù)據(jù)
    for i in range(biao_tou_num):
        all_data2.insert(i, biao_tou[i])
    # 表頭寫入
    new_name=input("清輸入新表的名稱:")
    # 下面開始文件數(shù)據(jù)的寫入
    new_exce = new_name+".xlsx"
    # 新建的exce文件名字
    fh1 = xlsxwriter.Workbook(new_exce)
    # 新建一個(gè)exce表
    new_sheet = fh1.add_worksheet()
    # 新建一個(gè)sheet表
    for i in range(len(all_data2)):
        for j in range(len(all_data2[i])):
            c = all_data2[i][j]
            new_sheet.write(i, j, c)
    fh1.close()
    # 關(guān)閉該exce表
    print("文件合并成功,請(qǐng)查看"+new_exce+"文件！")
    end=input("按Enter結(jié)束程序")