Python学习（十一）—— 程序解耦、模块、包

  1 import os
  2 
  3 def file_handler(backend_data, res=None, type=‘fetch‘):
  4     if type == ‘fetch‘:
  5         with open(‘haproxy.conf‘, ‘r‘) as read_f:
  6             tag = False
  7             ret = []
  8             for read_line in read_f:
  9                 if read_line.strip() == backend_data:
 10                     tag = True
 11                     continue
 12                 if tag and read_line.startswith(‘backend‘):
 13                     # tag=False
 14                     break
 15                 if tag:
 16                     print(‘\033[1;45m%s\033[0m‘ % read_line, end=‘‘)
 17                     ret.append(read_line)
 18         return ret
 19     elif type == ‘change‘:
 20         with open(‘haproxy.conf‘, ‘r‘) as read_f,  21                 open(‘haproxy.conf_new‘, ‘w‘) as write_f:
 22             tag = False
 23             has_write = False
 24             for read_line in read_f:  # server
 25                 if read_line.strip() == backend_data:
 26                     tag = True
 27                     continue
 28                 if tag and read_line.startswith(‘backend‘):
 29                     tag = False
 30                 if not tag:
 31                     write_f.write(read_line)
 32                 else:
 33                     if not has_write:
 34                         for record in res:
 35                             write_f.write(record)
 36                         has_write = True
 37         os.rename(‘haproxy.conf‘, ‘haproxy.conf.bak‘)
 38         os.rename(‘haproxy.conf_new‘, ‘haproxy.conf‘)
 39         os.remove(‘haproxy.conf.bak‘)
 40 
 41 def fetch(data):
 42     # print(‘\033[1;43m这是查询功能\033[0m‘)
 43     # print(‘\033[1;43m用户数据是\033[0m‘,data)
 44     backend_data = ‘backend %s‘ % data
 45     return file_handler(backend_data)
 46 
 47 def add():
 48     pass
 49 
 50 def change(data):
 51     # print(‘这是修改功能‘)
 52     # print(‘用户输入的数据是‘,data)
 53     backend = data[0][‘backend‘]  # 文件当中的一条记录 www.oldboy1.org
 54     backend_data = ‘backend %s‘ % backend  # backend www.oldboy1.org
 55     #       server 2.2.2.4 2.2.2.4 weight 20 maxconn 3000
 56 
 57     old_server_record = ‘%sserver %s %s weight %s maxconn %s\n‘ % (‘ ‘ * 8, data[0][‘record‘][‘server‘],
 58                                                                    data[0][‘record‘][‘server‘],
 59                                                                    data[0][‘record‘][‘weight‘],
 60                                                                    data[0][‘record‘][‘maxconn‘])
 61 
 62     new_server_record = ‘%sserver %s %s weight %s maxconn %s\n‘ % (‘ ‘ * 8, data[1][‘record‘][‘server‘],
 63                                                                    data[1][‘record‘][‘server‘],
 64                                                                    data[1][‘record‘][‘weight‘],
 65                                                                    data[1][‘record‘][‘maxconn‘])
 66     print(‘用户想要修改的记录是‘, old_server_record)
 67     res = fetch(backend)  # fetch(‘www.oldboy1.org‘)
 68     print(‘来自change函数--》‘, res)
 69     if not res or old_server_record not in res:
 70         return ‘你要修改的记录不存在‘
 71     else:
 72         index = res.index(old_server_record)
 73         res[index] = new_server_record
 74 
 75     res.insert(0, ‘%s\n‘ % backend_data)
 76     file_handler(backend_data, res=res, type=‘change‘)
 77 
 78 def delete():
 79     pass
 80 
 81 if __name__ == ‘__main__‘:
 82     msg = ‘‘‘
 83     1:查询
 84     2:添加
 85     3:修改
 86     4:删除
 87     5:退出
 88     ‘‘‘
 89     msg_dic = {
 90         ‘1‘: fetch,
 91         ‘2‘: add,
 92         ‘3‘: change,
 93         ‘4‘: delete,
 94     }
 95 
 96     while True:
 97         print(msg)
 98         choice = input(‘请输入你的选项：‘).strip()
 99         if not choice: continue
100         if choice == ‘5‘: break
101 
102         data = input(‘请输入你的数据：‘).strip()
103 
104         if choice != ‘1‘:
105             data = eval(data)
106 
107         res = msg_dic[choice](data)
108         print(‘最终结果是--》‘, res)
109 # 修改的格式：[{‘backend‘:‘www.oldboy1.org‘,‘record‘:{‘server‘:‘2.2.2.4‘,‘weight‘:20,‘maxconn‘:3000}},{‘backend‘:‘www.oldboy1.org‘,‘record‘:{‘server‘:‘2.2.2.5‘,‘weight‘:30,‘maxconn‘:4000}}]