作业：老板现在给你任务，公司有haproxy配置文件，希望通过python程序可以对ha配置文件进行增删改

# 老板现在给你任务，公司有haproxy配置文件，希望通过python程序可以对ha配置文件进行增删改
#分析：对文件进行增删改，首先需要找到需要修改文件的位置，即必须先把文件读取出来，找到对应
#位置，进行内容的修改，增加和删除；
import json,os  #json模块用于将用户输入的字符串转换为字典

#首先定义fetch函数，同时传入指定参数backend,用来将修改的地方找出来
def fetch(backend):
    flag=False #定义flag，目的是为了判断后面是否有可取的backend
    fetch_list=[]  #定义空列表，用来存储读取的backend

    #以读的形式打开"ha"文件，一行行的读，然后我们将满足我们要求的行存入到列表fetch_list中
    #最后返回装有我们指定backend信息的列表
    with open("ha.txt","r") as obj:
        for line in obj:
            if line.strip()=="backend %s" %(backend):
                flag=True
                continue
            if line.strip().startswith("backend"):
                flag=False
            if flag and line.strip():
                fetch_list.append(line.strip())
    return fetch_list

#定义add1函数，并同时传入参数dict_info ,此字典参数包含我们要传入的server信息
def add1(dict_info):
    backend_title=dict_info.get("backend")  #首先获取要插入的backend的名称
    context_title="backend %s"%(backend_title)  #要插入的backend的整个字段
    record_title=dict_info["record"]
    #要插入的记录信息
    context_record="server %s %s weight %s maxconn %s" %(record_title["server"],record_title["server"],
                                                         record_title["weight"],record_title["maxconn"])
    #将backend获取，并存储在fetch_list列表中
    fetch_list=fetch(backend_title)

     #如果要插入的backend存在
        #1.首先设定2个标志：flag和has_write
            #flag:用于找到要添加的backend下的内容
            #has_write:用于判断fetch_list中的内容是否重新写入到了文件
        #2.遍历源文件ha：
            #1首先逐行读取，并同时进行判断，
                # 1.判断backend是否已经找到，先把backend写入文件，然后找到后将flag设置为True，最后跳出本次循环
                # 2.判断找到的backend里面的record是否为空，若为空将flag设置为False
                # 3.对flag进行判断
                    #1.将flag为True的record从fetch_list写入到文件中，在这里面同时要进行是否已经写入的判断，
                    # 即定义has_write的作用
                    #2.若flag为False，就直接写入新文件，即将非列表里的内容直接写入
    if fetch_list:
        flag=False
        has_write=False
        with open("ha.txt","r") as read_obj,open("ha.new","w") as write_obj:
            for line in read_obj:
                if line.strip()==context_title:
                    write_obj.write("
"+line)
                    flag=True
                    continue
                if flag and line.startswith("backend"):
                    flag=False
                if flag:
                    for new_line in fetch_list:
                        if not has_write:
                            temp="%s%s" % (" "*8,new_line)
                            write_obj.write(temp)
                            has_write=True
                else:
                    write_obj.write(line)
    else:
         #如果要插入的backend不存在
        #分2部分写入：
            # 1部分.从ha文件里面直接读取，并同时将读取的内容写入新的文件ha.new
            # 2部分.将新的backend和context信息，直接写到新的文件ha.new的文件尾部
            with open("ha.txt","r") as read_obj,open("ha.new","w") as write_obj:
                for line in read_obj:
                    write_obj.write(line)
                write_obj.write("
"+context_title+"
")
                temp=" "*8+context_record+"
"
                write_obj.write(temp)
    #将原文件ha改名备用文件为ha.bak，即将ha下线
    #将新文件ha.new改名为线上文件ha.txt，即将ha.new上线
    os.rename("ha.txt","ha.bak")
    os.rename("ha.new","ha.txt")

#删除函数
def delete(dict_info):
    #1、首先获取用户的记录并以列表的形势存储，以便于我们调用
    del_backend=dict_info["backend"]
    del_record=dict_info["record"]

    context_title="backend %s" %(del_backend)
    context_record="server %s %s weight %s maxconn %s" %(del_record["server"],del_record["server"],
                                                         del_record["weight"],del_record["maxconn"])
    #调用fetch函数，得到包含backend内容的列表
    fetch_list=fetch(del_backend)
    #判断1、若列表为空格，直接返回，即没有可以删除的对象
    if not fetch_list:
        return
    else:
        #判断1、若用户输入的server信息不在列表里面，直接返回
        if context_record not in fetch_list:
            print("服务器信息不存在")
            return
        else:
            fetch_list.remove(context_record)  #若输入的server信息在列表中，则将此信息从列表中删除

        #逐行从ha文件中读取，并同时进行判断
        #核心思想：
            #首先边读取、边判断、边写入，
            # 判断其实就是我们自己输入的backend，找到后将列表fetch_list信息写入到新的文件里面。
            # 其余的非列表的内容直接从源文件ha写到新文件ha.new即可
        with open("ha.txt","r") as read_obj,open("ha.new","w") as write_obj:
            flag=False
            has_write=False
            for line in read_obj:
                if line.strip()==context_title:
                    write_obj.write(line)
                    flag=True
                    continue
                if flag and line.startswith("backend"):
                    flag=False

                if flag:
                    if not has_write:
                        print(fetch_list)
                        for new_line in fetch_list:
                            temp="%s%s
" %(" "*8,new_line)
                            write_obj.write(temp)
                        has_write=True
                else:
                    write_obj.write(line)
    #将原文件ha改名备用文件为ha.bak，即将ha下线
    #将新文件ha.new改名为线上文件ha，即将ha.new上线
    os.rename("ha.txt","ha.bak")
    os.rename("ha.new","ha.txt")


if __name__=="__main__":
    #获取用户的操作选择
    print("1、获取记录
2、添加记录
3、删除记录
")
    select_num=input("请输入你的选择：")
    #根据用户的选择，进行函数调用
    if select_num=="1":
        backend=input("请输入backend:")
        fetch_list=fetch(backend)
        for i in fetch_list:
            print(i)
    else:
        print("请按如下例子输入：")
        print('{"backend":"ttt.oldboy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}')
        data_str=input("请输入record:")
        data_dict=json.loads(data_str)
        if select_num=="2":
            add1(data_dict)
        elif select_num=="3":
            delete(data_dict)
        else:
            print("输入错误")

新手，编码有不成熟的地方，还望各位同道多多指正！