python之haproxy配置文件操作（第三天）

作业：

对haproxy配置文件进行操作

要求：

对haproxy配置文件中backend下的server实现增删改查的功能

一、这个程序有二个版本

1、 python2.7版本见haproxy_python27.py

2、 python3.4版本见haproxy_python34.py

二、具体实现了如下功能：

1、输入1，进入backend菜单，查询server信息

2、输入2，进入backend菜单，添加server条目

3、输入3，进入backend菜单，选择server条目，进入修改环节

4、输入4，进入backend菜单，选择server条目，进入删除环节

5、输入5，退出程序

三、haproxy配置文件如下：

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 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.10 100.1.7.10 weight 10 maxconn 2000 frontend mysql bind *:3306 mode tcp log global default_backend mysqlserver backend mysqlserver server mysql1 10.1.1.110:3306 weight 20 maxconn 300 server mysql2 10.1.1.120:3306 weight 10 maxconn 200 haproxy配置文件

四、流程图如下：

五、python2.7代码：

#!/usr/bin/python27
#_*_ coding=utf-8 _*_  import os,sys,re,time from collections import defaultdict,OrderedDict ######################raw_input输入字符类型转化函数####################### def input_handle(s): if str.isdigit(s): ###对输入是否是数字进行判断### s = int(s) ###如果输出的是个数字，则转化为整数类型### return s ###返回输入字符### #####################名称或变量名输入判断函数############################# def name_decide(): (name,name_flag) = ('',True) ###初始化返回的名称和判断标志位### while name_flag: name_input = raw_input('\033[32m请输入新服务名:(退出输入q)\033[0m') if len(name_input) == 0: ###如果输入为空则直接下一次循环### continue elif name_input == 'q': ###输入q，退出本次输入### name_flag = False elif re.match('[0-9a-zA-Z\_]+',name_input): ###匹配输入是否以字符、数字或下划线开头### name = name_input name_flag = False ###输入成功后退出循环### else: print('\033[31m名称输入错误，请重新输入!!!\033[0m') return(name) ###返回输入的结果### #####################IP地址及端口输入判断函数############################# def ipaddress_decide(): (address,address_flag) = ('',True) ###初始化返回的IP地址和判断标志位### while address_flag: address_input = raw_input('\033[32m请输入新地址(IP哦):(退出输入q)\033[0m') if len(address_input) == 0: ###如果输入为空则直接下一次循环### continue elif address_input == 'q': ###输入q，退出本次输入### address_flag = False ##################匹配输入是否是ip:port的格式###################### elif re.match('(25[0-5]|2[0-4]\d|[0-1]?\d?\d)(\.(25[0-5]|2[0-4]\d|[0-1]?\d?\d)){3}(\:\d{1,5})?$',address_input): address = address_input address_flag = False ###输入成功后退出循环### else: print('\033[31m地址输入错误，请重新输入!!!\033[0m') return(address) ###返回输入的结果### ####################数字输入判断函数###################################### def number_decide(name): (number,number_flag) = ('',True) ###初始化返回的数字和判断标志位### while number_flag: number_input = raw_input('\033[32m请输入 %s (数字哦):(退出输入q)\033[0m' % name) if len(number_input) == 0: ###如果输入为空则直接下一次循环### continue elif number_input == 'q': ###输入q，退出本次输入### number_flag = False else: try: int(number_input) ###匹配输入是否是数字### except: print('\033[31m %s 输入错误，请重新输入!!!\033[0m' % name) else: number = number_input number_flag = False ###输入成功后退出循环### return(number) ###返回输入的结果### #############################backend输入判断函数########################## def backend_input_if(input_index,input_dict): (input_name,input_flag) = ('',True) ###初始化返回的名称和判断标志位### ###############如果输入非空，对输入进行判断并转化类型################# if len(input_index) != 0: input_index = input_handle(input_index) if input_index == 'b': ###如果输入为b,则退出程序### input_flag = False elif input_index in input_dict.keys(): ###如果输入为数字编号，则从字典中获取具体backend名称### input_name = input_dict[input_index] elif input_index in input_dict.values(): ###如果输入为backend名称，则直接获取### input_name = input_index else: input_name = '' ###输入其他字符，赋值变量为空### return(input_name,input_flag) ###返回输入的结果和循环标志位### ###########################backend列表展示函数############################ def backend_read(file): backend_list = [] ###初始化backend的列表### show_dict = {} ###初始化要返回的显示字典### backend_name_dict = defaultdict(list) ###定义一个value为列表的字典，用来存放server### server_flag = False ###初始化server判断标志位### with open(file,'r') as ha: ###打开haproxy配置文件### for line in ha: server_dict = OrderedDict() ###定义一个有序字典### line = line.strip('\n') if re.match('backend',line): ###匹配配置文件以backend开头的行### backend_name = re.split('\s+',line)[1] backend_list.append(backend_name) ###将配置到的backend的名称插入到列表### server_flag = True ###赋值标志位为真，用来与server关联### elif server_flag and re.match('\s+server',line): ###匹配配置文件以server开头的行### server_info = re.split('\s+',line) ###对server进行分隔### server_dict['name'] = server_info[2] ###对server的具体信息进行字典赋值### server_dict['address'] = server_info[3] server_dict['weight'] = server_info[5] server_dict['maxconn'] = server_info[7] backend_name_dict[backend_name].append(server_dict) ###将server字典与backend的名称进行关联### else: server_flag = False ###当server没匹配到，赋值标志位为假，结束关联### for k,v in enumerate(backend_list,1): show_dict[k] = v ###对backend名称进行新字典赋值，方便查询和展示### print('%d . %s' % (k,v)) ###输出backend列表### return(show_dict,backend_name_dict) ###返回显示的字典和backend-server字典### ########################显示backend后端服务器函数############################# def backend_server_show(backend_show_value,server_show_dict): ############对backend名称进行遍历并加上数字编号########################### print('\n================================================================') print('后端服务器(%s)信息如下 : ' % backend_show_value) print('%-5s %-20s %-20s %-20s %-20s' % ('id','name','address','weight','maxconn')) server_list = server_show_dict[backend_show_value] for k,v in enumerate(server_list,1): ###用enumerate进行server展示### print '%-5s' % k + ' ', for kk,vv in v.items(): print '%-20s' % vv + ' ', print print('\n================================================================') ########################backend后端服务操作并回写配置文件函数################# def backend_server_handle(file,handle_dict): newfile = '%s.new' % file ###定义回写的新文件### server_flag = False with open(file,'r') as read_file,open(newfile,'w') as write_file: ###同时打开二文件，一个读，一个写### for line in read_file: if re.match('backend',line): ###匹配到backend行时进行server信息插入### write_file.write(line) backend_name = re.split('\s+',line)[1] for server_dict in handle_dict[backend_name]: ###对backend-server字典进行遍历### server_line = '\tserver {name} {address} weight {weight} maxconn {maxconn}\n' write_file.write(server_line.format(**server_dict)) ###将指定的backend下的server条目插入文件### server_flag = True ###指定标志位为真，方便server判断### elif server_flag and re.match('\s+server',line): ###匹配server开头的行，跳过不做任何操作### pass else: write_file.write(line) ###其他的行，直接插入新文件### server_flag = False print('\033[33m server更新成功\033[0m') os.system('mv %s %s.bak' % (file,file)) ###对源配置文件进行备份### os.system('mv %s %s' % (newfile,file)) ###对新生成的配置文件进行改名### ##################################主程序开始################################## if __name__ == '__main__': flag = True haproxy_file = 'haproxy.cfg' ###指定haproxy配置文件### backend_name = '' haproxy_name_show_dict = {} ###初始化backend显示字典### haproxy_server_show_dict = {} ###初始化server显示字典### while flag: os.system('clear') ###清屏### print('\n================================================================') print('\033[33m 欢迎访问haproxy配置文件管理平台：\033[0m') print('\nbackend列表信息如下 : ') (haproxy_name_show_dict,haproxy_server_show_dict) = backend_read(haproxy_file) print('\n================================================================') print ''' \033[31m 1. 查询后端服务\033[0m \033[32m 2. 添加后端服务\033[0m \033[33m 3. 修改后端服务\033[0m \033[35m 4. 删除后端服务\033[0m \033[36m 5. 退出\033[0m ''' print('\n================================================================') select_num = raw_input('\033[33m请选择操作条目：\033[0m') #######################1. 查询后端服务################################ if select_num == '1': query_flag = True ###初始化查询循环标志### while query_flag: backend_index = raw_input('\033[32;1m请输入查询的backend编号或名称(返回上层菜单,请输入b)：\033[0m') ##############################对输入值进行判断################ (backend_name,query_flag) = backend_input_if(backend_index,haproxy_name_show_dict) if backend_name: ###显示对应backend下的server列表### backend_server_show(backend_name,haproxy_server_show_dict) #######################2. 添加后端服务################################ if select_num == '2': add_flag = True ###初始化添加循环标志### while add_flag: backend_index = raw_input('\033[32;1m请输入添加的backend的编号或名称(返回上层菜单,请输入b)：\033[0m') ##############################对输入值进行判断################ (backend_name,add_flag) = backend_input_if(backend_index,haproxy_name_show_dict) if backend_name: ###显示对应backend下的server列表### backend_server_show(backend_name,haproxy_server_show_dict) add_server_dict = OrderedDict() print('\033[34;1m请依次输入后端服务信息(name,address,weight,maxconn):\033[0m') add_server_dict['name'] = name_decide() ###对输入的name有效性进行判断### add_server_dict['address'] = ipaddress_decide() ###对输入的IP有效性进行判断### add_server_dict['weight'] = number_decide('权重值') ###对输入的权重有效性进行判断### add_server_dict['maxconn'] = number_decide('最大连接数') ###对输入的连接数有效性进行判断### print(add_server_dict['name'],add_server_dict['address'],add_server_dict['weight'],add_server_dict['maxconn']) #############对输入的四个服务信息是否成功判断############# if add_server_dict['name'] and add_server_dict['address'] and add_server_dict['weight'] and add_server_dict['maxconn']: add_commit = raw_input('请确认是否添加此条目（y | n）：') if add_commit == 'y': ###确认添加服务条目，并回写配置文件### haproxy_server_show_dict[backend_name].append(add_server_dict) backend_server_handle(haproxy_file,haproxy_server_show_dict) else: add_flag = False ###否则退出本次循环### else: print('\033[31m server输入信息有误，请重新输入!!!\033[0m') #######################3. 修改后端服务################################# if select_num == '3': backend_modify_flag = True ###初始化修改循环标志### while backend_modify_flag: backend_index = raw_input('\033[32;1m请输入修改的backend的编号或名称(返回上层菜单,请输入b)：\033[0m') ##############################对输入值进行判断################ (backend_name,backend_modify_flag) = backend_input_if(backend_index,haproxy_name_show_dict) if backend_name: ###显示对应backend下的server列表### backend_server_show(backend_name,haproxy_server_show_dict) server_modify_flag = True ###初始化server条目修改标志位### while server_modify_flag: server_index = raw_input('\033[32;1m请输入修改的server的编号(返回上层菜单,请输入b)：\033[0m') if len(server_index) != 0: server_index = input_handle(server_index) if server_index == 'b': ###输入b，返还上一层### server_modify_flag = False #####################指定具体的server编号进行判断##### elif server_index >= 1 and server_index <= len(haproxy_server_show_dict[backend_name]): modify_server_dict = OrderedDict() print('\033[34;1m请依次输入后端服务信息(name,address,weight,maxconn):\033[0m') modify_server_dict['name'] = name_decide() ###对输入的四个信息有效性进行判断### modify_server_dict['address'] = ipaddress_decide() modify_server_dict['weight'] = number_decide('权重值') modify_server_dict['maxconn'] = number_decide('最大连接数') print(modify_server_dict['name'],modify_server_dict['address'],modify_server_dict['weight'],modify_server_dict['maxconn']) ######对输入的四个服务信息是否成功判断############# if modify_server_dict['name'] and modify_server_dict['address'] and modify_server_dict['weight'] and modify_server_dict['maxconn']: modify_commit = raw_input('请确认是否修改此条目（y | n）：') if modify_commit == 'y': ###确认修改服务条目，并回写配置文件### haproxy_server_show_dict[backend_name][server_index - 1] = modify_server_dict backend_server_handle(haproxy_file,haproxy_server_show_dict) modify_server_flag = False else: ###否则退出本次循环### modify_server_flag = False else: print('\033[31m server输入信息有误，请重新输入!!!\033[0m') else: print('\033[31m server编号输入错误，请重新输入!!!\033[0m') #######################4. 删除后端服务################################ if select_num == '4': backend_delete_flag = True ###初始化删除循环标志### while backend_delete_flag: backend_index = raw_input('\033[32;1m请输入删除的backend下条目编号或名称(返回上层菜单,请输入b)：\033[0m') ##############################对输入值进行判断################ (backend_name,backend_delete_flag) = backend_input_if(backend_index,haproxy_name_show_dict) if backend_name: ###显示对应backend下的server列表### backend_server_show(backend_name,haproxy_server_show_dict) server_delete_flag = True ###初始化server条目删除标志位### while server_delete_flag: server_index = raw_input('\033[32;1m请输入删除的server的编号(返回上层菜单,请输入b)：\033[0m') if len(server_index) != 0: server_index = input_handle(server_index) if server_index == 'b': ###输入b，返还上一层### server_delete_flag = False #####################指定具体的server编号进行判断##### elif server_index >= 1 and server_index <= len(haproxy_server_show_dict[backend_name]): print(haproxy_server_show_dict[backend_name][server_index - 1]) delete_commit = raw_input('请确认是否删除此条目（y | n）：') if delete_commit == 'y': ###确认删除服务条目，并回写配置文件### del haproxy_server_show_dict[backend_name][server_index - 1] backend_server_handle(haproxy_file,haproxy_server_show_dict) delete_server_flag = False else: ###否则退出本次循环### delete_server_flag = False else: print('\033[31m server编号输入错误，请重新输入!!!\033[0m') #######################5. 退出程序###################### if select_num == '5': sys.exit('\033[32m 退出程序，欢迎下次光临，谢谢\033[0m') haproxy_python27

七、效果图：

1、  初始菜单直接显示backend列表，并列出选择菜单：

2、  输入1，选择查询菜单，并输入backend的名称或编号，均可，并展示对应名称下的server信息：

3、  输入b可以返回上层菜单，输入2进入添加server条目：

4、  输入server对应的name，address，weight，maxconn，并对有效性进行判断：

添加成功后查看结果：

注：

name以数字、字母、下划线开头（其中包含数字的原因是可以输入ip地址）

address可以是单独的ip地址，也可以是ip加端口的形式，例：192.168.100.3或192.168.100.3:8080

weight和maxconn必须输入数字，大小暂时没有限制

5、 输入b可以返回上层菜单，输入3进入修改server条目，并对 www.oldboy.org下的sky 条目进行修改：

同样对输入的server有效性进行判断，最后确认回写配置文件，查看修改结果：

6、  输入b可以返回上层菜单，输入4进入删除server条目，并对指定条目进行修改：

7、输入b可以返回上层菜单，输入5，即可退出程序。

八、针对python2.7和python3.4的代码区别：

1、print的使用。

2、python2.7下使用raw_input，python3.4下使用input。