Day3作业：ha_proxy配置文件修改

不废话，上代码

readme:

# Auther:ccorz Mail:ccniubi@163.com Blog:http://www.cnblogs.com/ccorz/
# GitHub:https://github.com/ccorzorz

ha_proxy配置文件修改程序
1.查询时输入域名即可查询，显示域名相关的backend配置
2.删除时输入域名，即可删除包括域名以及配置服务器信息的相关内容；如相关域名信息，会提示用户，不改动文件
3.修改时需要输入列表信息，列表中需要包括backend的域名信息，以及包含weight、server、maxconn的record列表
    3.1 修改需输入内容举例：{"backend": "test.oldboy.org","record":{"server": "100.1.7.9","weight": 20,"maxconn": 30}}
    3.2 程序会自动判断输入的格式是否正确，如果不正确，会提醒用户重新输入，直至正确格式
    3.3 程序会自动判断域名信息是否存在，如存在，增加服务器信息;如不存在，追加至配置文件结尾
    3.4 程序会自动判断用户想增加的服务器信息是否与已存在的配置重叠，如存在，提示用户，不修改文件；如不存在，在域名相关信息后追加
4.程序会自动备份修改之前的文件，后缀为修改时间
5.本程序有日志记录，以便管理员查询修改记录

流程图：

ha_proxy配置文件：

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 30

backend buy.oldboy.org
        server 100.1.7.90 100.1.7.90 weight 20 maxconn 3000

程序代码：

#!/usr/bin/env python
# -*-coding=utf-8-*-
# Auther:ccorz Mail:ccniubi@163.com Blog:http://www.cnblogs.com/ccorz/
# GitHub:https://github.com/ccorzorz

import json,time,os

time_now=time.strftime('%Y-%m-%d %H:%M:%S')
time_flag=time.strftime('%Y-%m-%d-%H-%M-%S')
#定义通过域名查询ha配置文件函数
def search_backend(search_info):
    """
    :param search_info: 用户输入的需要查询的域名
    :return: result_list :结果列表,供增删改使用
    """
    b_flag=1    #设置初始标识符
    result_list=[]  #设置查询结果为空列表
    with open('ha','r') as ha_read:
        for line in ha_read.readlines():    #将内容通过readlines转化为列表
            line=line.strip()   #每一行去掉两头的空格和回车
            if line=='backend %s'%search_info:  #发现内容匹配
                b_flag=0    #更改标识符值
            elif line.startswith('backend'):    #发现以'backend'开头的内容
                b_flag=1    #更改标识符为初始值
            elif b_flag==0 and  len(line)!=0:   #将标识符为0的非空内容加入查询结果列表
                result_list.append(line)
    return result_list

