对haproxy.conf文件的增删改查

# 增删改查haproxy
def add_conf(add_domain,add_record):
    """
    增加记录
    :param add_domain:
    :param add_record:
    :return:
    """
    with open("haproxy.conf", encoding="utf8") as file, 
    open("add_haproxy.conf", 'w', encoding="utf8") as afile:
        for line in file:
            if line.strip().startswith("backend") and add_domain in line:
                line = "".join([line,"
","		",add_record.strip(),'
'])
            afile.write(line)
    print("增加记录成功")
def del_conf(del_domain,num):
    """
    输入删除的域名和序号，删除源文件对应的行
    :param del_domain:
    :param num:
    :return:
    """
    with open("haproxy.conf", encoding="utf8") as file, 
    open("del_haproxy.conf", 'w', encoding="utf8") as dfile:
        l = find_conf(del_domain)
        print(l[(num-1)])
        for line in file:
            if line.strip() == l[num-1]:
               continue
            else:
                dfile.write(line)
    print("删除成功")
def change_conf(ch_domain, ch_num, ch_record):
    with open("haproxy.conf", encoding="utf8") as file, 
    open("ch_haproxy.conf", 'w', encoding="utf8") as chfile:
        l = find_conf(ch_domain)
        for line in file:
            if line.strip() == l[ch_num - 1]:
                line = "		"+ch_record+"
"
            chfile.write(line)
    print("修改成功")
def find_conf(domain):
    """
    查询并打印server信息
    :param domain: 查询域名
    :return: 包含server信息的列表L
    """
    flag = False
    l = []
    with open("haproxy.conf", encoding="utf8") as find_file:
        for lines in find_file:
            if lines.strip().startswith("backend") and lines.strip().endswith(domain):
                flag = True
                continue
            if lines.strip().startswith("backend") and flag == True:
                flag = False
            if flag == True:
                l.append(lines.strip())
    return l
def menu():
    print( """-------------------
    1.查询
    2.增加
    3.删除
    4.修改
    ------------------------""")
def main():
    menu()
    choice=int(input("请输入操作序号"))
    # 查询
    if choice == 1:
        name = input("请输入要查询的域名")
        server_list = find_conf(name) # 拿到含有server信息的列表
        count =1
        for i in server_list:
            print(count,i)
            count += 1
    # 增加
    if choice == 2:
        add_domain = input("请输入要增加的域名")
        add_record = input("请输入要增加的record")
        add_conf(add_domain,add_record)
    # 删除
    if choice == 3:
        del_domain = input("请输入要删除的域名")
        del_list = find_conf(del_domain)
        count = 1
        for i in del_list:
            print(count, i)
            count += 1
        del_num = int(input("请输入要删除的序号"))
        del_conf(del_domain, del_num)
    # 修改
    if choice == 4:
        ch_domain = input("请输入要修改的域名")
        ch_list = find_conf(ch_domain)
        count = 1
        for i in ch_list:
            print(count, i)
            count += 1
        ch_num = int(input("请输入需要修改的序号"))
        ch_record = input("请输入需要修改record")
        change_conf(ch_domain, ch_num, ch_record)
while True:
    main()