Day03 - 作业

作业：HAproxy配置文件操作

一、作业要求

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

二、简单思路

1. 用函数功能实现查询、添加、修改、删除和备份功能。
2. 每次增删改前都先进行备份。
3. 文件在已有内容的修改方面有困难点，需要查询有什么方法实现。

三、流程图

Day03 - HAproxy配置文件操作

四、参考站点

1. day3 作业详细
2. haproxy配置文件

五、代码

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

import time, json, os

def write_file(backend_title, exist_res):   # 向文件写入 backend 和 server 信息
    with open('haproxy', 'r', encoding='utf-8') as f, open('haproxy_new', 'w', encoding='utf-8') as f2:
        write_flag = False
        for line in f:
            if line.strip() == 'backend %s' % backend_title:    # 如果该行的内容为匹配，说明从此往后的内容为新内容
                write_flag = True
                f2.write(line)
                for i in exist_res:
                    f2.write(' '*8 + i + '\n')
                continue
            elif line.strip().startswith('backend') and write_flag:  # 如果以backend开头且write_flag为真，说明新加信息已写完
                write_flag = False
                f2.write(line)
                continue
            elif write_flag and line.strip():   # 如果write_flag为真且该行有内容，说明是不需写入的旧信息。
                continue
            else:   # 正常写入
                f2.write(line)

def search(search_input):   # 按 backend 查询
    search_res = []
    with open('haproxy', 'r', encoding='utf-8') as f:
        check_flag = False
        for line in f:
            if line.strip() == 'backend %s' % search_input:  # 如果该行与'backend %s' % search_input相同
                check_flag = True
                continue
            elif line.strip().startswith('backend') and check_flag:  # 该行仅以'backend'开头，并且check_flag为真
                check_flag = False
                break
            elif check_flag and line.strip():  # 如果check_flag为真且该行有内容，说明是需查询到的server信息。
                search_res.append(line.strip())
    if not search_res:
        print("文件中没有 %s 的信息！" % search_input)
        return search_res
    else:
        return search_res

def add(backend_title, backend_info):
    exist_res = search(backend_title)   # 先查询输入记录是否已经存在
    if not exist_res:  # 输入的backend和server信息均不存在
        backup()
        with open('haproxy', 'r', encoding='utf-8') as f, open('haproxy_new', 'w', encoding='utf-8') as f2:
            for line in f:
                f2.write(line)
            f2.write('\n\nbackend %s\n' % backend_title)  # 开始添加新的backend和server信息
            f2.write(' '*8 + backend_info)  # 字符也可以用乘法，刷新认知了。
            print("新的 backend 和 server 信息已添加进入文件！")
        os.rename('haproxy_new', 'haproxy')  # 重命名文件
    else:   # 输入的backend信息已存在
        if backend_info in exist_res:   # 输入的server记录也已存在，即重复了
            print("您要添加的 backend 和 server 信息已经存在！")
        else:   # 输入的server记录为新记录
            backup()
            exist_res.append(backend_info)
            write_file(backend_title, exist_res)    # 调用write_file函数
            print("新的 server 信息已添加进入文件！")
            os.rename('haproxy_new', 'haproxy')  # 重命名文件

def delete(backend_title, backend_info):
    exist_res = search(backend_title)   # 先查询输入记录是否已经存在
    if not exist_res:  # 输入的backend和server信息均不存在
        return
    else:   # 输入的backend信息已存在
        if backend_info not in exist_res:   # 输入的server记录不存在
            print("%s 中没有您要删除的 server 信息！" % backend_title)
        else:   # 输入的server记录存在
            backup()
            exist_res.remove(backend_info)
            with open('haproxy', 'r', encoding='utf-8') as f, open('haproxy_new', 'w', encoding='utf-8') as f2:
                write_flag = False
                for line in f:
                    if line.strip() == 'backend %s' % backend_title:  # 如果该行与'backend %s' % backend_title相同
                        write_flag = True
                        if not exist_res:  # 如果exist_res列表为False，则跳过不写入
                            continue
                        else:  # 如果exist_res列表不为空，则写入已有内容
                            f2.write(line)
                            for i in exist_res:
                                f2.write(' '*8 + i + '\n')
                        continue
                    elif line.strip().startswith('backend') and write_flag:  # 该行仅以'backend'开头，并且write_flag为真
                        write_flag = False
                        f2.write(line)
                        continue
                    elif write_flag and line.strip():  # write_flag为真且不为空行
                        continue
                    else:
                        f2.write(line)
            print("backend 和 server 信息已删除！")
            os.rename('haproxy_new', 'haproxy')  # 重命名文件

def replace(backend_title, backend_info):
    exist_res = search(backend_title)
    if not exist_res:  # 根据server信息列表真假值，来判断要修改的backend是否存在于文件中。
        return
    elif backend_info in exist_res:  # 要修改的backend和server信息，文件中均已存在。
        print("文件中已经含有该条 backend 和 server 信息！")
        return
    else:
        backup()
        for server_info in enumerate(exist_res):
            print(server_info[0], server_info[1])
    server_num = input("请选择在 %s 中您要修改的记录：" % backend_title)
    while True:  # 判断选择是否正确，不然总是报错
        if server_num.isdigit():
            server_num = int(server_num)
            if server_num > len(exist_res):
                print("请输入正确的数字！")
            else:
                exist_res[server_num] = backend_info
            break
        else:
            print("请输入正确的格式！")
    write_file(backend_title, exist_res)
    print("已将 %s 中的信息修改为 %s" % (backend_title, exist_res[server_num]))
    os.rename('haproxy_new', 'haproxy')

def backup():  # 文件操作前先备份已有文件
    backup_time = time.strftime('%Y%m%d_%H%M%S')
    with open('haproxy', 'r', encoding='utf-8') as f,\
            open('haproxy_%s.bak' % backup_time, 'w', encoding='utf-8') as f2:
        for line in f:
            f2.write(line)
    return "原配置已备份为 haproxy_%s.bak" % backup_time

while True:
    print("1. 查询 backend 信息\n2. 添加 backend 和 server 信息\n3. 修改 backend 和 server 信息\n"
          "4. 删除 backend 和 server 信息\nq. 退出")
    choice_input = input("\n请输入您想进行的操作:").strip()
    if choice_input.isdigit():
        choice_input = int(choice_input)
        if choice_input == 1:
            user_input = input("请输入您想查询的 backend 信息:").strip()
            for info in search(user_input):
                print(info)
        elif choice_input in [2, 3, 4]:
            print('\n请以字典形式输入 backend 和 server 信息。\n例如：{"backend": "test.oldboy.org","record":'
                  '{"server": "100.1.7.9","weight": 20,"maxconn": 30}}\n')
            user_input = input("请输入您想进行操作的 backend 和 server 信息：").strip()
            backend_dict = json.loads(user_input)
            backend_title = backend_dict['backend']
            backend_info = 'server %s %s weight %d maxconn %d' % (backend_dict['record']['server'],
                                                                  backend_dict['record']['server'],
                                                                  backend_dict['record']['weight'],
                                                                  backend_dict['record']['maxconn'])
            if choice_input == 2:
                add(backend_title, backend_info)
            elif choice_input == 3:
                replace(backend_title, backend_info)
            elif choice_input == 4:
                delete(backend_title, backend_info)
        else:
            print("请输入正确的数字！")
    elif choice_input == 'q':
        exit("Bye！")
    else:
        print("请输入正确的格式！")