import gtake
import modbus
import os

TXT_PATH = ""  # 这个安符串用来存txt的路径


# 打印菜单
def print_menu():
    print("*" * 50)
    print("     吉泰科变频器参数下载软件V0.0")
    print("1.恢复出厂")
    print("2.选择txt")
    print("3.参数下载")
    print("4.系统退出")
    print("*" * 50)


# 恢复出厂
def restore_factory(md):
    try:
        md.write_int(3, 2)  # 恢复出厂，A0-03 = 2
        print("初始化完成！")
    except Exception as ex:
        print(str(ex))


# 文件夹遍历，并输出
def folder_txt_list():
    global TXT_PATH
    folder_path = os.getcwd()  # E:\群晖同步盘\原创开发\28.变频器参数下载20211214
    txt_list = []  # 创建一个空列表，用来存所有txt的文件名
    for x in os.listdir(folder_path):
        # 判断文件结尾是否是 .txt 是的话，就加入到列表
        if x.endswith(".txt"):
            txt_list.append(x)
    # 打印一下，txt_list
    if txt_list:
        i = 0
        print("序号\t文件名")
        for x in txt_list:
            print("%d\t%s" % (i, x))
            i += 1
        # 文件选择
        path_num = input("请输入要下载的txt文件的序号：")
        # 先判断是否是数字
        if path_num.isdigit():
            # 判断了一下是否超过列表的索引范围
            if int(path_num) < len(txt_list):
                TXT_PATH = r"{0}\{1}".format(folder_path, txt_list[int(path_num)])
            else:
                print("输入的序号【超过索引啦】！！！")
        else:
            print("输入的序号【不是数字】！！！")
        print(TXT_PATH)
    else:
        print("当前文件目录下，未找到txt文件！")


# 参数下载
def download(gk820, md):
    for x in gk820.list4:
        if x[0] != "":
            try:
                md.write_int(x[0], x[1])
                print("已写入参数：%s" % x[2])
            except Exception as ex:
                print(str(ex))
                print("写入失败！")
                break
        else:
            print(x[2])


def main():
    md = modbus.PlcModbus()
    gk820 = gtake.Gtake()
    while True:
        # 打印功能菜单
        print_menu()
        # 从键盘获取功能选择
        select_num = input("请输入选择的功能序号：")
        if select_num == "1":
            restore_factory(md)
        elif select_num == "2":
            folder_txt_list()
            gk820.read_txt_2_list(TXT_PATH)
        elif select_num == "3":
            download(gk820, md)
        elif select_num == "4":
            print("系统已退出")
            break
        input("按任意键，继续~~~")
        os.system("cls")


if __name__ == "__main__":
    main()