Python小程序之动态修改Haproxy配置文件

需求如下：

　　1、动态的查询添加删除haproxy节点信息

　　2、程序功能：add（添加）、Del（删除）、Query（查询）

　　3、添加时实例字符串为：  {'backend': 'www.oldboy.com','record': {'server': '100.1.7.9','weight': 20,'maxconn': 30}}

代码：

 # Author:Lee Sir
 import sys

 haproxyconf = r"E:\Python练习脚本\day3\haproxy.txt"
 haproxyconf_temp = r"E:\Python练习脚本\day3\haproxy_temp.txt"

 def openconf(file,mode):
     fd = open(file,'%s' % mode)
     return fd

 def closeconf(fd):
     fd.close()

 def user_choice():
     choice = ['Query','Add','Delete','Update']
     for index,key in enumerate(choice):
         print(index,key)
     while True:
         userchoice = input('Please number for your Choice: ')
         if userchoice.isdigit() and int(userchoice) >= 0 and int(userchoice) < 3:
             return userchoice
         elif userchoice == 'q' or userchoice == 'Q':
             sys.exit('Bye Bye')
         else:
             print('Input Error,Please try again')

 def listnode():
     conf = openconf(haproxyconf,'r')
     key = input('Please Input Your Query Node Name: ')
     count = 0
     for line in conf:
         if key in line and line.strip().startswith('backend'):
             result = True
             nodenumber = 1
             while result:
                 nodeinfo = conf.readline()
                 if nodeinfo.strip().startswith('server'):
                     print('节点 %s:%s ' % (nodenumber,nodeinfo.strip()))
                     count += 1
                     nodenumber += 1
                 else:
                     result = False
     if count == 0:
         print('\033[31;1mNo Node for %s\033[0m' % key)
     closeconf(conf)

 def delenode():
     conf = openconf(haproxyconf,'r')
     key = input('Please input Delete Node: ')
     conf_list = []
     for line in conf:
         if line.strip().startswith('backend'):
             if key in line:
                 conf_list.append(line)
                 while True:
                     nodeinfo = conf.readline()
                     if nodeinfo.strip().startswith('server'):
                         conf_list.append(nodeinfo)
                     else:
                         break
     else:
         if len(conf_list) > 0:
             old_conf = openconf(haproxyconf,'r')
             write_conf = openconf(haproxyconf_temp,'w')
             for info in old_conf:
                 if info in conf_list:
                     continue
                 else:
                     write_conf.write(info)
             else:
                 closeconf(old_conf)
                 closeconf(write_conf)
                 print('Delete Node:%s Successful!'  % key )
         else:
             sys.exit('No Node info for %s' % key)

 def addnode():
     conf = openconf(haproxyconf_temp,'r')
 #    nodeinfo = '''{'backend': 'www.oldboy.com','record': {'server': '100.1.7.9','weight': 20,'maxconn': 30}}'''
     nodeinfo = input('Please input Your Add node info:')
     nodeinfo = eval(nodeinfo)
     if isinstance(nodeinfo,dict):
         backend = nodeinfo['backend']
         for line in conf:
             if line.strip().startswith('backend'):
                 if backend in line:
                     sys.exit('The node are exist!')
         else:
             add_wirte = openconf(haproxyconf_temp,'a')
             backend = nodeinfo['backend']
             record = nodeinfo['record']
             server = record['server']
             weight = record['weight']
             maxconn = record['maxconn']
             node_info = '''\nbackend %s\n\t\tserver %s %s weight %s maxconn %s
             ''' %  (backend,server,server,weight,maxconn)
             add_wirte.write(node_info)
             print('The Node:%s Add Successful!' % backend )
     else:
         sys.exit('Input Error,bye bye!')

 def main():
     userchoice = user_choice()
     if userchoice == '':
         listnode()
     elif userchoice == '':
         addnode()
     elif userchoice == '':
         delenode()

 if __name__ == '__main__':
     main()