用python修改haproxy配置文件

  1 # ！/usr/bin/env python
  2 # -*- coding:utf-8 -*-
  3 
  4 def select(arg):
  5     """
  6     要求用户输入域名，可以查看webserver的子作用域
  7     针对haproxy的配置文件
  8     思路：
  9     a.逐行读取文件
 10     b.当我遇到backend + 域名的行的时候，将其子作用域放到一个空列表显示给用户看
 11     c.再当我遇到以backend开头的时候就结束循环就ok了
 12     d.因为这个子作用域是需要显示 的，就要拿出来，所以要特别显示出来，要与众不同，所以我们在这里定义标志位来显示这个与众不同
 13     :param arg:
 14     :return:
 15     """
 16     result = []
 17     flag = False
 18     with open("haproxy.txt", "r", encoding="utf-8") as f:
 19         for line in f :
 20             if line.strip() == "backend %s" % arg:
 21                 flag = True
 22                 continue
 23             if line.strip().startswith("backend"):
 24                 flag  = False
 25             if flag:
 26                 result.append(line)
 27     return result
 28 ret = select("www.oldboy.org")
 29 
 30 for i in ret :
 31     print(i)
 32 #++++++++++++++++++++++++++++
 33 def sele_add(yuming, record):
 34     """
 35     增加：
 36     a.先查看有没有这个域名，如果没有在最后添加就可以了
 37     b.如果有，在检测有没有要和你相同的那个子作用域，
 38     c.如果有，则pass,如果没有就在最后一行添加
 39 
 40     方法一：
 41     a.查询webserver域名是用户要的那个域名
 42     b.把查询到的结果添加新纪录写入到新的文件中
 43     c.在继续把除了这个域名意外的作用域及其他内容写入到新文件中
 44     方法二：
 45     a.就是在下一个新的域名之前加入用户需要加入的新纪录
 46     #也可以理解为先插入那行在写backend
 47     b.在排除是否已经存在
 48     c.当伪代码已经写完的时候，肯定不能实现，所以我们得要修改代码，那么就需要用到标志位 因为只在当前域名后第一个backend
 49     前添加纪录，所以标志位在标记的时候需要 更改为数字
 50     :param arg:
 51     :return:
 52     """
 53 
 54     """
 55     biaozhiwei = False
 56     with open("haproxy.txt", "r", encoding="utf-8") as old, open("haproxy2.txt", "w", encoding="utf-8") as new:
 57         res_list = select(yuming)
 58         res_list.append(" "*8 + record + "\n")
 59         for line in old:
 60             if line.strip() == "backend %s" % yuming:
 61                 biaozhiwei = True#注意
 62                 new.write(line)
 63                 for i in res_list:
 64                     new.write(i )
 65 
 66                 continue
 67             if line.strip().startswith("backend"):
 68                 biaozhiwei = False
 69             if not biaozhiwei:
 70                 new.write(line)
 71     """
 72 #No.2
 73 # 方法二：
 74 # a.就是在下一个新的域名之前加入用户需要加入的新纪录
 75 #也可以理解为先插入那行在写backend
 76 # b.在排除是否已经存在
 77 #----------------
 78 #判断域名是否存在，如果存在再次判断记录是否存在，如果存在，就pass,如果不存在就添加
 79     bzw = 2
 80     result = []
 81     a = 0
 82     with open ("haproxy.txt", "r", encoding="utf-8") as old , open("haproxy2.txt", "w", encoding="utf-8") as new:
 83         if ("backend %s" % yuming ) in old:
 84             print("cunz")
 85 
 86             for line in old:
 87 
 88                 if line.strip() == "backend %s" % yuming:
 89                     bzw = 0
 90                     result.append(line)
 91                     new.write(line)
 92                     continue
 93                 if line.strip().startswith("backend"):
 94                     bzw += 1   #因为我们会遇到很多个backend,所以就自加1 了
 95                     if bzw == 1:
 96                         if (" "*8 + record + "\n") in result:
 97                             pass
 98                         else:
 99                             new.write(" "*8 + record + "\n")
100                 new.write(line)
101         else:
102             print("bucz")
103             with open("haproxy.txt", "r", encoding="utf-8") as old, open("haproxy2.txt", "w", encoding="utf-8") as new:
104                 for line in old:
105                     new.write(line)
106                 new.write("\nbackend %s\n" % yuming)
107                 new.write(" "*8 + record + "\n")
108 
109 sele_add("www.oldboy.orgt","1233446")