Day3作业：ha_proxy配置文件修改
Posted ccorz
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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\'])
以上是关于Day3作业：ha_proxy配置文件修改的主要内容，如果未能解决你的问题，请参考以下文章