初学 python 之 HAproxy配置文件操作
HAproxy配置文件操作：
1. 根据用户输入输出对应的backend下的server信息
2. 可添加backend 和sever信息
3. 可修改backend 和sever信息
4. 可删除backend 和sever信息
5. 操作配置文件前进行备份
6 添加server信息时，如果ip已经存在则修改;如果backend不存在则创建；若信息与已有信息重复则不操作
配置文件 参考 http://www.cnblogs.com/alex3714/articles/5717620.html
目标:
练习文件读写,列表和列表解析,函数使用,循环表达式,if...else流程控制表达式.
程序流程:
1.开始,读取配置文件,用f.readlins()生成一个内容列表,列表元素按顺序依次与文本各行内容一一对应!
2.输入操作符
输入是c,进入21
21.输入网址, 网址不存在则返回2; 网址存在,则显示该网址的server信息,进入211 # 查看信息
211.输入ip,则删除该ip的server信息,备份和更新文件,显示新信息,然后返回2;输入b则直接返回2 # 删除server或返回不操作
输入m,进入22 # 添加修改信息
22.按格式输入网址以及server信息,网址存在则进入221,网址不存在则添加网址以及server信息,备份和更新文件,显示新信息,然后返回2 # 网址不存在则添加
221.ip存在,则进入2211;ip不存在就添加server信息,备份和更新文件,显示新信息,然后返回2 # ip不存在则添加
2211.weight 或 maxconn 的值相同,则显示网址和server信息,然后返回2;值不同,则修改值,备份和更新文件,显示新信息,然后返回2 # 输入信息已存在则显示信息,返回不操作
输入q,直接退出
重点:
利用列表有序特性,定位文本内容,实现查询,重点是,实现修改,插入添加等文本文件操作!
因为列表元素按顺序依次与文本各行内容一一对应,所以提取列表元素下标即可定位内容在文本位置,对列表的查询,修改,添加,然后写入文件即可!
使用函数处理 多处文件备份和更新等写入操作和多处显示列表信息操作 优化代码
代码:
1 # Author:Jay He 2 3 # 整个程序围绕data = f.readlines()列表做处理! 4 5 6 def get_server_list(net): 7 index_st = data.index(net) 8 index_follow = [data.index(i) for i in data[index_st+1:] if i[0] == 'b'] 9 # 把net往后的顶格字母是b的所有行下标放入index_follow 10 if index_follow: # 空列表本身就是False 11 index_end = min(index_follow) 12 # 是非空列表,就提取下标最小来截取ip下的server信息 13 get_list = [i.strip() for i in data[index_st+1:index_end]] 14 # 截取ip下包含的所有server信息,传递给列表get_list 15 else: 16 get_list = [i.strip() for i in data[index_st+1:]] 17 # 是空列表,直接截取net往后的所有信息,传递给列表get_list 18 return get_list 19 # 提取网址下server信息列表函数 20 21 22 def show_info(): 23 print(' %s' % add_info[0]) 24 for j in search_get: 25 if j == add_info[1]: 26 print(' 33[1;31m%s 33[0m' % j) 27 else: 28 print(' %s' % j) 29 # 显示server信息函数 30 31 32 def output_write(): 33 with open('profile_bak', 'w', encoding='utf-8') as f_backup, 34 open('profile', 'w', encoding='utf-8') as f_new: 35 for line in data_bak: 36 f_backup.write(line) 37 # 原内容备份到profile_bak 38 for line in data: 39 f_new.write(line) 40 # 更新修改过的内容到profile 41 # 备份和更新 42 43 44 while True: 45 with open('profile', 'r', encoding='utf-8') as f: 46 data = f.readlines() 47 input_choice = input('''输入c查询和删除,输入m添加和修改;输入q退出>>>''') 48 if input_choice == 'q': 49 break 50 elif input_choice == 'c': 51 input_net = input('请输入需要查询的网址,例如:www.baidu.com>>>') 52 add_info = ['backend '+input_net, 'none'] # 按统一格式处理输入信息 53 check_net = '%s ' % add_info[0] # 调整输入信息为data列表元素格式 54 if check_net in data: 55 search_get = get_server_list(check_net) # 获取server列表 56 print('参看网址的server信息如下:') 57 show_info() # 显示server信息 58 input_del = input('输入ip,删除server,例如:100.1.7.9;输入b返回>>>') 59 server_list = [k.split() for k in search_get if k != ''] # 调整server列表,可读取ip 60 ip_list = [k2[1] for k2 in server_list if k2 != ''] # 提取ip列表 61 if input_del in ip_list: 62 index_del = data.index(check_net)+ip_list.index(input_del)+1 # 定位输入ip在data中的位置 63 add_info[1] = search_get[ip_list.index(input_del)] # 把定位好的server信息传递给add_info方便下面用 64 del data[index_del] # 在data中删除server信息,准备输出 65 del search_get[ip_list.index(input_del)] # 在search_get中删除server信息,准备显示 66 data_bak = data # 设置备份列表data_bak,准备输出 67 output_write() # 输出 68 print('