HAproxy配置文件操作

要求

1. 根据用户输入输出对应的backend下的server信息
2. 可添加backend 和sever信息
3. 可修改backend 和sever信息
4. 可删除backend 和sever信息
5. 操作配置文件前进行备份
6 添加server信息时，如果ip已经存在则修改;如果backend不存在则创建；若信息与已有信息重复则不操作

  1 def find(backend):
  2     '''
  3     查看backend下sever信息
  4     :param backend:
  5     :return:
  6     '''
  7     ls = []
  8     with open('HA.txt', 'r') as f:
  9         T = True
 10         for line in f:
 11             # 找到第一个backend，做记录 T = False
 12             if line.strip() == 'backend %s' % (backend):
 13                 T = False
 14                 continue
 15             # 找到第二个backend，做记录 T = True
 16             if T == False and line.strip().startswith('backend'):
 17                 T = True
 18                 break
 19             # 把关于False的加入到列表中
 20             if T == False and line.strip():
 21                 ls.append(line.strip())
 22     return ls
 23 
 24 
 25 def check_bk():
 26     '''
 27     列出backend的信息
 28     :return: 选中的backend
 29     '''
 30     ls = []
 31     T = True
 32     with open('HA.txt', 'r') as f1:
 33          for line in f1:
 34             if line.strip().startswith('backend') == True:
 35                 to = line.split(' ')
 36                 ls.append(to[1])
 37          for i, v in enumerate(ls,1):
 38             print (i, v)
 39          while T:
 40              t1 = input('请输入查看的server的 backend信息')
 41              if t1.isdecimal() and int(t1) in range(0,len(ls)+1):
 42                 a = ls[int(t1)-1]
 43                 break
 44              else:
 45                 print('输入有误，重新输入')
 46     return a
 47 
 48 def backup():
 49     '''
 50     备份
 51     :return:
 52     '''
 53     with open('HA.txt', 'r') as f1, open('HA_old.txt', 'w') as f2:
 54         for line in f1:
 55             f2.write(line)
 56 
 57 
 58 def add(bk, sr):
 59     '''
 60     添加记录
 61     :param bk:
 62     :param sr:
 63     :return:
 64     '''
 65     r_list = find(bk)
 66     # 没有关于bk的记录，直接添加内容
 67     T = True
 68     a = 'backend %s' % bk
 69     backup()
 70     if not r_list:
 71         with open('HA.txt', 'a') as f1:
 72             f1.write('
%s
'% a)
 73             f1.write("        %s
" % sr)
 74     # 有关于bk的记录
 75     else:
 76         # 没有sever记录
 77         backup()
 78         if sr not in r_list:
 79             with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
 80                 for line in f1:
 81                     if line.strip() == 'backend %s' % (bk):
 82                         T = False
 83                         f2.write(line)
 84                         continue
 85                     if T == False and line.strip().startswith('backend'):
 86                         T = True
 87                     if T == True:
 88                         f2.write(line)
 89                     else:
 90                         if sr not in r_list:
 91                             r_list.append(sr)
 92                             t = '
        '.join(r_list)
 93                             f2.write(' '* 8 + t + '

')
 94     # 有sever记录,直接复制
 95         else:
 96             print('记录已存在，请重新选择')
 97             caidan()
 98 
 99 
100 def update_bk ():
101     '''
102     需要修改或删除backend
103     :return:
104     '''
105     backup()
106     T = True
107     bk = check_bk()
108     ls =[]
109     with open('HA.txt', 'r') as f1:
110          for line in f1:
111             if line.strip().startswith('backend') == True:
112                 to = line.split(' ')
113                 ls.append(to[1].strip())
114     c = input('''
115     请选择需要选项：
116     1 修改backend
117     2 删除backend
118     ''')
119     while T:
120         #修改backend信息
121         if c == '1':
122             bk1 = input('请输入修改后的backend信息')
123             t3 = input('信息确认请按 y, 重新输入请按其他键')
124             if bk1 in ls:
125                 print('信息已存在，请重新选择')
126                 break
127             else:
128                 if t3.lower() == 'y':
129                     with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
130                         for line in f1:
131                             #需要修改的backend
132                             if line.strip() == 'backend %s' % (bk.strip()):
133                                 f2.write('backend %s 
' % (bk1))
134                             else:
135                                 f2.write(line)
136                         print('修改成功')
137                         break
138                 else:
139                     continue
140             #删除backend信息
141         elif c == '2':
142             t = find(bk.strip())
143             while T:
144                 with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
145                     for line in f1:
146                         if line.strip() == 'backend %s' % (bk.strip()):
147                             continue
148                         if line.strip() in t:
149                             continue
150                         else:
151                             f2.write(line)
152                     print('删除成功')
153                     break
154             break
155 
156 def check_sr():
157     T1 = True
158     while T1:
159         cont = 0
160         a1 = input('sever:')
161         ti = a1.split('.')
162         for ii in ti:
163             if ii.isdecimal() and int(ii) in range(0, 255) and len(ti) == 4:
164                 cont = cont + 1
165                 continue
166             else:
167                 print('输入有误，请输入由四组0-255组成的IP地址')
168                 break
169         if cont == 4:
170             break
171     return a1
172 
173 
174 def update_sr():
175     '''
176     修改或删除sever
177     :return:
178     '''
179     t = check_bk()
180     backup()
181     T = True
182     T1 = True
183     ls = find(t.strip())
184     cc = 0
185     with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
186         for line in f1:
187             if line.strip() == 'backend %s' %(t.strip()):
188                 T = False
189                 f2.write(line)
190                 continue
191             if T == False and line.strip().startswith('backend'):
192                 T = True
193             if T == True:
194                 if line.strip() != cc and line.strip() not in ls:
195                     f2.write(line)
196 
197             else:
198                 for i, v in enumerate(ls, 1):
199                     print ('%s: %s' % (i, v))
200                 while T1:
201                     v1 = input('请输入要修改或删除的编号')
202                     if v1.isdecimal() and int(v1) in range(len(ls)+1):
203                         break
204                     else:
205                         print('输入有误，请重新输入')
206                         continue
207                 x = input('请选择 1 = 修改 其他键 = 删除')
208                 if x == '1':
209                     while T1:
210                         a1 = check_sr()
211                         a2 = input('weight:')
212                         a3 = input('maxconn:')
213                         if a2.isdecimal() and a3.isdecimal():
214                             t1 = 'server %s %s weight %s maxconn %s' % (a1, a1, a2, a3)
215                             print('backend: %s 
%s' % (t, t1))
216                         else:
217                             print('所输入weight 和 maxconn 必须为数字，请重新输入')
218                             continue
219                         li_sr = 'server %s %s weight %s maxconn %s' % (a1, a1, a2, a3)
220                         # server
221                         if li_sr in ls:
222                             tt = input('server已存在，请按 y = 重新输入，或按 other key = 返回')
223                             if tt == 'y':
224                                 continue
225                             else:
226                                 break
227                         else:
228                             a4 = input('请确认信息正确，y = 确认, other key  = 重新输入')
229 
230                             if a4 == 'y':
231                                 f2.write(' '*8 + li_sr +'
')
232                             else:
233                                 continue
234                             print('修改成功')
235                             break
236                     cc = ls[int(v1)-1]
237                     del ls[int(v1)-1]
238                     for tt in ls:
239                         f2.write(' '*8 + tt +'
')
240                     T = True
241                 else:
242                     print('删除成功')
243                     cc = ls[int(v1)-1]
244                     del ls[int(v1)-1]
245                     for tt in ls:
246                         f2.write(' '*8 + tt +'
')
247                     T = True
248 
249 
250 def caidan ():
251     T = True
252     while T:
253         ls = input('''请输入所需编号：
254         1 查看backend下sever信息
255         2 添加记录
256         3 修改或删除backend
257         4 修改或删除sever
258         5 退出
259         ''')
260         if ls == '1':
261             i = check_bk()
262             l1 = find(i.strip())
263             for a in l1:
264                 print (a)
265         elif ls == '2':
266             while T:
267                 t = input('请输入backend信息')
268                 print('请输入sever信息')
269                 cont = 0
270                 while T:
271                     a1 = input('sever:')
272                     ti = a1.split('.')
273                     for ii in ti:
274                         if ii.isdecimal() and int(ii) in range(0, 255) and len(ti) == 4:
275                             cont = cont + 1
276                             continue
277                         else:
278                             print('输入有误，请输入由四组0-255组成的IP地址')
279                             break
280                     if cont == 4:
281                         break
282                 while T:
283                     a2 = input('weight:')
284                     a3 = input('maxconn:')
285                     if a2.isdecimal() and a3.isdecimal():
286                         t1 = 'server %s %s weight %s maxconn %s' % (a1, a1, a2, a3)
287                         print('backend: %s 
%s' % (t, t1))
288                         break
289                     else:
290                         print('所输入weight 和 maxconn 必须为数字，请重新输入')
291                         continue
292                 t3 = input('信息确认请按 y, 重新输入请按其他键')
293                 if t3.lower() == 'y':
294                     add (t, t1)
295                     print('添加成功')
296                     break
297                 else:
298                     continue
299         elif ls == '3':
300             update_bk()
301         elif ls == '4':
302             update_sr()
303         elif ls == '5':
304             exit()
305         else:
306             print('输入有误，请重新输入')
307 
308 caidan()

def find(backend):
    '''
    查看backend下sever信息
:param backend:
:return:
    '''
ls = []
    with open('HA.txt', 'r') as f:
        T = True
        for line in f:
            # 找到第一个backend，做记录 T = False
if line.strip() == 'backend %s' % (backend):
                T = False
                continue
# 找到第二个backend，做记录 T = True
if T == False and line.strip().startswith('backend'):
                T = True
                break
# 把关于False的加入到列表中
if T == False and line.strip():
                ls.append(line.strip())
    return ls


def check_bk():
    '''
    列出backend的信息
:return: 选中的backend
    '''
ls = []
    T = True
    with open('HA.txt', 'r') as f1:
         for line in f1:
            if line.strip().startswith('backend') == True:
                to = line.split(' ')
                ls.append(to[1])
         for i, v in enumerate(ls,1):
            print (i, v)
         while T:
             t1 = input('请输入查看的server的 backend信息')
             if t1.isdecimal() and int(t1) in range(0,len(ls)+1):
                a = ls[int(t1)-1]
                break
             else:
                print('输入有误，重新输入')
    return a

def backup():
    '''
    备份
:return:
    '''
with open('HA.txt', 'r') as f1, open('HA_old.txt', 'w') as f2:
        for line in f1:
            f2.write(line)


def add(bk, sr):
    '''
    添加记录
:param bk:
:param sr:
:return:
    '''
r_list = find(bk)
    # 没有关于bk的记录，直接添加内容
T = True
a = 'backend %s' % bk
    backup()
    if not r_list:
        with open('HA.txt', 'a') as f1:
            f1.write('
%s
'% a)
            f1.write("        %s
" % sr)
    # 有关于bk的记录
else:
        # 没有sever记录
backup()
        if sr not in r_list:
            with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
                for line in f1:
                    if line.strip() == 'backend %s' % (bk):
                        T = False
f2.write(line)
                        continue
                    if T == False and line.strip().startswith('backend'):
                        T = True
                    if T == True:
                        f2.write(line)
                    else:
                        if sr not in r_list:
                            r_list.append(sr)
                            t = '
        '.join(r_list)
                            f2.write(' '* 8 + t + '

')
    # 有sever记录,直接复制
else:
            print('记录已存在，请重新选择')
            caidan()


def update_bk ():
    '''
    需要修改或删除backend
:return:
    '''
backup()
    T = True
bk = check_bk()
    ls =[]
    with open('HA.txt', 'r') as f1:
         for line in f1:
            if line.strip().startswith('backend') == True:
                to = line.split(' ')
                ls.append(to[1].strip())
    c = input('''
    请选择需要选项：
    1 修改backend
    2 删除backend
    ''')
    while T:
        #修改backend信息
if c == '1':
            bk1 = input('请输入修改后的backend信息')
            t3 = input('信息确认请按 y, 重新输入请按其他键')
            if bk1 in ls:
                print('信息已存在，请重新选择')
                break
            else:
                if t3.lower() == 'y':
                    with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
                        for line in f1:
                            #需要修改的backend
if line.strip() == 'backend %s' % (bk.strip()):
                                f2.write('backend %s 
' % (bk1))
                            else:
                                f2.write(line)
                        print('修改成功')
                        break
                else:
                    continue
#删除backend信息
elif c == '2':
            t = find(bk.strip())
            while T:
                with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
                    for line in f1:
                        if line.strip() == 'backend %s' % (bk.strip()):
                            continue
                        if line.strip() in t:
                            continue
                        else:
                            f2.write(line)
                    print('删除成功')
                    break
            break

def check_sr():
    T1 = True
    while T1:
        cont = 0
a1 = input('sever:')
        ti = a1.split('.')
        for ii in ti:
            if ii.isdecimal() and int(ii) in range(0, 255) and len(ti) == 4:
                cont = cont + 1
continue
            else:
                print('输入有误，请输入由四组0-255组成的IP地址')
                break
        if cont == 4:
            break
    return a1


def update_sr():
    '''
    修改或删除sever
:return:
    '''
t = check_bk()
    backup()
    T = True
T1 = True
ls = find(t.strip())
    cc = 0
with open('HA_old.txt', 'r') as f1, open('HA.txt', 'w') as f2:
        for line in f1:
            if line.strip() == 'backend %s' %(t.strip()):
                T = False
f2.write(line)
                continue
            if T == False and line.strip().startswith('backend'):
                T = True
            if T == True:
                if line.strip() != cc and line.strip() not in ls:
                    f2.write(line)

            else:
                for i, v in enumerate(ls, 1):
                    print ('%s: %s' % (i, v))
                while T1:
                    v1 = input('请输入要修改或删除的编号')
                    if v1.isdecimal() and int(v1) in range(len(ls)+1):
                        break
                    else:
                        print('输入有误，请重新输入')
                        continue
x = input('请选择 1 = 修改 其他键 = 删除')
                if x == '1':
                    while T1:
                        a1 = check_sr()
                        a2 = input('weight:')
                        a3 = input('maxconn:')
                        if a2.isdecimal() and a3.isdecimal():
                            t1 = 'server %s %s weight %s maxconn %s' % (a1, a1, a2, a3)
                            print('backend: %s 
%s' % (t, t1))
                        else:
                            print('所输入weight 和 maxconn 必须为数字，请重新输入')
                            continue
li_sr = 'server %s %s weight %s maxconn %s' % (a1, a1, a2, a3)
                        # server
if li_sr in ls:
                            tt = input('server已存在，请按 y = 重新输入，或按 other key = 返回')
                            if tt == 'y':
                                continue
                            else:
                                break
                        else:
                            a4 = input('请确认信息正确，y = 确认, other key  = 重新输入')

                            if a4 == 'y':
                                f2.write(' '*8 + li_sr +'
')
                            else:
                                continue
print('修改成功')
                            break
cc = ls[int(v1)-1]
                    del ls[int(v1)-1]
                    for tt in ls:
                        f2.write(' '*8 + tt +'
')
                    T = True
                else:
                    print('删除成功')
                    cc = ls[int(v1)-1]
                    del ls[int(v1)-1]
                    for tt in ls:
                        f2.write(' '*8 + tt +'
')
                    T = True


def caidan ():
    T = True
    while T:
        ls = input('''请输入所需编号：
        1 查看backend下sever信息
        2 添加记录
        3 修改或删除backend
        4 修改或删除sever
        5 退出
        ''')
        if ls == '1':
            i = check_bk()
            l1 = find(i.strip())
            for a in l1:
                print (a)
        elif ls == '2':
            while T:
                t = input('请输入backend信息')
                print('请输入sever信息')
                cont = 0
while T:
                    a1 = input('sever:')
                    ti = a1.split('.')
                    for ii in ti:
                        if ii.isdecimal() and int(ii) in range(0, 255) and len(ti) == 4:
                            cont = cont + 1
continue
                        else:
                            print('输入有误，请输入由四组0-255组成的IP地址')
                            break
                    if cont == 4:
                        break
                while T:
                    a2 = input('weight:')
                    a3 = input('maxconn:')
                    if a2.isdecimal() and a3.isdecimal():
                        t1 = 'server %s %s weight %s maxconn %s' % (a1, a1, a2, a3)
                        print('backend: %s 
%s' % (t, t1))
                        break
                    else:
                        print('所输入weight 和 maxconn 必须为数字，请重新输入')
                        continue
t3 = input('信息确认请按 y, 重新输入请按其他键')
                if t3.lower() == 'y':
                    add (t, t1)
                    print('添加成功')
                    break
                else:
                    continue
        elif ls == '3':
            update_bk()
        elif ls == '4':
            update_sr()
        elif ls == '5':
            exit()
        else:
            print('输入有误，请重新输入')

caidan()