from TSExcel2GPG.TSExcelConverMod1 import *
from TSExcel2GPG.TSExcelConverMod2 import *

MOD1 = 0
MOD2 = 1


def detect_excel_type(file_name):
    wb = load_workbook(file_name)
    sheet_names = wb.worksheets
    for sheet_name in sheet_names:
        if sheet_name.title.lower() == 'action' or sheet_name.title.lower() == 'main1':
            return MOD1
    return MOD2



if __name__ == "__main__":
    import sys
    if len(sys.argv)<3:
        print("input arg count error")
    start_time = time.perf_counter()
    ret = detect_excel_type(sys.argv[1])
    if ret == MOD1:
        wb = ExcelConverMod1(sys.argv[1],sys.argv[2])
        if wb.Ret:
            print("Load Successed")
    elif ret == MOD2:
        wb = ExcelConverMod2(sys.argv[1],sys.argv[2])
        if wb.Ret:
            print("Load Successed")
        else:
            print(wb.Ret)
    else:
        print("unknow excel type")