python第九天----今天来晚了!

作业 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

前后两天!终于搞定了!明天,嗯,今天就可以交作业啦!