用python修改haproxy配置文件

需求：

    当用户输入域名的时候，显示出来下面的记录

    当用户需要输入添加纪录的时候，添加到你需要的那个域名下面

# ！/usr/bin/env python
# -*- coding:utf-8 -*-

def select(arg):
    """
    要求用户输入域名，可以查看webserver的子作用域
    针对haproxy的配置文件
    思路：
    a.逐行读取文件
    b.当我遇到backend + 域名的行的时候，将其子作用域放到一个空列表显示给用户看
    c.再当我遇到以backend开头的时候就结束循环就ok了
    d.因为这个子作用域是需要显示 的，就要拿出来，所以要特别显示出来，要与众不同，所以我们在这里定义标志位来显示这个与众不同
    :param arg:
    :return:
    """
    result = []
    flag = False
    with open("haproxy.txt", "r", encoding="utf-8") as f:
        for line in f :
            if line.strip() == "backend %s" % arg:
                flag = True
                continue
            if line.strip().startswith("backend"):
                flag  = False
            if flag:
                result.append(line)
    return result
ret = select("www.oldboy.org")

for i in ret :
    print(i)
#++++++++++++++++++++++++++++
def sele_add(yuming, record):
    """
    增加：
    a.先查看有没有这个域名，如果没有在最后添加就可以了
    b.如果有，在检测有没有要和你相同的那个子作用域，
    c.如果有，则pass,如果没有就在最后一行添加

    方法一：
    a.查询webserver域名是用户要的那个域名
    b.把查询到的结果添加新纪录写入到新的文件中
    c.在继续把除了这个域名意外的作用域及其他内容写入到新文件中
    方法二：
    a.就是在下一个新的域名之前加入用户需要加入的新纪录
    #也可以理解为先插入那行在写backend
    b.在排除是否已经存在
    c.当伪代码已经写完的时候，肯定不能实现，所以我们得要修改代码，那么就需要用到标志位 因为只在当前域名后第一个backend
    前添加纪录，所以标志位在标记的时候需要 更改为数字
    :param arg:
    :return:
    """

    """
    biaozhiwei = False
    with open("haproxy.txt", "r", encoding="utf-8") as old, open("haproxy2.txt", "w", encoding="utf-8") as new:
        res_list = select(yuming)
        res_list.append(" "*8 + record + "\n")
        for line in old:
            if line.strip() == "backend %s" % yuming:
                biaozhiwei = True#注意
                new.write(line)
                for i in res_list:
                    new.write(i )

                continue
            if line.strip().startswith("backend"):
                biaozhiwei = False
            if not biaozhiwei:
                new.write(line)
    """
#No.2
# 方法二：
# a.就是在下一个新的域名之前加入用户需要加入的新纪录
#也可以理解为先插入那行在写backend
# b.在排除是否已经存在
#----------------
#判断域名是否存在，如果存在再次判断记录是否存在，如果存在，就pass,如果不存在就添加
    bzw = 2
    result = []
    a = 0
    with open ("haproxy.txt", "r", encoding="utf-8") as old , open("haproxy2.txt", "w", encoding="utf-8") as new:
        if ("backend %s" % yuming ) in old:
            print("cunz")

            for line in old:

                if line.strip() == "backend %s" % yuming:
                    bzw = 0
                    result.append(line)
                    new.write(line)
                    continue
                if line.strip().startswith("backend"):
                    bzw += 1   #因为我们会遇到很多个backend,所以就自加1 了
                    if bzw == 1:
                        if (" "*8 + record + "\n") in result:
                            pass
                        else:
                            new.write(" "*8 + record + "\n")
                new.write(line)
        else:
            print("bucz")
            with open("haproxy.txt", "r", encoding="utf-8") as old, open("haproxy2.txt", "w", encoding="utf-8") as new:
                for line in old:
                    new.write(line)
                new.write("\nbackend %s\n" % yuming)
                new.write(" "*8 + record + "\n")

sele_add("www.oldboy.orgt","1233446")

配置文件

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
        server 100.1.7.92 100.1.7.93 weight 20 maxconn 3000
        server 100.1.7.94 100.1.7.93 weight 20 maxconn 3000

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