1 #!/usr/bin/env python
2 #=*- coding:UTF-8 -*-
3
4 #查找
5 def search(website ):
6 reault_list = []#存放查询结果
7 search_flag = False #定义查询结果标志
8
9 with open('haproxy','r',encoding= 'utf-8') as f:
10 for line in f:#if语句顺序进行，当前条件判断完，进入下一判断
11 if 'backend {}'.format(website) == line.strip():#找到backend www.oldboy.org 开始截取
12 search_flag = True
13 if line.strip().startswith('backend') and 'backend {}'.format(website) != line.strip() :#找到下一个 backend www.xxx.com部分结束，截取其中部分
14 search_flag = False
15 if search_flag:
16 reault_list.append(line .strip() )
17 print(reault_list )
18 f.close()
19 if reault_list == []:
20 print('对不起，您查询的网址不存在！')
21 else:
22 return reault_list
23
24 #增加
25 def add(website):
26 arg = eval(website) #将输入的字符串转变为字典
27 backend_title = 'backend {}'.format(arg['backend'])# 要插入backend的字段
28 record_title = arg['record']
29 context_record = 'server {0}{0} weight {1} maxconn {2}'.\
30 format(record_title['server'],record_title['weight'],record_title['maxconn'] )
31 add_flag = True#设置新增标志位
32 with open('haproxy', 'r+', encoding='utf-8') as f:
33 for line in f:
34 if line.strip() == backend_title:
35 print("您新增的网址已经存在！")
36 add_flag = False
37 if add_flag:
38 f.write('\n{}'.format(backend_title))
39 f.write('\n\t\t{}'.format(context_record))
40 f.close()
41
42 #删除
43 def delete(website):
44 delete_flag = False #设置删除标志位
45 delete_result = True
46 with open('haproxy', 'r', encoding='utf-8') as f,\
47 open('haproxy_bak', 'w') as f1:
48 for line in f:
49 if 'backend {}'.format(website ) == line.strip():
50 delete_flag = True
51 delete_result = False
52 continue
53 if line.strip().startswith('backend') and line.strip() != website:
54 delete_flag = False
55 if not delete_flag:
56 f1.write(line)
57 if delete_result :
58 print("您删除的网址不存在！")
59 with open('haproxy', 'w') as f, \
60 open('haproxy_bak', 'r', encoding='utf-8') as f1:
61 for line in f1:
62 f.write(line)
63
64 #修改
65 def update(website):
66 arg = eval(website )
67 # print(arg)
68 backend_title = "backend {}".format(arg['backend']) # 要插入backend整个字段
69 # print(backend_title)
70 record_title = arg["record"]
71 context_record = "server {0} {0} weight {1} maxconn {2}". \
72 format(record_title['server'], record_title['weight'], record_title['maxconn'])
73 # print(context_record)
74
75 update_flag = False #设置修改标志位
76 update_re = False #设置重复修改位
77
78 with open('haproxy', 'r', encoding='utf-8') as f, \
79 open('haproxy_bak', 'w') as f1:
80 for line in f:
81 if line.strip() == backend_title:
82 update_flag = True
83 continue #如果出现修改backend_title,设置修改标志位为True。
84 if line.strip().startswith('backend') and line.strip() != backend_title :
85 update_flag = False #捕捉到下一个backend_title,设置修改标志位为False。
86 if not update_flag:
87 f1.write(line)
88 if update_flag and not update_re: #修改标志位为True,并且没有被修改过
89 f1.write('\n{}'.format(backend_title))
90 f1.write('\n\t\t{}\n'.format(context_record))
91 update_re = True
92
93 with open('haproxy', 'w') as f, \
94 open('haproxy_bak', 'r', encoding='utf-8') as f1:
95 for line in f1:
96 f.write(line)
97
98
99
100
101 while True:
102 print('欢迎进入haproxy配置程序'.center(50,'-') )
103 print( '\n1 查询\n')
104 print(' 2 新增\n')
105 print(' 3 删除\n')
106 op_haproxy = input('选择要进入模式的ID-》：')
107 if op_haproxy == '1':
108 website = input('请输入要查询的网址：\n'
109 '例如www.oldboy.org')
110 search(website )
111 elif op_haproxy == '2':
112 website = input("请输入要新增的网址配置："
113 "例如：{'backend': 'www.baidu.com','record': {'server': '100.1.7.8',"
114 "'weight': 20,'maxconn': 3000}}\n")
115 add(website)
116 elif op_haproxy == '3':
117 website = input("请输入要删除的网址配置："
118 "例如：www.baidu.com\n")
119
120 delete(website)
121 elif op_haproxy == '4':
122 website = input("请输入要修改的网址配置："
123 "例如：{'backend': 'www.baidu.com','record': {'server': '100.1.7.8',"
124 "'weight': 20,'maxconn': 3000}}\n")
125 update(website)
126 elif op_haproxy == 'q':
127 break
128 else:
129 print("请检查您的输入！")
