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('\t%s' % add_info[0])
 24     for j in search_get:
 25         if j == add_info[1]:
 26             print('\t\t\033[1;31m%s\033[0m' % j)
 27         else:
 28             print('\t\t%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\n' % 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('\033[1;31m%s %s ip已删除!\033[0m当前server信息如下:' % (add_info[0], add_info[1]))
 69                 show_info()                                                      # 显示
 70             else:
 71                 continue
 72         else:
 73             print('没有网址%s的server信息' % input_net)
 74             continue
 75         # 查看功能,方便删除server
 76     elif input_choice == 'm':
 77         input_info = input('''
 78 请输入需要添加或者修改的信息,
 79 例如:backend www.oldboy.org,server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000
 80 >>>''')
 81         add_info = input_info.split(',')                                # 按统一格式处理输入信息
 82         add_net = '%s\n' % add_info[0]                                  # 调整输入信息为data列表元素格式
 83         add_server = '        %s\n' % add_info[1]                       # 调整输入信息为data列表元素格式
 84         if add_net in data:
 85             search_get = get_server_list(add_net)                       # 获取server列表
 86             begin = data.index(add_net)                                 # 提取网址在data中的位置
 87             server_list = [k.split() for k in search_get if k != '']   # 调整server列表,可读取ip
 88             ip_list = [k2[1] for k2 in server_list if k2 != '']        # 提取ip列表
 89             server_info = add_info[1].split()                           # 调整输入信息各元素可提取
 90             ip = server_info[1]                                         # 提取输入ip值
 91             weight = server_info[4]                                     # 提取输入weight值
 92             maxconn = server_info[6]                                    # 提取输入maxconn值
 93             if ip in ip_list:
 94                 index_ip = ip_list.index(ip)
 95                 # 确定ip在ip_list的位置,对应的也是ip在server_list的位置,同时也是ip在search_get的位置
 96                 if weight == server_list[index_ip][4] and maxconn == server_list[index_ip][6]:
 97                     print('需添加的\033[1;31m%s %s ip已存在!各值无变化!\033[0m当前server信息如下:'
 98                           % (add_info[0], add_info[1]))
 99                     show_info()                        # 显示
100                     continue
101                     # 网址以及ip都存在,weight和maxconn值无变化,则显示ip信息
102                 else:
103                     the_ip = begin+index_ip+1            # 确定需要修改的ip在data的位置,由于server_list和server_get位置顺序一致
104                     data_bak = data                      # 设置备份列表data_bak,准备输出
105                     data[the_ip] = add_server            # 修改内容,准备输出
106                     search_get[index_ip] = add_info[1]   # 更新显示列表search_get,准备显示
107                     output_write()                       # 输出
108                     print('需添加的\033[1;31m%s %s ip已存在!各值已更新!\033[0m当前server信息如下:'
109                           % (add_info[0], add_info[1]))
110                     show_info()                          # 显示
111                     continue
112                     # 网址以及ip都存在,weight或maxconn的值有变化,更新值并显示ip信息
113             else:
114                 data_bak = data                                              # 设置备份列表data_bak,准备输出
115                 insert_index = begin+len(search_get)                         # 在data中设置插入位置
116                 data = data[:insert_index]+[add_server]+data[insert_index:]  # data中插入内容,准备输出
117                 search_get = search_get[:-1]+[add_info[1]]+search_get[-1:]   # 更新显示列表search_get,准备显示
118                 output_write()                                               # 输出
119                 print('需添加的\033[1;31m%s %s 已加入配置!\033[0m当前IP信息如下:' % (add_info[0], add_info[1]))
120                 show_info()                                                  # 显示
121                 continue
122                 # 网址存在,ip不存在,则添加ip信息
123         else:
124             data_bak = data                          # 设置备份列表data_bak
125             data.append(add_net)                     # 更新输出列表data
126             data.append(add_server)                  # 更新输出列表data
127             search_get = [add_info[1]]               # 更新显示列表search_get
128             output_write()                           # 输出
129             print('需添加的\033[1;31m%s %s 已加入配置!\033[0m当前IP信息如下:' % (add_info[0], add_info[1]))
130             show_info()                              # 显示
131             continue
132             # 网址不存在,添加网址及ip信息