#作业名称：配置文件的增删改查
#作者：Marvin
#作业需求：
配置文件增删改查
#本次完成作业需求：
配置文件增删改查
#测试：
##运行环境：python3.4
第一步：
运行程序
第二步：
选择1，进行查询
第三步：
输入查询内容
{'backend':'www.oldboy.org','record':{'server':'100.1.7.10','weight':'20','maxconn':'3000'}}
第四步：
选择2，进行添加
第五步：
输入添加内容
{'backend':'www.oldboy.org','record':{'server':'100.1.7.1','weight':'20','maxconn':'3000'}}
第六步：
选择3，进行删除
第七步：
输入删除内容
{'backend':'www.oldboy.org','record':{'server':'100.1.7.1','weight':'20','maxconn':'3000'}}
第八步：
选择4，进行修改
第九步：
输入要修改的新数据和修改后的新数据
[{'backend':'www.oldboy.org','record':{'server':'100.1.7.10','weight':'20','maxconn':'3000'}},{'backend':'www.oldboy.org','record':{'server':'100.1.7.1','weight':'20','maxconn':'3000'}}]
第十步：
选择5，进行退出
#备注：
1、输入的格式固定
import os
def fetch(data):
backend_data="backend %s"%(data)
# print(backend_data)
searched_data = []
with open("haproxy配置文件.txt","r")as f:
tag=False
for line in f:
#print(line)
if line.strip() ==backend_data:
#print("1")
tag=True
continue
if line.startswith("frontend"):
#print("2")
tag=False
break
if tag and line:
#print("3")
searched_data.append(line.strip())
# for item in searched_data:
# print(item)
return searched_data
def add(data):
backend_title="backend %s"%(data["backend"])
record="server %s %s weight %s maxconn %s" %(data["record"]["server"],\
data["record"]["server"],\
data["record"]["weight"],\
data["record"]["maxconn"])
record_list=fetch(data["backend"])
# 如果要插入的backend存在
# 1.首先设定2个标志：flag和has_write
# flag:用于找到要添加的backend下的内容
# has_write:用于判断fetch_list中的内容是否重新写入到了文件
# 2.遍历源文件ha：
# 1首先逐行读取，并同时进行判断，
# 1.判断backend是否已经找到，先把backend写入文件，然后找到后将flag设置为True，最后跳出本次循环
# 2.判断找到的backend里面的record是否为空，若为空将flag设置为False
# 3.对flag进行判断
# 1.将flag为True的record从fetch_list写入到文件中，在这里面同时要进行是否已经写入的判断，即定义has_write的作用
# 2.若flag为False，就直接写入新文件，即将非列表里的内容直接写入
if record_list:
flag=False
has_write=False
with open("haproxy配置文件.txt","r")as f_read,\
open("haproxy配置文件_new.txt","w")as f_write:
for line in f_read:
if line.strip()==backend_title:
f_write.write(line)
flag=True
continue
if flag and line.startswith("frontend"):
flag=False
if flag:
if not has_write:
f_write.write("%s%s\n"%(" "*8,record))
f_write.write(line)
has_write=True
print("1")
else:
f_write.write(line)
print("2")
# print("1")
# f_write.write("\n")
else:
f_write.write(line)
#如果要插入的backend不存在
#分2部分写入：
# 1部分.从ha文件里面直接读取，并同时将读取的内容写入新的文件ha.new
# 2部分.将新的backend和context信息，直接写到新的文件ha.new的文件尾部
else:
with open("haproxy配置文件.txt", "r")as f_read, \
open("haproxy配置文件_new.txt", "w")as f_write:
for line in f_read:
f_write.write(line)
f_write.write(backend_title+"/n")
f_write.write(record+"/n")
os.rename("haproxy配置文件.txt","haproxy配置文件bak.txt")
os.rename("haproxy配置文件_new.txt", "haproxy配置文件.txt")
os.remove("haproxy配置文件bak.txt")
def remove(data):
backend_data = "backend %s" % (data["backend"])
record = "server %s %s weight %s maxconn %s" % (data["record"]["server"], \
data["record"]["server"], \
data["record"]["weight"], \
data["record"]["maxconn"])
record_list = fetch(data["backend"])
if not record or record not in record_list:
print("\033[33:1m无此条记录\033[0m")
return
else:
record_list.insert(0,backend_data)
record_list.remove(record)
with open("haproxy配置文件.txt","r")as f_read,\
open("haproxy配置文件_new.txt","w")as f_write:
tag =False
has_writed=False
for r_line in f_read:
if r_line.strip()==backend_data:
tag=True
continue
if tag and r_line.startswith("frontend"):
tag=False
if not tag:
f_write.write(r_line)
else:
if not has_writed:
for new_line in record_list:
if new_line.startswith("backend"):
f_write.write(new_line+"\n")
else:
f_write.write("%s%s\n" %(" "*8,new_line))
has_writed=True
os.rename("haproxy配置文件.txt","haproxy配置文件bak.txt")
os.rename("haproxy配置文件_new.txt", "haproxy配置文件.txt")
os.remove("haproxy配置文件bak.txt")
def fix(data):
backend_title = "backend %s" % (data[0]["backend"])
old= "server %s %s weight %s maxconn %s" % (data[0]["record"]["server"], \
data[0]["record"]["server"], \
data[0]["record"]["weight"], \
data[0]["record"]["maxconn"])
new = "server %s %s weight %s maxconn %s" % (data[1]["record"]["server"], \
data[1]["record"]["server"], \
data[1]["record"]["weight"], \
data[1]["record"]["maxconn"])
fetch_list = fetch(data[0]["backend"])
if not fetch_list or old not in fetch_list:
print(
"\033[33:1m无此条记录\033[0m"
)
else:
remove(data[0])
add(data[1])
if __name__=='__main__':
msg='''
1：查询
2：添加
3：删除
4：修改
5：退出
'''
menu_dic={
"1":fetch,
"2":add,
"3":remove,
"4":fix,
"5":exit
}
while True:
print(msg)
choice=input("操作>>:")
if len(choice)==0 or choice not in menu_dic:continue
if choice=="5":break
data=input("数据>>:").strip()
if choice !="1":
data=eval(data)
menu_dic[choice](data)
