Python-24_综合练习-01_函数_文件处理_解耦--查询功能

--------------------------------------------- 一、查询功能：---------------------------------------------

def fetch(data):
    print("33[1;43m查询功能33[0m")          # 33[1;43m---33[0m  给---添加颜色，43代表颜色
    print("33[1;43m用户数据33[0m",data)
    backend_data="backend %s" %data
    with open("haproxy.conf","r") as read_f:
        arg=False
        res=[]
        for read_line in read_f:
            if read_line.strip()==backend_data:
                arg=True
                continue
            if arg and read_line.startswith("backend"):
                break
            if arg:
                print("33[1;43m%s33[0m"%read_line,end="")
                res.append(read_line)
    return res

def add():
    print("添加功能")
def change():
    print("修改功能")
def delete():
    print("删除功能")

msg="""
1:查询
2:添加
3:修改
4:删除
5:退出
"""
msg_dic={"1":fetch,"2":add,"3":change,"4":delete}

while True:
    print(msg)
    choice=input("请输入你需要的选项：").strip()
    if not choice:continue
    if choice=="5":break
    data=input("请输入您的数据：").strip()
    msg_dic[choice](data)


# 退出当前循环、退出全部循环：
arg=True
while arg:
    print("level1")
    choice=input("level1 >>>").strip()
    if choice == "quit": break
    if choice == "quit_all": arg = False
    while arg:
        print("level2")
        choice = input("level2 >>>").strip()
        if choice == "quit": break
        if choice == "quit_all": arg = False
        while arg:
            print("level3")
            choice = input("level3 >>>").strip()
            if choice == "quit": break
            if choice == "quit_all": arg = False

--------------------------------------------- 二、修改功能：---------------------------------------------

import os
def fetch(data):
    print("33[1;43m查询功能33[0m")  # 33[1;43m---33[0m  给---添加颜色，43代表颜色
    print("33[1;43m用户数据33[0m", data)
    backend_data = "backend %s" % data
    with open("haproxy.conf", "r") as read_f:
        arg = False
        res = []
        for read_line in read_f:
            if read_line.strip() == backend_data:
                arg = True
                continue
            if arg and read_line.startswith("backend"):
                break
            if arg:
                print("33[1;43m%s33[0m" % read_line, end="")
                res.append(read_line)
    return res


def add():
    print("添加功能")


def change(data):
    print("修改功能")
    print("用户输入数据%s" % data)
    backend = data[0]["backend"]  # 文件当中的一条记录
    backend_data="backend %s" %backend
    old_server_record="%sserver %s %s weight %s maxconn %s
" %(" "*8,data[0]["record"]["server"],
                                                              data[0]["record"]["server"],
                                                              data[0]["record"]["weight"],
                                                              data[0]["record"]["maxconn"])
    new_server_record="%sserver %s %s weight %s maxconn %s