def add_backend():
    """
    :return: None
    """
    while True:
        add_contect_str=input('Input the server_info:') #提示用户输入配置文件信息
        res=enter_query(add_contect_str)    #将确认格式函数的返回值赋予res变量
        if res==False:  #如果变量为Flase,提醒用户输入的字符串格式不正确
            print('String format error,try again!')
        if res == True:     #如果变量值为True,即用户输入格式正确，执行以下
            # print(result_list)
            add_contect=json.loads(add_contect_str) #通过json模块，将字符串转化为列表
            search_info=add_contect['backend']  #变量赋值
            weight=add_contect['record']['weight']
            server=add_contect['record']['server']
            maxconn=add_contect['record']['maxconn']
            server_info='server %s %s weight %s maxconn %s'%(server,server,weight,maxconn)
            result_list=search_backend(search_info)     #将字符串中解析得到的列表赋值于result_list
            print(result_list)
            if len(result_list)==0:     #如果无对应的域名，新增域名以及对应的记录
                with open('ha','r') as ha_read,open('ha_new','w') as ha_wirte:
                    con_list=ha_read.readlines()        #通过readlines将ha文件中的内容转化为列表
                    con_list.append('\n')   #列表中加一空行
                    con_list.append('backend %s\n'%search_info)     #列表中新增域名信息
                    con_list.append('%s%s'%(8*' ',server_info))     #列表中追加域名对应服务器信息
                    ha_wirte.writelines(con_list)       #将列表通过writelines全部写入ha_write文件
                #回显，提示用户无域名信息，将新增信息到配置文件最后
                print('No domain target,will add to the end of the "ha",has been add the end of file..')
                time.sleep(1)
                print('\033[31;1mInformation has been added successly!!!!\033[0m')
                os.rename('ha', 'ha_bak.%s' % time_flag)  # 系统改名
                os.rename('ha_new', 'ha')
                with open('modify.log', 'a') as log:    #写入日志文件
                    log.write('%s domain %s add server info: %s\n' % (time_now, search_info, server_info))
            else:       #列表不为空
                if server_info in result_list:  #如果服务器信息重叠，提示用户信息已重叠，不操作文件
                    print('\033[31;1mThe server info had been in %s domain,do not need to add...\033[0m'%search_info)
                else:
                    result_list.append(server_info)     #服务器信息不重叠
                    with open('ha','r') as ha_read,open('ha_add','w') as ha_add:
                        a_flag=1    #设置取配置文件中目标内容的标识符
                        exit2_flag=1    #设置限制目标内容循环的标识符
                        for line in ha_read.readlines():    #转化列表，遍历列表
                            if line.strip()=='backend %s'%search_info:  #如果其中一行的内容为“backend 域名名称”,更改目标内容标识符的值
                                a_flag=0
                            # 更改目标内容标识符后，继续遍历列表，发现以backend开头的行，再次更改目标内容标识符的值，
                            # 两个标识符中间的内容为即将修改的目标内容
                            elif line.strip().startswith('backend'):
                                a_flag=1
                            if a_flag==1:   #如果非目标内容写入ha_add文件
                                ha_add.write(line)
                            elif a_flag==0 and exit2_flag==1:   #如果是目标内容，并且限制循环的标识符为初始值
                                ha_add.write('backend %s\n'%search_info)    #添加一行域名信息
                                for line in result_list:    #遍历即将修改的列表，在元素内容前加入8个空格
                                    num=result_list.index(line)
                                    result_list[num]='%s%s\n'%(8*' ',line)
                                ha_add.writelines(result_list)  #将列表中内容写入ha_add文件
                                ha_add.write('\n')  #添加一行空白行
                                exit2_flag=0    #更改限制循环标识符的初始值，使其只循环一次，避免多次写入列表中的服务器信息
                    os.rename('ha','ha_bak.%s'%time_flag)   #系统改名
                    os.rename('ha_add','ha')
                    with open('modify.log','a') as log: #写入修改日志文件
                        log.write('%s %s add server info: %s\n'%(time_now,search_info,server_info))
                    print('\033[31;1mModify success!!!!\033[0m')    #回显
            break

#定义删除函数
def del_backend(del_domain):
    with open('ha','r') as ha_read,open('ha_del','w') as ha_del:
        b_flag=1    #设置目标内容初始值
        for line in ha_read.readlines():    #将内容通过readlines转化为列表
            if line.strip()=='backend %s'%del_domain:  #发现内容匹配
                b_flag=0    #更改标识符值
            elif line.strip().startswith('backend'):    #发现以'backend'开头的内容
                b_flag=1    #更改标识符为初始值
            if b_flag==1:   #只将非目标内容写入ha_del文件
                ha_del.write(line)
    with open('modify.log', 'a') as log:    #操作日志写入日志文件
        log.write('%s delete the domain %s and the related info.\n' % (time_now, del_domain))
    os.rename('ha','ha_bak.%s'%time_flag)   #系统改名ha以及操作文件
    os.rename('ha_del','ha')
    print('域名以及域名信息已全部删除...')

#定义确认用户输入格式是否为ha配置文件所需格式函数
def enter_query(add_contect_str):
    """
    :param add_contect_str: 用户选择修改配置文件后，输入的格式必须为字典
    :return: Flase=用户输入格式不正确  True=用户输入格式正确
    """
    try:    #尝试将用户输入字符串转化为列表，并给变量赋值
        add_contect=json.loads(add_contect_str)
        search_info=add_contect['backend']
        weight=add_contect['record']['weight']
        server=add_contect['record']['server']
        maxconn=add_contect['record']['maxconn']
    except: #如果异常，return 值为False
        return False
    else:   #如果无报错，return 值为False
        return True

