用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