python day3 配置文件作业

1.需求

HAproxy配置文件操作

根据用户输入，输出对应的backend下的server信息
可添加backend 和sever信息
可修改backend 和sever信息
可删除backend 和sever信息
操作配置文件前进行备份

2.流程图

3.代码

#!/usr/bin/env python
# -*- coding:utf-8 -*-
#Author Kevin Liu
import json
import shutil

def input_check(s):
    """
    检查用户输入是否是数字
    :param s:用户输入
    :return:返回用户选择
    """
    if str.isdigit(s):
        s = int(s)
    return s

def fetch(backend):     #取出配置文件的匹配信息
    result = []  #把返回值取到这个列表
    with open("conf","r") as f:
        flag = False
        for line in f:    #一次读取一行文件内容
            #如果找到的不是空行而且是以"backend"开头，而且内容是"backend +用户输入，就是找到了配置项，"backend "这里有个空格
            if line.strip().startswith("backend") and line.strip() == "backend " + backend:
                flag = True
                #找到了backend 可以跳出本次循环，需要取出backend后面的配置项
                continue
            if flag and line.strip().startswith("backend"):
                #找到第二个backend，则可以退出了
                flag = False
                break
            if flag and line.strip():
                #如果flag是真，并且不是空行，就把内容写入result
                result.append(line.strip())
            #print(result)
    return result

def delete(backend,record):
    """
    删除函数
    :param backend:需要删除的配置项
    :param record: 需要删除的记录
    :return:
    """
    global delete_flag   #删除标记，用户控制是否删除
    record_list = fetch(backend)  #先用fetch找到需要删除的地方
    if not record_list:
        print(("没有找到需要删除的项。").center(50,"*"))
        delete_flag = False    #没有找到需要删除的对象，所以标记为假
    else:
        if record in record_list:    #找到了删除的backend和record，执行删除
            record_list.remove(record)
            print("\033[31;已经删除完成\033[0m")
            delete_flag = True       #删除标记设为真
        else:
            print(("配置项下没有没有可以删除的记录").center(50,"*"))   #找到了backend，但是没有找到record，也不删除
            delete_flag = False
    return delete_flag    #返回是否删除

def modify(backend,record_list):
    """
    修改配置文件函数
    :param backend:需要修改的配置项
    :param record_list: 需要修改的记录
    :return: 没有返回值
    """
    with open("conf",'r') as old,open("new",'w') as new:
        #打开2个配置文件，把修改record写入新配置文件
        flag = False
        for line in old:
            #找到需要修改的配置项
            if line.strip().startswith("backend") and line.strip() == "backend " + backend:
                flag = True
                new.write(line)
                #先写入record
                for new_line in record_list:
                    #写入需要修改的recored
                    new.write(" " * 6 + new_line + "\n")

                continue   #跳出本次循环，继续查找下一个backend下面的record
            #找到下一个backend
            if flag and line.strip().startswith("backend"):
                #写入backend后面的record
                flag = False
                new.write(line)
                continue
            if not flag and line.strip():
                #写入配置文件剩下的内容
                new.write(line)

def add(backend,record):
    """
    增加函数
    :param backend:需要增加的配置项
    :param record: 需要增加的record
    :return: 没有返回值
    """
    #先找到需要增加的配置项backend
    record_list = fetch(backend)
    #如果record不存在，而且backend也不存在
    if not record_list:
        with open("conf","r") as old,open("new","w")as new:
            for line in old:
                #备份配置文件
                new.write(line)
            #写入配置项和record
            new.write("\nbackend " + backend + "\n")
            new.write(" " * 6 + record + "\n")
        print("\033[32;1m添加完成\033[0m")

    else:
        #backend 存在 record也存在
        if record in record_list:
            print("\033[31;1m记录已经存在，无需添加\033[0m")
        else: #backend存在，record不存在，则追加记录，并调用modify函数修改完成
            record_list.append(record)
            modify(backend,record_list)
            print("\033[32;1m添加完成\033[0m")

def input_json():
    """
    判断用户输入是否是json格式
    :return: 返回配置项的字典值
    """
    flag = False
    while flag is not True:
        backend = input("输入配置文件信息").strip()
        try:
            backend_dict = json.loads(backend)
        except Exception as e:
            print("请输入json格式")
            continue
        flag = True
    return backend_dict

def input_judge():
    """
    用户输入判断函数
    :return: 根据用户输入返回对应的函数值
    """
    input_str = input("输入数字选择：").strip()
    if input_str == "1":
        func = "fetch"
    elif input_str == "2":
        func = "add"
    elif input_str == "3":
        func = "delete"
    elif input_str == "q" or input_str == "quit":
        exit(("拜拜！").center(50,'*'))
    else:
        print("输入有误，再来")
        func = None
    return func

def operation(action):
    """
    操作函数，再用户输入完成后调用对应的函数
    :param action: 根据input_judge的返回值输入
    :return: 没有返回值
    """
    global change_flag
    if action == "fetch":
        backend_info = input("输入需要操作的配置项").strip()
        #查找配置项信息
        result = fetch(backend_info)
        #找到了
        if result:
            output(result)
        else:
            print("没有找到匹配项")
    elif action is None:
        print("没有任何输入！")
    else:
        #获取输入字典值
        backend_record_dict = input_json()  #对输入是不是json格式判断
        for key in backend_record_dict:
            backend = key
            record = backend_record_dict[key]
            if action == "add":
                add(backend,record)
            elif action == "delete":
                delete(backend,record)

        if change_flag is True:
            shutil.copy("conf","old")
            shutil.copy("new","conf")
            result = fetch(backend)
            output(result)

def list():
    """
    用户ui接口函数
    :return: 没有返回值
    """
    print("选择一个需要的操作".center(60,"*"))
    print("【1】查找配置项信息")
    print("【2】增加配置项信息")
    print("【3】删除配置项信息")
    print("-".center(60,"-"))

def output(servers):
    """
    输出record的内容信息
    :param servers: record列表
    :return:
    """
    print("匹配到的配置项信息：".center(60,"*"))
    for server in servers:
        print("%s" % server)
    print("*".center(60,"*"))

def main():
    """
    主函数入口
    :return:
    """
    exit_flag = False
    while exit_flag is not True:
        global change_flag
        change_flag = False
        list()
        action = input_judge()
        operation(action)

main()