#定义主函数
def main():
    while True:
        start_num=input("""1.查询
2.修改
3.删除
q.退出""")
        if start_num.isdigit():
            start_num=int(start_num)
            while True:
                if start_num==1:
                    search_info=input('Please enter the domain name:')
                    result_list=search_backend(search_info)
                    if len(result_list)==0: #如果配置文件中无查询的内容,回显
                        print('\033[31;1mSorry,the configuration file does not have the information you want to query!\033[0m')
                    else:   #如果配置文件中有要查询的内容,打印查询结果
                        for line in result_list:
                            print('\033[31;1m%s\033[0m'%line)
                    break
                elif start_num==2:
                    add_backend()
                    break
                elif start_num==3: #如果选择删除，提示用户输入所需删除的域名
                    del_domain=input('Input the domain want to delete:')
                    result_list=search_backend(del_domain)  #列表赋值于变量
                    if len(result_list)==0:     #如果列表为空，无域名相应信息，提示用户
                        print('\033[31;1mSorry,the configuration file does not have the information you want to delete!\033[0m')
                    else:   #如果有相应信息，执行删除函数
                        del_backend(del_domain)
                    break
                else:   #其他输入，提示用户输入错误，请重新输入
                    print('Information is incorrect, please re-enter...')
                    break
        elif start_num=='q':    #选择退出
            print('Bye!!!!')
            break
        else:   #输入无相应功能，提示重新输入
            print('Information is incorrect, please re-enter...')

#执行主函数
main()

日志内容：

2016-05-23 21:32:33 delete the domain www.oldboy.org and the related info.
2016-05-23 21:36:45 delete the domain www.oldboy.org and the related info.
2016-05-23 21:38:29 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 21:41:38 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 21:43:50 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 21:45:59 www.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 21:47:48 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 22:06:05 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 22:08:34 domain test.oldboy.org add server info: server 100.1.7.9 100.1.7.9 weight 20 maxconn 30
2016-05-23 22:19:40 delete the domain test.oldboy.org and the related info.

沛sir的demo：

#!/usr/bin/env python
# -*- coding:utf-8 -*-

def fetch(backend):
    # backend = "www.oldboy.org"
    result = []
    with open('ha.conf', 'r', encoding='utf-8') as f:
        flag = False
        for line in f:
            if line.strip().startswith("backend") and line.strip() == "backend " + backend:
                flag = True
                continue
            if flag and line.strip().startswith("backend"):
                flag = False
                break
            if flag and line.strip():
                result.append(line.strip())

    return result

# ret = fetch("www.oldboy.org")
# print(ret)

def add(backend, record):
    # 思路一：
    # 思路二：
    # 先检查记录存不存
    record_list = fetch(backend)
    if not record_list:
        # backend不存在
        with open('ha.conf', 'r') as old, open("new.conf", 'w') as new:
            for line in old:
                new.write(line)
            new.write("\nbackend " + backend + "\n")
            new.write(" " * 8 + record + "\n")
    else:
        # backend存在
        if record in record_list:
            # record已经存在
            # import shutil
            # shutil.copy("ha.conf", 'new.conf')
            pass
        else:
            # backend存在,record不存在
            record_list.append(record)
            with open('ha.conf', 'r') as old, open('new.conf', 'w') as new:
                flag = False
                for line in old:
                    if line.strip().startswith("backend") and line.strip() == "backend " + backend:
                        flag = True
                        new.write(line)
                        for new_line in record_list:
                            new.write(" "*8 + new_line + "\n" )
                        continue
                    if flag and line.strip().startswith("backend"):
                        flag = False
                        new.write(line)
                        continue
                    if line.strip() and not flag:
                        new.write(line)

bk = "www.oldboy.org"
rd = "server 100.1.7.49 100.1.7.49 weight 20 maxconn 30"
add(bk, rd)
# s = "[11,22,33,44,5, 66]"
# s = '{"k1":"v1"}'
# print(type(s),s)
#
# import json
# n = json.loads(s) # 将一个字符串，转换成python的基本数据类型; 注意：字符串形式的字典'{"k1":"v1"}'内部的字符串必须是 双引号
# print(type(n), n)
# import json
# r = input("input:")
# dic = json.loads(r)
# bk = dic['backend']
# rd = "server %s %s weight %d maxconn %d" %(dic['record']['server'],
#                                            dic['record']['server'],
#                                            dic['record']['weight'],
#                                            dic['record']['maxconn'])