Day3作业：ha_proxy配置文件修改
不废话，上代码
readme:
# Auther:ccorz Mail:ccniubi@163.com Blog:http://www.cnblogs.com/ccorz/
# GitHub:https://github.com/ccorzorz
ha_proxy配置文件修改程序
1.查询时输入域名即可查询，显示域名相关的backend配置
2.删除时输入域名，即可删除包括域名以及配置服务器信息的相关内容；如相关域名信息，会提示用户，不改动文件
3.修改时需要输入列表信息，列表中需要包括backend的域名信息，以及包含weight、server、maxconn的record列表
3.1 修改需输入内容举例：{"backend": "test.oldboy.org","record":{"server": "100.1.7.9","weight": 20,"maxconn": 30}}
3.2 程序会自动判断输入的格式是否正确，如果不正确，会提醒用户重新输入，直至正确格式
3.3 程序会自动判断域名信息是否存在，如存在，增加服务器信息;如不存在，追加至配置文件结尾
3.4 程序会自动判断用户想增加的服务器信息是否与已存在的配置重叠，如存在，提示用户，不修改文件；如不存在，在域名相关信息后追加
4.程序会自动备份修改之前的文件，后缀为修改时间
5.本程序有日志记录，以便管理员查询修改记录
流程图：
ha_proxy配置文件：
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.9 100.1.7.9 weight 20 maxconn 30
backend buy.oldboy.org
server 100.1.7.90 100.1.7.90 weight 20 maxconn 3000
程序代码：
#!/usr/bin/env python # -*-coding=utf-8-*- # Auther:ccorz Mail:ccniubi@163.com Blog:http://www.cnblogs.com/ccorz/ # GitHub:https://github.com/ccorzorz import json,time,os time_now=time.strftime('%Y-%m-%d %H:%M:%S') time_flag=time.strftime('%Y-%m-%d-%H-%M-%S') #定义通过域名查询ha配置文件函数 def search_backend(search_info): """ :param search_info: 用户输入的需要查询的域名 :return: result_list :结果列表,供增删改使用 """ b_flag=1 #设置初始标识符 result_list=[] #设置查询结果为空列表 with open('ha','r') as ha_read: for line in ha_read.readlines(): #将内容通过readlines转化为列表 line=line.strip() #每一行去掉两头的空格和回车 if line=='backend %s'%search_info: #发现内容匹配 b_flag=0 #更改标识符值 elif line.startswith('backend'): #发现以'backend'开头的内容 b_flag=1 #更改标识符为初始值 elif b_flag==0 and len(line)!=0: #将标识符为0的非空内容加入查询结果列表 result_list.append(line) return result_list def add_backend(): """ :return: None """ while True: add_contect_str=input('Input the server_info:') #提示用户输入配置文件信息 res=enter_query(add_contect_str) #将确认格式函数的返回值赋予res变量 if res==False: #如果变量为Flase,提醒用户输入的字符串格式不正确 print('String format error,try again!') if res == True: #如果变量值为True,即用户输入格式正确，执行以下 # print(result_list) add_contect=json.loads(add_contect_str) #通过json模块，将字符串转化为列表 search_info=add_contect['backend'] #变量赋值 weight=add_contect['record']['weight'] server=add_contect['record']['server'] maxconn=add_contect['record']['maxconn'] server_info='server %s %s weight %s maxconn %s'%(server,server,weight,maxconn) result_list=search_backend(search_info) #将字符串中解析得到的列表赋值于result_list print(result_list) if len(result_list)==0: #如果无对应的域名，新增域名以及对应的记录 with open('ha','r') as ha_read,open('ha_new','w') as ha_wirte: con_list=ha_read.readlines() #通过readlines将ha文件中的内容转化为列表 con_list.append('\n') #列表中加一空行 con_list.append('backend %s\n'%search_info) #列表中新增域名信息 con_list.append('%s%s'%(8*' ',server_info)) #列表中追加域名对应服务器信息 ha_wirte.writelines(con_list) #将列表通过writelines全部写入ha_write文件 #回显，提示用户无域名信息，将新增信息到配置文件最后 print('No domain target,will add to the end of the "ha",has been add the end of file..') time.sleep(1) print('\033[31;1mInformation has been added successly!!!!\033[0m') os.rename('ha', 'ha_bak.%s' % time_flag) # 系统改名 os.rename('ha_new', 'ha') with open('modify.log', 'a') as log: #写入日志文件 log.write('%s domain %s add server info: %s\n' % (time_now, search_info, server_info)) else: #列表不为空 if server_info in result_list: #如果服务器信息重叠，提示用户信息已重叠，不操作文件 print('\033[31;1mThe server info had been in %s domain,do not need to add...\033[0m'%search_info) else: result_list.append(server_info) #服务器信息不重叠 with open('ha','r') as ha_read,open('ha_add','w') as ha_add: a_flag=1 #设置取配置文件中目标内容的标识符 exit2_flag=1 #设置限制目标内容循环的标识符 for line in ha_read.readlines(): #转化列表，遍历列表 if line.strip()=='backend %s'%search_info: #如果其中一行的内容为“backend 域名名称”,更改目标内容标识符的值 a_flag=0 # 更改目标内容标识符后，继续遍历列表，发现以backend开头的行，再次更改目标内容标识符的值， # 两个标识符中间的内容为即将修改的目标内容 elif line.strip().startswith('backend'): a_flag=1 if a_flag==1: #如果非目标内容写入ha_add文件 ha_add.write(line) elif a_flag==0 and exit2_flag==1: #如果是目标内容，并且限制循环的标识符为初始值 ha_add.write('backend %s\n'%search_info) #添加一行域名信息 for line in result_list: #遍历即将修改的列表，在元素内容前加入8个空格 num=result_list.index(line) result_list[num]='%s%s\n'%(8*' ',line) ha_add.writelines(result_list) #将列表中内容写入ha_add文件 ha_add.write('\n') #添加一行空白行 exit2_flag=0 #更改限制循环标识符的初始值，使其只循环一次，避免多次写入列表中的服务器信息 os.rename('ha','ha_bak.%s'%time_flag) #系统改名 os.rename('ha_add','ha') with open('modify.log','a') as log: #写入修改日志文件 log.write('%s %s add server info: %s\n'%(time_now,search_info,server_info)) print('\033[31;1mModify success!!!!\033[0m') #回显 break #定义删除函数 def del_backend(del_domain): with open('ha','r') as ha_read,open('ha_del','w') as ha_del: b_flag=1 #设置目标内容初始值 for line in ha_read.readlines(): #将内容通过readlines转化为列表 if line.strip()=='backend %s'%del_domain: #发现内容匹配 b_flag=0 #更改标识符值 elif line.strip().startswith('backend'): #发现以'backend'开头的内容 b_flag=1 #更改标识符为初始值 if b_flag==1: #只将非目标内容写入ha_del文件 ha_del.write(line) with open('modify.log', 'a') as log: #操作日志写入日志文件 log.write('%s delete the domain %s and the related info.\n' % (time_now, del_domain)) os.rename('ha','ha_bak.%s'%time_flag) #系统改名ha以及操作文件 os.rename('ha_del','ha') print('域名以及域名信息已全部删除...') #定义确认用户输入格式是否为ha配置文件所需格式函数 def enter_query(add_contect_str): """ :param add_contect_str: 用户选择修改配置文件后，输入的格式必须为字典 :return: Flase=用户输入格式不正确 True=用户输入格式正确 """ try: #尝试将用户输入字符串转化为列表，并给变量赋值 add_contect=json.loads(add_contect_str) search_info=add_contect['backend'] weight=add_contect['record']['weight'] server=add_contect['record']['server'] maxconn=add_contect['record']['maxconn'] except: #如果异常，return 值为False return False else: #如果无报错，return 值为False return True #定义主函数 def main(): while True: start_num=input("""1.查询 2.修改 3.删除 q.退出""") if start_num.isdigit(): start_num=int(start_num) while True: if start_num==1: search_info=input('Please enter the domain name:') result_list=search_backend(search_info) if len(result_list)==0: #如果配置文件中无查询的内容,回显 print('\033[31;1mSorry,the configuration file does not have the information you want to query!\033[0m') else: #如果配置文件中有要查询的内容,打印查询结果 for line in result_list: print('\033[31;1m%s\033[0m'%line) break elif start_num==2: add_backend() break elif start_num==3: #如果选择删除，提示用户输入所需删除的域名 del_domain=input('Input the domain want to delete:') result_list=search_backend(del_domain) #列表赋值于变量 if len(result_list)==0: #如果列表为空，无域名相应信息，提示用户 print('\033[31;1mSorry,the configuration file does not have the information you want to delete!\033[0m') else: #如果有相应信息，执行删除函数 del_backend(del_domain) break else: #其他输入，提示用户输入错误，请重新输入 print('Information is incorrect, please re-enter...') break elif start_num=='q': #选择退出 print('Bye!!!!') break else: #输入无相应功能，提示重新输入 print('Information is incorrect, please re-enter...') #执行主函数 main()
日志内容：
2016-05-23 21:32:33 delete the domain www.oldboy.org and the related info. 2016-05-23 21:36:45 delete the domain www.oldboy.org and the related info. 2016-05-23 21:38:29 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 21:41:38 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 21:43:50 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 21:45:59 www.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 21:47:48 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 22:06:05 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 22:08:34 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30 2016-05-23 22:19:40 delete the domain test.oldboy.org and the related info.
沛sir的demo：
#!/usr/bin/env python # -*- coding:utf-8 -*- def fetch(backend): # backend = "www.oldboy.org" result = [] with open('ha.conf', 'r', encoding='utf-8') as f: flag = False for line in f: if line.strip().startswith("backend") and line.strip() == "backend " + backend: flag = True continue if flag and line.strip().startswith("backend"): flag = False break if flag and line.strip(): result.append(line.strip()) return result # ret = fetch("www.oldboy.org") # print(ret) def add(backend, record): # 思路一： # 思路二： # 先检查记录存不存 record_list = fetch(backend) if not record_list: # backend不存在 with open('ha.conf', 'r') as old, open("new.conf", 'w') as new: for line in old: new.write(line) new.write("\nbackend " + backend + "\n") new.write(" " * 8 + record + "\n") else: # backend存在 if record in record_list: # record已经存在 # import shutil # shutil.copy("ha.conf", 'new.conf') pass else: # backend存在,record不存在 record_list.append(record) with open('ha.conf', 'r') as old, open('new.conf', 'w') as new: flag = False for line in old: if line.strip().startswith("backend") and line.strip() == "backend " + backend: flag = True new.write(line) for new_line in record_list: new.write(" "*8 + new_line + "\n" ) continue if flag and line.strip().startswith("backend"): flag = False new.write(line) continue if line.strip() and not flag: new.write(line) bk = "www.oldboy.org" rd = "server 100.1.7.49 100.1.7.49 weight 20 maxconn 30" add(bk, rd) # s = "[11,22,33,44,5, 66]" # s = '{"k1":"v1"}' # print(type(s),s) # # import json # n = json.loads(s) # 将一个字符串，转换成python的基本数据类型; 注意：字符串形式的字典'{"k1":"v1"}'内部的字符串必须是 双引号 # print(type(n), n) # import json # r = input("input:") # dic = json.loads(r) # bk = dic['backend'] # rd = "server %s %s weight %d maxconn %d" %(dic['record']['server'], # dic['record']['server'], # dic['record']['weight'], # dic['record']['maxconn'])