" %(" "*8,data[1]["record"]["server"],
                                                              data[1]["record"]["server"],
                                                              data[1]["record"]["weight"],
                                                              data[1]["record"]["maxconn"])
    print("用户想要修改的记录是",old_server_record)
    res=fetch(backend)
    print("来自修改功能",res)
    if not res or old_server_record not in res:
        return "修改内容不存在"
    else:
        index=res.index(old_server_record)
        res[index]=new_server_record

    res.insert(0,"%s
" %backend_data)
    with open("haproxy.conf","r") as read_f,
        open("haproxy.conf_new","w") as write_f:
        arg=False
        arg_write=False
        for read_line in read_f:
            if read_line.strip()==backend_data:
                arg=True
                continue
            if arg and read_line.startswith("backend"):
                arg=False
            if not arg:
                write_f.write(read_line)
            else:
                if not arg_write:
                    for record in res:
                        write_f.write(record)
                    arg_write=True
    os.rename("haproxy.conf","haproxy.conf.bak")
    os.rename("haproxy.conf_new","haproxy.conf")
    os.remove("haproxy.conf.bak")


def delete():
    print("删除功能")


msg = """
1:查询
2:添加
3:修改
4:删除
5:退出
"""
msg_dic = {"1": fetch, "2": add, "3": change, "4": delete}

while True:
    print(msg)
    choice = input("请输入你需要的选项：").strip()
    if not choice: continue
    if choice == "5": break
    data = input("请输入您的数据：").strip()
    if choice != "1":
        data = eval(data)
    res = msg_dic[choice](data)
    print(res)

# data --->> [{"backend":"www.newmet1.top","record":{"server":"2.2.2.5","weight":"30","maxconn":"4000"}},{"backend":"www.newmet1.top","record":{"server":"2.2.2.4","weight":"20","maxconn":"3000"}}]

# 查询和修改功能，代码都涉及到文件处理的操作，可以单独定义一个函数，用来处理文件的操作！！

------------------------------------ 二、修改功能--程序的解耦：-------------------------------------

import os
def file_handle(backend_data,res=None,type="fetch"):
    if type=="fetch":
        with open("haproxy.conf", "r") as read_f:
            tag = False
            res = []
            for read_line in read_f:
                if read_line.strip() == backend_data:
                    tag = True
                    continue
                if tag and read_line.startswith("backend"):
                    break
                if tag:
                    print("33[1;43m%s33[0m" % read_line, end="")
                    res.append(read_line)
        return res
    elif type == "change":
        with open("haproxy.conf", "r") as read_f, 
                open("haproxy.conf_new", "w") as write_f:
            tag = False
            tag_write = False
            for read_line in read_f:
                if read_line.strip() == backend_data:
                    tag = True
                    continue
                if tag and read_line.startswith("backend"):
                    tag = False
                if not tag:
                    write_f.write(read_line)
                else:
                    if not tag_write:
                        for record in res:
                            write_f.write(record)
                        tag_write = True
        os.rename("haproxy.conf", "haproxy.conf.bak")
        os.rename("haproxy.conf_new", "haproxy.conf")
        os.remove("haproxy.conf.bak")

def fetch(data):
    # print("33[1;43m查询功能33[0m")  # 33[1;43m---33[0m  给---添加颜色，43代表颜色
    # print("33[1;43m用户数据33[0m", data)
    backend_data = "backend %s" % data
    return file_handle(backend_data)


def add():
    print("添加功能")


def change(data):
    print("修改功能")
    print("用户输入数据%s" % data)
    backend = data[0]["backend"]  # 文件当中的一条记录
    backend_data="backend %s" %backend
    old_server_record="%sserver %s %s weight %s maxconn %s
" %(" "*8,data[0]["record"]["server"],
                                                              data[0]["record"]["server"],
                                                              data[0]["record"]["weight"],
                                                              data[0]["record"]["maxconn"])
    new_server_record="%sserver %s %s weight %s maxconn %s
" %(" "*8,data[1]["record"]["server"],
                                                              data[1]["record"]["server"],
                                                              data[1]["record"]["weight"],
                                                              data[1]["record"]["maxconn"])
    print("用户想要修改的记录是",old_server_record)
    res=fetch(backend)
    print("来自修改功能",res)
    if not res or old_server_record not in res:
        return "修改内容不存在"
    else:
        index=res.index(old_server_record)
        res[index]=new_server_record
    res.insert(0,"%s
" %backend_data)
    file_handle(backend_data,res=res,type="change")

def delete():
    print("删除功能")


msg = """
1:查询
2:添加
3:修改
4:删除
5:退出
"""
msg_dic = {"1": fetch, "2": add, "3": change, "4": delete}

while True:
    print(msg)
    choice = input("请输入你需要的选项：").strip()
    if not choice: continue
    if choice == "5": break
    data = input("请输入您的数据：").strip()
    if choice != "1":
        data = eval(data)
    res = msg_dic[choice](data)
    print(res)

# data --->> [{"backend":"www.newmet1.top","record":{"server":"2.2.2.5","weight":"30","maxconn":"4000"}},{"backend":"www.newmet1.top","record":{"server":"2.2.2.4","weight":"20","maxconn":"3000"}}]

# 查询和修改功能，代码都涉及到文件处理的操作，可以单独定义一个函数，用来处理文件的操作！！