day2-python

# Author:daemon chai

f=open("yesterday2","w+",encoding="utf-8")
f.write("------add-----------\n")
f.write("------df-----------\n")
f.write("------dfasfasdf---------\n")

print(f.tell())
print(f.seek(0))
print(f.readline())

f.write("------czk-----------")
f.close()

with open("record", "w", encoding="utf-8") as f:
    f.write("")

# Author:daemon chai
old_index = -1
new_dict={}
other_list=[]
#将配置文件读取出来，需要修改的信息存储在字典里面
with open("haproxy","r",encoding="utf-8") as f :
    for index,line in enumerate(f) :
        #存储配置文件中节点的bakend
        if line.strip().startswith("bakend") :
            old_index = index + 1
            user_name=line.strip().split()[1].split(‘.‘)[1]
            new_dict[user_name]={}
            new_dict[user_name]["bakend"]=line.strip().split()[1]
            name=line.strip().split()[1]
            # print(new_list[user_name])
        #存储节点信息的record
        elif old_index == index :
            new_dict[user_name]["record"]={}
            new_dict[user_name]["record"]["server"]=line.strip().split()[1]
            new_dict[user_name]["record"]["weight"] = line.strip().split()[3]
            new_dict[user_name]["record"]["maxconn"] = line.strip().split()[5]
        #存储其他不需要修改的信息
        else:
            other_list.append(line.strip())

while True:
    print("请选择需要的功能".center(50,"="))
    #打印已有节点信息
    print("已有的节点：")
    for dict_one in new_dict:
        print(dict_one)
#提供功能选择界面
    print("现有功能：\nc、查询；\na、增加；\nr、修改；\nd、删除；\nq、退出。")
    choice = input("请选择你需要的功能：")
    #查询操作
    if choice ==‘c‘ :
        while True:
            choice_query = input("请输入查询的节点,退出（q）：")
            if choice_query == "q":
                break
            elif choice_query in new_dict:
                print("节点名称：", new_dict[choice_query]["bakend"])
                print("ip地址：", new_dict[choice_query]["record"]["server"])
                print("权重：", new_dict[choice_query]["record"]["weight"])
                print("连接数：", new_dict[choice_query]["record"]["maxconn"])
            else:
                print("输入错误，请重新输入")
    #添加操作
    elif choice==‘a‘:
        while True:
            add_simple_name = input("请输出节点简称")
            add_name = input("请输出节点名称")
            add_ip = input("请输出节点ip")
            add_weight = input("请输出节点权重")
            add_maxconn = input("请输出节点最大连接数")
            new_dict[add_simple_name]={}
            new_dict[add_simple_name]["bakend"] = add_name
            new_dict[add_simple_name]["record"] = {}
            new_dict[add_simple_name]["record"]["server"] = add_ip
            new_dict[add_simple_name]["record"]["weight"] = add_weight
            new_dict[add_simple_name]["record"]["maxconn"] = add_maxconn
            print("添加成功",new_dict[add_simple_name])
            break
    #修改节点信息操作
    elif choice == ‘r‘:
        while True:
            return_name = input("你想修改哪个节点啊？退出（q）")
            if return_name == "q":
                break
            elif return_name in new_dict:
                new_dict[return_name]["bakend"] = input("请输出修改后节点名称(例如www.user.org)")
                new_dict[return_name]["record"]["server"] = input("请输出修改后节点ip")
                new_dict[return_name]["record"]["weight"] = input("请输出修改后节点权重")
                new_dict[return_name]["record"]["maxconn"] = input("请输出修改后节点最大连接数")
                print("保存退出后修改才会生效")
                break
    #删除节点操作
    elif choice == ‘d‘:
        while True:
            delete_name = input("你想删除那个节点呢,退出（q）？")
            if delete_name == "q":
                break
            elif delete_name in new_dict:
                del new_dict[delete_name]
                print("删除成功！")
            else:
                print("没有这个节点，重新输入！")
    #退出操作，将修改后的节点信息存储进文件
    elif choice == ‘q‘:
        with open("haproxy", "w") as f1 :
            f1.write("")
        for i in other_list:
            with open("haproxy", "a") as f2 :
                f2.write("%s\n"%i)
        for i in new_dict:
            with open("haproxy", "a") as f3 :
                f3.write("bakend %s\n\tserver %s weight %s maxconn %s\n\n"                         %(new_dict[i]["bakend"], new_dict[i]["record"]["server"], new_dict[i]["record"]["weight"],                           new_dict[i]["record"]["maxconn"]))
        print("退出程序成功")
        break
    else:
        print("输入错误，请重新输入\n")