作业 1. HAproxy配置文件操作
1. 根据用户输入输出对应的backend下的server信息
2. 可添加backend 和sever信息
3. 可修改backend 和sever信息
4. 可删除backend 和sever信息
5. 操作配置文件前进行备份
6 添加server信息时，如果ip已经存在则修改;如果backend不存在则创建；若信息与已有信息重复则不操作
照旧,上一下流程图:
详细代码:
#！usr/bin/env python
#-*-coding:utf-8-*-
# Author calmyan
import time
time_format='%y-%m-%d_%H-%M-%S'#格式化时间输出格式
times=time.strftime(time_format)#将时间转为字符串 def filebak():#自动备份配置文件函数
haproxy_bak='HA_bak_'+times
with open('HAproxy','r+',encoding='utf-8') as ha_file,\
open(haproxy_bak,'w',encoding='utf-8') as new_file:
for line in ha_file:
new_file.write(line)
return haproxy_bak#返回备份文件名 def file_query():#查询server信息函数
file_dict={}#创建一个空字典
with open('HAproxy','r+',encoding='utf-8') as ha_file:
for line in ha_file:
line=line.strip()#去除空格与回车
try:
if line.startswith('backend'):#判断以backend开头的行
key_1=line.split()[1]#以空格进行分割赋于变量
file_dict[key_1]={}#设置字典格式
file_dict_key1=file_dict[key_1]#赋于变量
if line.startswith('server'):
b={'server':[line.split()[1],line.split()[2]],'weight':line.split()[4],'maxconn':line.split()[6]}#
file_dict_key1.update(b)#更新字典
except Exception as e:#出错可返回操作
print("\033[31;1mHAproxy文件出错,请仔细检查！\033[0m")
exit()
return file_dict#返回字典 所有backend的信息 def out():#返回退出函数
qu=input("请按提示操作！按\033[31;1mY/y\033[0m退出修改器!其它键返回！")
if qu=='Y'or qu=='y':
print("您已经退出修改器！" )
exit()#退出程序
else :
pass def show():#提示示例
arg_list="{'backend':'www.oldboy3.org','record':{'server':['110.110.1.1','110.110.1.1'],'weight':20,'maxconn':344}}"
print(arg_list) def add():#添加backend信息函数
while True:
print('\033[43;1m添加backend信息！输入格式示例:\033[0m'.center(40,'-'))
#arg_list="{'backend':'www.oldboy3.org','record':{'server':['110.110.1.1','110.110.1.1'],'weight':20,'maxconn':344}}"
#print(arg_list)
show()#显示提示
print('\n操作提示：按 \033[31;1m B/b\033[0m返回上层菜单<<< \033[31;1mQ/q\033[0m直接退出！\n')
arg=input('>>>>>请输入你要添加的信息:')
if arg=='b' or arg=='B':
break
elif arg=='q' or arg=='Q':
out()
else :
try:#进行异常处理
arg_dict=eval(arg)#输入信息转为字典
if judge(arg_dict):#判断是否已经存在
print('\033[41;1mbackend已经存在\033[0m'.center(40,'*'))
continue
else:
with open('HAproxy','a',encoding='utf-8') as add_file:
backend=arg_dict['backend']#key赋于变量 backend信息
record=arg_dict['record']##key赋于变量
print_info(backend,record)
#要添加的字符串变量
backend_add='''\nbackend {_backend_}
server {_server_1} {_server_2} weight {_weight_} maxconn {_maxconn_}'''
backend_add_for=backend_add.format(_backend_=backend,_server_1=record[server][0],_server_2=record[server][1],_weight_=record[weight],_maxconn_=record[maxconn])#进行格式化
q=input("\033[43;1m警告！\033[0m是否进行添加操作,按\033[43;1mY/y！\033[0m确认！否则返回")
if q=='y' or q=='Y':
add_file.write(backend_add_for)#写入文件
add_file.flush()#实时写入
print('\033[32;1m恭喜！添加成功！\033[0m\n\n')
continue
else:
break
except Exception as e:#出错可返回操作
print("\033[31;1m输入的格式有误，请重新输入！\033[0m")
pass def char_file():#修改backend信息
while True:
print('\033[35;1m修改backend信息！输入格式示例：\033[0m'.center(40,'='))
show()#显示提示
print('\n操作提示：按 \033[35;1m B/b\033[0m返回上层菜单<<< \033[35;1mQ/q\033[0m直接退出！\n')
arg=input('>>>>>请输入你要修改的信息:')
if arg=='b' or arg=='B':
break
elif arg=='q' or arg=='Q':
out()
else:
try:
arg_dict=eval(arg)#转为字典
if judge(arg_dict):#判断是否存在
backend=arg_dict['backend']#key赋于变量 backend信息
record=arg_dict['record']##key赋于变量
print('\033[32;1m新的配置\033[0m'.center(50,'-'))
print_info(backend,record)#显示要修改后的信息
dict_file=file_query()#执行查询函数
print('\033[37;1m旧的配置\033[0m'.center(50,'-'))
print_info(backend,dict_file[backend] )#显示修改前的信息
server_infor='server '+ dict_file[backend][server][0]+' '+dict_file[backend][server][1]#判断ip文本
backend_add='''\tserver {_server_1} {_server_2} weight {_weight_} maxconn {_maxconn_}'''#新的配置信息
backend_add_for=backend_add.format(_server_1=record[server][0],_server_2=record[server][1],_weight_=record[weight],_maxconn_=record[maxconn])#进行格式化
#file_char_list=[]
with open('HAproxy','r',encoding='utf-8') as chars_file:
for line in chars_file:
if server_infor in line:#判断 backend所在行
line=backend_add_for
file_pop_list.append(line)#写入列表
file_pop_list.append(line)#写入列表
q=input("\033[45;1m警告！\033[0m是否进行修改操作,按\033[45;1mY/y！\033[0m确认！否则退出修改操作!")
if q=='y' or q=='Y':
write_file(file_pop_list)#写文件函数
print("\033[31;1m修改完成!\033[0m")
continue
else:
break
else:
jud=input('该信息不存在，是否进行添加？按 \033[35;1m B/b\033[0m返回不添加 <<< \033[35;1mY/y\033[0m确定添加！\n')
if jud=='b' or jud=='B':
break
elif jud=='Y' or jud=='y':
add()
else:
pass
except Exception as e:#出错可返回操作
print("\033[31;1m输入的格式有误，请重新输入！\033[0m")
pass def write_file(file_pop_list):#写文件函数
with open('HAproxy','w',encoding='utf-8') as pop_file:
for i in range(len(file_pop_list)):
pop_file.write(file_pop_list[i])#修改文件
pop_file.flush() def judge(arg_dict):#判断是否存在的函数
backend=arg_dict['backend']#key赋于变量 backend信息
record=arg_dict['record']##key赋于变量
file_dict=file_query()#调用查询函数
if backend in file_dict:#判断是否已经存在
return True#返回真
else:
return False#返回假 def infor_pop():#删除backend信息
while True:
print('\033[31;1m删除backend信息！输入格式示例:\033[0m'.center(40,'-'))
#arg_list="{'backend':'www.oldboy3.org','record':{'server':['110.110.1.1','110.110.1.1'],'weight':20,'maxconn':344}}"
#print(arg_list)
show()#显示提示
print('\n操作提示：按 \033[31;1m B/b\033[0m返回上层菜单<<< \033[31;1mQ/q\033[0m直接退出！\n')
arg=input('>>>>>请输入你要删除的信息:')
if arg=='b' or arg=='B':
break
elif arg=='q' or arg=='Q':
out()
else:
try:#进行异常处理
arg_dict=eval(arg)#转为字典
if judge(arg_dict):#进行判断是否存在
backend=arg_dict['backend']#key赋于变量 backend信息
record=arg_dict['record']##key赋于变量
server_infor='server '+ record[server][0]+' '+record[server][1]#判断ip文本
#file_pop_list=[]
with open('HAproxy','r+',encoding='utf-8') as pop_file:
for line in pop_file:
if line.startswith('backend') and backend in line:#判断 backend所在行
line=''#
file_pop_list.append(line)#写入列表
if server_infor in line:#判断server所在行
line=''
file_pop_list.append(line)#写入列表
file_pop_list.append(line)#写入列表 追加
print_info(backend,record)
#print(file_pop_list)
q=input("\033[41;1m警告！\033[0m是否进行删除操作,按\033[41;1mY/y！\033[0m确认！否则退出返回!")
if q=='y' or q=='Y':
# with open('HAproxy','w',encoding='utf-8') as pop_file:
# for i in range(len(file_pop_list)):
# pop_file.write(file_pop_list[i])
# file_pop_list=[]
# pop_file.flush()
print(1)
write_file(file_pop_list)#写文件函数
print("\033[31;1m删除完成!\033[0m")
continue
else:
break
else:
print("\033[31;1m您所的信息不存,请重新输入!\033[0m")
except Exception as e:#出错可返回操作
print("\033[31;1m输入的格式有误，请重新输入！\033[0m")
pass def print_info(backend,record):#输入相关信息
infor_pr='''
\033[35;1m=======确认信息=========\033[0m
backend:{arg}
server: {server} {server1}
weight: {weight}
maxconn: {maxconn}
\033[35;1m=======================\033[0m
'''#定义输出文本
infor_pr=infor_pr.format(arg=backend,server=record[server][0],server1=record[server][1],weight=record[weight],maxconn=record[maxconn])#格式化文本
def print_sss():
infor_pr_1='''
\033[35;1m===========确认信息=============\033[0m \033[36;1m温馨提示:\033[0m\033[31;1m操作有风险,\033[0m\033[32;1m阁下需谨慎!\033[0m\033[33;1m请再次确认!'\033[0m
\033[37;1m确认操作系统将为您自动备份!\033[0m
\033[35;1m===============================\033[0m
'''#定义输出文本
print(infor_pr_1) #程序开始
print('\033[35;1mHAproxy 配置文件修改器\033[0m'.center(40,'='))
file_pop_list=[]#修改,删除,时读写的文件列表
backend_key='backend'
server='server'#定义字符串变量
weight='weight'
maxconn='maxconn'
while True:
print('''您可以进行以下相关操作:\n
按\033[32;1m 1 >>> 查询backend信息.\033[0m
按\033[33;1m 2 >>> 添加修改相关backend信息.\033[0m
按\033[35;1m 3 >>> 修改相关backend信息.\033[0m
按\033[31;1m 4 >>> 删除相关backend信息.\033[0m
按\033[30;1m 5 >>> 直接出退修改器.\033[0m
''')
choose=input('请输入您想要的操作 >>> ：')
if choose=='':
while True:
print('\033[32;1m查询backend信息界面！\033[0m'.center(50,'='))
#print_sss()
print('操作提示：按 \033[31;1m B/b\033[0m返回上层菜单<<< \033[31;1mQ/q\033[0m直接退出！')
arg=input('请输您要查询的信息:')
if arg=='b' or arg=='B':
break
elif arg=='q' or arg=='Q':
out()
else :
dict_file=file_query()#执行查询函数
if arg in dict_file:#如果所输入的存在
dict_file_key=dict_file[arg]#所对应的信息赋于变量
#infor=infor.format(arg=arg,server=dict_file_key[server][0],server1=dict_file_key[server][1],weight=dict_file_key[weight],maxconn=dict_file_key[maxconn])#格式化文本
#print(infor)
print_info(arg,dict_file_key)#调用信息函数
out()
else:
print('您输入的不存在,请重新输入')
pass
elif choose=='':
while True:
print('\033[33;1m添加backend信息界面！\033[0m'.center(50,'='))
print_sss()
qch=input('按提示进行操作：\033[33;1mB/b<<<\033[0m返回上层 \033[33;1mY/y>>>\033[0m确定开始添加 ')
if qch=='b' or qch=='B':
break
elif qch=='Y' or qch=='y':
print('添加backend信息，有一定的风险！已经为您自动备份配置文件！')
print('备份文件：\033[33;1m%s\033[0m'%filebak())#自动备份配置文件
add()#增加信息函数\
else:
pass
elif choose=='':
while True:
print('\033[35;1m修改backend信息界面！\033[0m'.center(50,'='))
print_sss()
qch=input('按提示进行操作：\033[35;1mB/b<<<\033[0m返回上层 \033[35;1mY/y>>>\033[0m确定开始添加 ')
if qch=='b' or qch=='B':
break
elif qch=='Y' or qch=='y':
print('修改backend信息，有一定的风险！已经为您自动备份配置文件！')
print('备份文件：\033[35;1m%s\033[0m'%filebak())
char_file()#修改文件函数
else:
pass
elif choose=='':
while True:
print('\033[31;1m删除backend信息界面！\033[0m'.center(40,'='))
print_sss()
qch=input('按提示进行操作：\033[31;1mB/b<<<\033[0m返回上层 \033[31;1mY/y>>>\033[0m确定开始删除 ')
if qch=='b' or qch=='B':
break
elif qch=='Y' or qch=='y':
print('删除信息，有一定的风险！已经为您自动备份配置文件！')
print('备份文件：\033[31;1m%s\033[0m'%filebak())
infor_pop()#删除信息函数
else:
pass
elif choose=='':
print('\033[30;1m您已经出退修改器.\033[0m')
exit()
else:
print('\033[41;1m您的输入有误！请重新输入！\033[0m')
pass
前后两天!终于搞定了!明天,嗯,今天就可以交作业啦!