s13 day3作业

ha_proxy配置文件修改程序
ha_file 为存储配置信息的文件。运行的时候对该文件进行操作。
1.查询信息：用户输入域名，获得域名相关信息
2.修改配置文件：用户输入的格式应该为 {"backend": "test.oldboy.org","record":{"server": "100.1.7.9","weight": 20,"maxconn": 30}}
然后执行判断，如果输入的backend存在，则判断对应的信息是否重复，如果重复，则提醒用户，同时对配置文件不做更改。
如果输入的信息与原配置信息不符，则追加到相应的backend下。
如果输入的backend不存在，则追加到文件末尾。
3.删除信息：用户输入存在的域名，则删除域名和相应的配置文件。如果不存在，则提示错误。

下面是ha_file文件信息

 global
         log 127.0.0.1 local2
         daemon
         maxconn 256
         log 127.0.0.1 local2 info
 defaults
         log global
         mode http
         timeout connect 5000ms
         timeout client 50000ms
         timeout server 50000ms
         option  dontlognull

 listen stats :8888
         stats enable
         stats uri       /admin
         stats auth      admin:1234

 frontend oldboy.org
         bind 0.0.0.0:80
         option httplog
         option httpclose
         option  forwardfor
         log global
         acl www hdr_reg(host) -i www.oldboy.org
         use_backend www.oldboy.org if www

 backend www.oldboy.org
         server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000
         server 100.1.7.9 100.1.7.9 weight 20 maxconn 300
         server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000

ha_file

下面是主要程序代码，

 #!/use/bin/env python
 #_*_ coding:utf_8 _*_
 import json,time,os,sys

 time_now=time.strftime('%Y-%m-%d %H:%M:%S')
 time_flag=time.strftime('%Y-%m-%d-%H-%M-%S')

 if __name__ =='__main__':
     """
     主函数，通过判断用户输入执行相应操作
     :return:
     """
     while True:
         print(
             '''
 ---------------------------------------
     请输入下列编号进行操作,按q退出：
         1：获取haproxy记录
         2：增加haproxy记录
         3: 删除haproxy记录
 ----------------------------------------
          ''')
         get_mun = input('请输入操作序号：')
         if get_mun == 'q':
             exit('bye')
         if get_mun.isdigit:
             get_mun = int(get_mun)
         if get_mun == 1: #如果选择1，则执行get_ha()
             read = input("请输入backend：（例如：www.oldboy.org）")
             ha_list = get_ha(read)
             if len(ha_list)!= 0:
                 print(ha_list)
             else:
                 print('没有找到相应记录，请检查输入是否正确')
         elif get_mun == 2:
             add_ha()
         elif get_mun == 3:
             del_domain = input('请输入想要删除的值：')
             del_ha(del_domain)

main()

main函数调用其他几个函数，分别为查询记录，增加记录和删除记录

定义查询函数

 def get_ha(get_read):
     """
     定义查询函数
     :param get_read:用户输入想查询的信息
     :return:返回查询结果
     """
     get_flags = 0 #设置标识符，当匹配到输入时置为1
     get_list = [] #将匹配到的内容放在列表中
     with open('ha_file','r',encoding='utf-8') as f:
         for line in f.readlines():
             line = line.strip() #去掉多余的空格和换行符
             if line == 'backend %s'%get_read:
                 get_flags = 1
             elif line.startswith('backend'):
                 get_flags = 0
             elif get_flags == 1 and len(line)!= 0 :
                 get_list.append(line)
     return get_list

get_ha()

定义增加函数

 def add_ha():
     while True:
         add_resource_str = input('请输入想要增加的记录： ')
         result_check = check_input(add_resource_str)
         if result_check == False:
             print('请输入正确格式')
         if result_check == True:
             add_resource = json.loads(add_resource_str) #取得输入的值
             domain_info = add_resource['backend']
             server_info = add_resource['record']['server']
             weight_info = add_resource['record']['weight']
             maxcont_info = add_resource['record']['maxconn']
             match_info = ('server {} {} weight {} maxconn {}').format(server_info,server_info,weight_info,maxcont_info)
             #match_info 判断信息是否存在，不存在则添加进文件
             # print(match_info,type(match_info))
             # print(domain_info)
             get_match = get_ha(domain_info) #原文件已经存在的记录
             # print(get_match[0])
             if get_match != []:#域名信息存在
                 if get_match[0] == match_info:
                     print('信息已存在,不做更改')
                 else:
                     #域名信息存在，但是有更新，f1为原文件，f2为备份文件
                     with open('ha_file','r',encoding='utf-8') as f1:
                         all_config = f1.readlines()
                     match_item = 'backend {}\n'.format(domain_info) #获得更新后的配置信息
                     #print(match_item)
                     if str(match_item) in all_config:
                         update_index = all_config.index(str(match_item))
                         insert_record_index = int(update_index) +1
                         all_config.insert(int(insert_record_index),8*' '+match_info+'\n') #将数据插入
                         with open('ha_new','w',encoding='utf-8') as f2:
                             f2.writelines(all_config)
                         os.rename('ha_file','ha_bak%s'%(time_flag))
                         os.rename('ha_new','ha_file')
             else:  #原文件不存在相同域名，追加到文件末尾
                 with open('ha_file','r',encoding='utf-8') as f1,open('ha_new','w',encoding='utf-8') as f2:
                     all_config = f1.readlines()
                     str_append = ('''
 backend %s
         %s
         '''%(domain_info,match_info))# 字符串预格式化
                     all_config.append(str_append)
                     f2.writelines(all_config)
                     os.rename('ha_file','ha_bak')
                     os.rename('ha_new','ha_file')
         return

add_ha

定义删除函数

 def get_ha(get_read):
     """
     定义查询函数
     :param get_read:用户输入想查询的信息
     :return:返回查询结果
     """
     get_flags = 0 #设置标识符，当匹配到输入时置为1
     get_list = [] #将匹配到的内容放在列表中
     with open('ha_file','r',encoding='utf-8') as f:
         for line in f.readlines():
             line = line.strip() #去掉多余的空格和换行符
             if line == 'backend %s'%get_read:
                 get_flags = 1
             elif line.startswith('backend'):
                 get_flags = 0
             elif get_flags == 1 and len(line)!= 0 :
                 get_list.append(line)
     return get_list

del_ha

由于本次为手动输入记录，对格式有严格的要求，所以定义一个检查输入的函数。实际环境中应该是从其他地方get到格式化好的值。

 def check_input(check_dir):
     """
     检查输入格式是否正确
     :param check_str:
     :return:
     """
     try:
         check = json.loads(check_dir)
         domain_info = check['backend']
         server_info = check['record']['server']
         weight_info = check['record']['weight']
         maxcont_info = check['record']['maxconn']

     except:
         return False
     else:
         return True

check_input