haproxy配置文件简单管理

版本：python3
功能：对haproxy配置文件进行简单的查询、添加以及删除功能
操作流程：
1.根据提示选择相应的选项
2.进入所选项后，根据提示写入相应的参数
3.查询功能会返回查询结果，添加、删除以及回退功能无回显

流程图：

 global
         log 127.0.0.1 local2
         daemon
         maxconn 256
         log 127.0.0.1 local2 info
 defaults
         log global
         mode http
         timeout connect 5000ms
         timeout client 50000ms
         timeout server 50000ms
         option  dontlognull

 listen stats :8888
         stats enable
         stats uri       /admin
         stats auth      admin:1234

 frontend oldboy.org
         bind 0.0.0.0:80
         option httplog
         option httpclose
         option  forwardfor
         log global
         acl www hdr_reg(host) -i www.oldboy.org
         use_backend www.oldboy.org if www

 backend www.oldboy.org
         server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000

 backend buy.oldboy.org
         server 100.1.7.90 100.1.7.90 weight 20 maxconn 3000

 backend nb.oldboy.org
         server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000

 backend ttt.oldboy.org
         server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000

 backend www.oldboy.com
 server  192.168.1.1 192.168.1.1 weight 30 maxcoon 3000

haproxy配置示例文件

 import os
 import os.path
 import time
 def query(back_name):#定义查询函数
     flage=False    #[通过用户给定的backend，找到backend所在行，在寻找下一个以backend开头的行，将两个backend中间的内容添加进列表中]
     query_list=[]
     with open('ha','r') as ha:  #打开配置文件
         for line in ha:          #获取配置文件的内容
             if line.strip()=='backend %s'%(back_name): #获取要查询的backend所在行的内容
                 flage=True#善用标志位
                 continue
             if line.strip().startswith('backend'):#判断是否行开头为backend
                 flage=False
             if flage and line.strip(): #判断flage是否为真
                 query_list.append(line.strip())#将符合的内容添加单独进列表中
     return query_list

 def add_backend(back_name,ip,weight,maxcoon):#定义添加函数
     current_title='backend %s'%back_name  #获得用户输入backend所在行内容
     current_record='server  %s %s weight %s maxcoon %s'%(ip,ip,weight,maxcoon)  #获得用户所要添加的server
     fetch_list=query(back_name)     #获取指定的backend下的记录
     if fetch_list:  #如果添加的backend存在
         if current_record in fetch_list: #如果添加的子项存在backend中，则不进行操作
             pass      # '''【主要思路：通过判断，将需要添加的backend下的内容和原来已经存在的内容存入一个列表，将需要添加的
                       # backend之前的内容写入新文件,再将backend和存入列表中需要添加的sever写入，最后添加其余内容】'''
         else:
             fetch_list.append(current_record)#如果不存在backend中，添加进指定的backend中
         flage=False  #定义标示位，用于判断前后backend之间需要写入的内容
         has_write=False##定义标示位，用于判断需要添加的内容是否已存在
         with open('ha','r') as ha,open('ha.new','w') as new_ha:  #打开ha配置文件，创建一个新的文件
             for line in ha: #迭代配置文件
                 if line.strip() == current_title: #判断文件中是否有用户输入的backend
                     flage=True
                     new_ha.write(line)#把backend写入到新配置文件内
                     continue
                 if flage and line.strip().startswith('backend'):
                     flage=False
                 if flage :
                     if not has_write: #通过标示位判断
                         for line in fetch_list:#获得需要添加的backend下的所以server
                             temp="%s%s\n" %(" "*8,line)
                             new_ha.write(temp) #写入新的文件中
                             has_write = True
                 else:
                     new_ha.write(line)#将其余内容写入到新的文件中
     else:#如果用户输入的backend不存在配置文件中
         with open('ha','r') as ha,open('ha.new','w') as new_ha:
             for line in ha:
                 new_ha.write(line)#将配置文件的所以内容写入新的文件中
             new_ha.write('\n'+current_title)#将用户输入的backend写入新的文件中
             tem=''*8+current_record+'\n'
             new_ha.write('\n'+tem)#将用户输入的sever写入文件中
     os.rename('ha','ha_bak')
     os.rename('ha.new','ha')

 def quit_time():#查询完成后退出
     for i in [3,2,1]:
         print('\033[32;1m查询完毕，正在返回主菜单.....\033[1m',i)
         time.sleep(1)

 def del_backend(back_name,ip,weight,maxcoon):  #与添加类似，通过将需要的server存入一个列表中，最后将其余内容和列表中的内容写入新的文件中
     current_title='backend %s'%back_name     #通过文件重命名的方式实现功能
     current_record='server  %s %s weight %s maxcoon %s'%(ip,ip,weight,maxcoon)
     fetch_list=query(back_name)     #获取指定的backend下的记录
     if fetch_list:  #如果添加的backend存在
         if current_record in fetch_list: #如果添加的子项存在backend中
                 fetch_list.remove(current_record)
         else:
             pass
         flage=False
         has_write=False
         with open('ha','r') as ha,open('ha.new','w') as new_ha:
             for line in ha:
                 if line.strip() == current_title:
                     flage=True
                     new_ha.write(line)#把backend写入到新配置文件内
                     continue
                 if flage and line.strip().startswith('backend'):
                     flage=False

                 if flage :
                     if not has_write:
                         for line in fetch_list:
                             temp="%s%s\n" %(" "*8,line)
                             new_ha.write(temp)
                             has_write = True
                 else:
                     new_ha.write(line)
     else:
         with open('ha','r') as ha,open('ha.new','w') as new_ha:
             for line in ha:
                 new_ha.write(line)
             new_ha.write('\n'+current_title)
             tem=''*8+current_record+'\n'
             new_ha.write('\n'+tem)
     if os.path.isfile('ha_bak'): #文件重命名需要判断文件是否存在，当第一次时，备份文件不存在会报错
         os.remove('ha_bak')
     os.rename('ha','ha_bak')
     os.rename('ha.new','ha')

 def fallback(): #回退代码，将备份文件和当前配置文件重命名的方式实现功能
     os.rename('ha','ha_new')
     os.rename('ha_bak','ha')

 while True:
     print('\033[34;1m#######################################   \033[1m') #打印选项
     user_enter=input('''\033[32;1m 1.查询后端backend
  2.添加后端backend
  3.删除后端backend
  4.回滚至前一版本
  5.退出\033[1m
  请输入你要选择的操作：''' )
     if user_enter =='':
         back_name=input('\033[32;1m请输入需要查询的backend名称，example:www.oldboy.org：\033[1m')
         result=query(back_name)  #调用查询函数
         if result:
             print('\033[33;1m查询结果如下：\033[1m')
             for i in result:
                 print(i)
             quit_time()
             continue
         else:
             print('\033[34;1m查询错误，backend不存在，结果为空！\033[1m')
     elif user_enter=='':
          back_name=input('\033[32;1m请输入需要查询的backend名称，example:www.oldboy.org：\033[1m')
          ip=input('\033[33;1m请输入要添加的serverIP地址，example192.168.1.1：\033[1m')
          weight=input('\033[34;1m请输入该server的weight：\033[1m')
          maxcoon=input('\033[35;1m请输入该server的maxcoon：\033[1m')
          add_backend(back_name,ip,weight,maxcoon)
          print('添加完成')
          quit_time()
     elif user_enter=='':
          back_name=input('\033[32;1m请输入需要删除的backend名称，example:www.oldboy.org：\033[1m')
          ip=input('\033[33;1m请输入要添加的serverIP地址，example192.168.1.1：\033[1m')
          weight=input('\033[34;1m请输入该server的weight：\033[1m')
          maxcoon=input('\033[35;1m请输入该server的maxcoon：\033[1m')
          del_backend(back_name,ip,weight,maxcoon)
          print('删除完成')
          quit_time()
     elif user_enter=='':
          try:
            fallback()
          except FileNotFoundError: #回退时，备份文件不存在报错
              print('\033[31;1m回退失败，当前为最新版本，无法回退！\033[1m')
          except FileExistsError:#用户重复回退
              print('已经是最新版本')
              quit_time()
     elif user_enter=='':
           quit_time()
           break
     else:#非法输入时
         print('\033[31;1m您输入的选项错误，请重新选择！\033[1m')

程序代码