Python蜕变-2017-4-23

　　我的第一篇博客，这是试水练习。这次上的菜是Haporxy配置文件操作。

<1> 上需求：

具体配置文件如下：

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 weight 20 maxconn 3000

<2>分析需求：

　　1.将配置文件中的主要信息也就是“backend”后的数据展示出来。

　　2.能够对数据进行增删改查的操作。

　　3.每次操作后进行备份。

<3>功能分析：

　　1.读取配置文件并截取有用数据打印。

　　2.具备增删改查功能，在此有用户输入，所以有必要进行判断，防止错误输入导致程序崩溃。

　　3.将操作后的数据与原始的无关数据进行拼接写入配置文件

<4>完整代码：

#--------------------------Haporxy配置文件操作--------------------------------------
#py3.4                            by:羽凡                                 2017-4-23
#----------------------------------------------读取文件处理数据(返回字典Info{})-------
def Deal_file():
    s = ""
    Info = {}
    with open("配置文件.txt","r") as f:
        for i in f:
            if i.startswith("backend"):
                backend = i.strip().split(" ")[1]
                Info[backend] = []
            elif i.strip().startswith("server"):
                server = i.strip().split(" ")[1]
                weight = i.strip().split(" ")[3]
                maxconn = i.strip().split(" ")[5]
                Server = {"server":server,"weight":weight,"maxconn":maxconn}
                Info[backend].append(Server)
            else:
                continue
    return Info
#----------------------------------------------信息保存-----------------------------
def Save_Info(Info):
    s = ""
    with open("配置文件.txt","r") as f:
        for i in f:
            if not i.startswith("backend"):
                s = s + i
            else:
                break
    s1 = ""
    for backend in Info:
        s2 = "backend " + backend
        s1 = s1 + s2 + "
"
        for di in Info[backend]:
            server = di["server"]
            weight = di["weight"]
            maxconn = di["maxconn"]
            s3 = "      server {} weight {} maxconn {}".format(server,weight,maxconn)
            s1 = s1 + s3 + "
"
    S = s + s1
    with open("配置文件.txt","w+") as F:
        F.write(S)
    return(S)
#----------------------------------------------展示信息，返回指令---------------------
def Show(Info):
    show_list = []
    print("--------------------""33[1;31;0m""HaProxy配置文件管理""33[0m""-------------------------------")
    for i,n in enumerate(Info):
        show_list.append(n)
        print("33[1;32;0m",i+1,n,"33[0m")
    print("------------------------------------------------------------------------")
    order = input("请输入指令（A）新建、（D）删除、（M）修改、（C）查询、（Q）退出：")
    while order not in ("A","D","M","C","Q"):
        order = input("33[1;31;0m""输入错误，请重新输入：""33[0m")
    show_list = list(show_list)
    return order,show_list
#----------------------------------------------Backend编号输入判断-------------------
def Jude(choice,show_list):
    try:
        if eval(choice) not in range(len(show_list)+1):
            return True
        else:return False
    except:
        return True
#----------------------------------------------Server编号判断-----------------------
def JudeS(choiceS,a):
    try:
        if int(choiceS) in range(1,a+1) or int(choiceS) == 1:
            return False
        else:
            return True
    except:
        return True
#----------------------------------------------新增信息判断--------------------------
def JudeA(backend,weight,maxconn):
    try:
        if backend.startswith("www") and int(weight) in range(50) and int(maxconn) in range(5000):
            return False
    except:
        return True
#----------------------------------------------查询---------------------------------
def Check(show_list):
    choice = input("请选择backend：")
    while Jude(choice,show_list):
        choice = input("33[1;31;0m""输入错误，请重新输入：""33[0m")
    choice = int(choice)
    print("------------------------------",show_list[choice-1],"------------------------------")
    for a,b in enumerate(Info[show_list[choice-1]]):
            print("{} server:{}     weight:{}   maxconn:{}".format(a+1,b["server"],b["weight"],b["maxconn"]))
    print("----------------------------------------------------------------------------")
    select = input("请输入指令（Q）退出、（B）返回：")
    while select not in ("Q","B"):
        select = input("33[1;31;0m""请输入正确指令：""33[0m")
    if select == "Q":
        pass
    elif select == "B":
        main()
#----------------------------------------------删除---------------------------------
def Delete(show_list):
    choice = input("删除Backend（B） or  Server（S）？：")
    while(choice not in ("B","S")):
        choice = input("33[1;31;0m""请输入正确的指令：""33[0m")
    selection = input("请选择Backend编号：")
    while Jude(selection,show_list):
        selection = input("33[1;31;0m""请输入正确的编号：""33[0m")
    selection = int(selection)
    if choice =="B":
        del Info[show_list[selection-1]]
        Save_Info(Info)
        print("33[1;31;0m""删除成功！""33[0m")
        main()
    elif choice == "S":
        print("------------------------------",show_list[selection-1],"------------------------------")
        for a,b in enumerate(Info[show_list[selection-1]]):
            print("{} server:{}     weight:{}   maxconn:{}".format(a+1,b["server"],b["weight"],b["maxconn"]))
        print("----------------------------------------------------------------------------")
        choiceS = input("选择要删除的Server:")
        while JudeS(choiceS,a):
            choiceS = input("33[1;31;0m""请输入正确的编号：""33[0m")
        choiceS = int(choiceS)
        Info[show_list[selection-1]].pop(choiceS-1)
        Save_Info(Info)
        print("33[1;31;0m""删除成功！""33[0m")
        main()
#----------------------------------------------增加---------------------------------
def Add():
    print("————>请输入详细信息<————")
    backend = input("Backend名称：")
    server = input("Server名称：")
    weight = input("权重(weight)：")
    maxconn = input("最大链接数(maxconn):")
    while JudeA(backend,weight,maxconn):
        print("33[1;31;0m""输入信息错误，请重新输入！！！""33[0m")
        Add()
    if backend not in Info:
        Info[backend] = [{"server":server,"weight":weight,"maxconn":maxconn}]
        print(">>>>>新建成功")
    else:
        flag = 1
        for ser in Info[backend]:
            if server == ser["server"]:
                flag = 0
                ser["weight"] = weight
                ser["maxconn"] = maxconn
                print(">>>>>backend及server信息已存在，""33[1;31;0m""server信息已修改""33[0m")
        if flag:
            Info[backend].append({"server":server,"weight":weight,"maxconn":maxconn})
            print(">>>>>backend已存在，""33[1;31;0m""server添加成功""33[0m")
    Save_Info(Info)
    main()
#----------------------------------------------修改---------------------------------
def Modify(show_list):
    selection = input("修改backend(B)  or  server(S)?>")
    while (selection not in ("B","S")):
        selection = input("33[1;31;0m""请输入正确回答:""33[0m")
    choice = input("请输入要修改的backend编号:")
    while Jude(choice,show_list):
        choice = input("33[1;31;0m""请输入正确编号:""33[0m")
    choice = int(choice)
    if selection == "B":
        backend = input("修改backend名称为:")
        while not backend.startswith("www"):
            backend = input("33[1;31;0m""请输入合法的backend名称:""33[0m")
        temp = Info[show_list[choice-1]]
        del Info[show_list[choice-1]]
        Info[backend] = temp
        Save_Info(Info)
        print("33[1;31;0m""修改backend成功""33[0m")
        main()
    else:
        print("------------------------------",show_list[choice-1],"------------------------------")
        for a,b in enumerate(Info[show_list[choice-1]]):
            print("{} server:{}     weight:{}   maxconn:{}".format(a+1,b["server"],b["weight"],b["maxconn"]))
        print("----------------------------------------------------------------------------")
        selection = input("修改哪个server？>")
        while JudeS(selection,a):
            selection = input("33[1;31;0m""请输入正确的server编号:""33[0m")
        selection = int(selection)
        sername = input("修改server为：")
        Info[show_list[choice-1]][a-1]["server"] = sername
        Save_Info(Info)
        print("33[1;31;0m""修改server成功""33[0m")
        main()
#----------------------------------------------指令处理------------------------------
def Deal_order(order,show_list):
    if order =="Q":          #退出
        pass
    elif order =="A"  :      #新建
        Add()
    elif order == "D":       #删除
        Delete(show_list)
    elif order == "M":       #修改
        Modify(show_list)
    elif order == "C":
        Check(show_list)
    else:
        order = input("33[1;31;0m""输入错误，请重新输入：""33[0m")
        Deal_order(order,show_list)
#----------------------------------------------主函数--------------------------------
def main():
    order,show_list = Show(Info)
    Deal_order(order,show_list)
if __name__=="__main__":
    Info = Deal_file()
    main()

<5>处理流程：

<6>主要函数功能介绍：

 　　1.Deal_file():读取配置文件，并截取操作数据，通过String.startswith()找到截取位置获取数据，返回字典形式。

def Deal_file():
    s = ""
    Info = {}
    with open("配置文件.txt","r") as f:
        for i in f:
            if i.startswith("backend"):
                backend = i.strip().split(" ")[1]
                Info[backend] = []
            elif i.strip().startswith("server"):
                server = i.strip().split(" ")[1]
                weight = i.strip().split(" ")[3]
                maxconn = i.strip().split(" ")[5]
                Server = {"server":server,"weight":weight,"maxconn":maxconn}
                Info[backend].append(Server)
            else:
                continue
    return Info

　　2.Save_Info（）：将操作后的数据与原始无关数据拼接保存，接收字典作为参数。

def Save_Info(Info):
    s = ""
    with open("配置文件.txt","r") as f:
        for i in f:
            if not i.startswith("backend"):
                s = s + i
            else:
                break
    s1 = ""
    for backend in Info:
        s2 = "backend " + backend
        s1 = s1 + s2 + "
"
        for di in Info[backend]:
            server = di["server"]
            weight = di["weight"]
            maxconn = di["maxconn"]
            s3 = "      server {} weight {} maxconn {}".format(server,weight,maxconn)
            s1 = s1 + s3 + "
"
    S = s + s1
    with open("配置文件.txt","w+") as F:
        F.write(S)
    return(S)

　　3.Show（）：将要操作的数据打印出来提示输入操作指令，白名单判断输入指令，进行判断，返回输入指令和打印列表。

　　效果：

def Show(Info):
    show_list = []
    print("--------------------""33[1;31;0m""HaProxy配置文件管理""33[0m""-------------------------------")
    for i,n in enumerate(Info):
        show_list.append(n)
        print("33[1;32;0m",i+1,n,"33[0m")
    print("------------------------------------------------------------------------")
    order = input("请输入指令（A）新建、（D）删除、（M）修改、（C）查询、（Q）退出：")
    while order not in ("A","D","M","C","Q"):
        order = input("33[1;31;0m""输入错误，请重新输入：""33[0m")
    show_list = list(show_list)
    return order,show_list

　　4.Jude（）、JudeA（）、JudeS（）分别对输入的backend编号、server编号和新建的数据信息进行简单判断。

#----------------------------------------------Backend编号输入判断-------------------
def Jude(choice,show_list):
    try:
        if eval(choice) not in range(len(show_list)+1):
            return True
        else:return False
    except:
        return True
#----------------------------------------------Server编号判断-----------------------
def JudeS(choiceS,a):
    try:
        if int(choiceS) in range(1,a+1) or int(choiceS) == 1:
            return False
        else:
            return True
    except:
        return True
#----------------------------------------------新增信息判断--------------------------
def JudeA(backend,weight,maxconn):
    try:
        if backend.startswith("www") and int(weight) in range(50) and int(maxconn) in range(5000):
            return False
    except:
        return True

　　5.Deal_order（）：接受来自Show（）的返回值，对输入的指令进行分流给不同的函数处理。

def Deal_order(order,show_list):
    if order =="Q":          #退出
        pass
    elif order =="A"  :      #新建
        Add()
    elif order == "D":       #删除
        Delete(show_list)
    elif order == "M":       #修改
        Modify(show_list)
    elif order == "C":
        Check(show_list)
    else:
        order = input("33[1;31;0m""输入错误，请重新输入：""33[0m")
        Deal_order(order,show_list)

　　6.Check（）：查询，主要查看backend下的server信息。

　　效果：

#----------------------------------------------查询---------------------------------
def Check(show_list):
    choice = input("请选择backend：")
    while Jude(choice,show_list):
        choice = input("33[1;31;0m""输入错误，请重新输入：""33[0m")
    choice = int(choice)
    print("------------------------------",show_list[choice-1],"------------------------------")
    for a,b in enumerate(Info[show_list[choice-1]]):
            print("{} server:{}     weight:{}   maxconn:{}".format(a+1,b["server"],b["weight"],b["maxconn"]))
    print("----------------------------------------------------------------------------")
    select = input("请输入指令（Q）退出、（B）返回：")
    while select not in ("Q","B"):
        select = input("33[1;31;0m""请输入正确指令：""33[0m")
    if select == "Q":
        pass
    elif select == "B":
        main()

　　7.Add（）：新建，可以新建backend若已存在则在backend下新建server信息，若两种都存在则修改server信息。

#----------------------------------------------增加---------------------------------
def Add():
    print("————>请输入详细信息<————")
    backend = input("Backend名称：")
    server = input("Server名称：")
    weight = input("权重(weight)：")
    maxconn = input("最大链接数(maxconn):")
    while JudeA(backend,weight,maxconn):
        print("33[1;31;0m""输入信息错误，请重新输入！！！""33[0m")
        Add()
    if backend not in Info:
        Info[backend] = [{"server":server,"weight":weight,"maxconn":maxconn}]
        print(">>>>>新建成功")
    else:
        flag = 1
        for ser in Info[backend]:
            if server == ser["server"]:
                flag = 0
                ser["weight"] = weight
                ser["maxconn"] = maxconn
                print(">>>>>backend及server信息已存在，""33[1;31;0m""server信息已修改""33[0m")
        if flag:
            Info[backend].append({"server":server,"weight":weight,"maxconn":maxconn})
            print(">>>>>backend已存在，""33[1;31;0m""server添加成功""33[0m")
    Save_Info(Info)
    main()

　　8.Delete（）：删除，可以删除backend或Server信息。

　　效果：

#----------------------------------------------删除---------------------------------
def Delete(show_list):
    choice = input("删除Backend（B） or  Server（S）？：")
    while(choice not in ("B","S")):
        choice = input("33[1;31;0m""请输入正确的指令：""33[0m")
    selection = input("请选择Backend编号：")
    while Jude(selection,show_list):
        selection = input("33[1;31;0m""请输入正确的编号：""33[0m")
    selection = int(selection)
    if choice =="B":
        del Info[show_list[selection-1]]
        Save_Info(Info)
        print("33[1;31;0m""删除成功！""33[0m")
        main()
    elif choice == "S":
        print("------------------------------",show_list[selection-1],"------------------------------")
        for a,b in enumerate(Info[show_list[selection-1]]):
            print("{} server:{}     weight:{}   maxconn:{}".format(a+1,b["server"],b["weight"],b["maxconn"]))
        print("----------------------------------------------------------------------------")
        choiceS = input("选择要删除的Server:")
        while JudeS(choiceS,a):
            choiceS = input("33[1;31;0m""请输入正确的编号：""33[0m")
        choiceS = int(choiceS)
        Info[show_list[selection-1]].pop(choiceS-1)
        Save_Info(Info)
        print("33[1;31;0m""删除成功！""33[0m")
        main()

　　9.Modify（）：修改，修改backend或是Server信息。

　　效果：

def Modify(show_list):
    selection = input("修改backend(B)  or  server(S)?>")
    while (selection not in ("B","S")):
        selection = input("33[1;31;0m""请输入正确回答:""33[0m")
    choice = input("请输入要修改的backend编号:")
    while Jude(choice,show_list):
        choice = input("33[1;31;0m""请输入正确编号:""33[0m")
    choice = int(choice)
    if selection == "B":
        backend = input("修改backend名称为:")
        while not backend.startswith("www"):
            backend = input("33[1;31;0m""请输入合法的backend名称:""33[0m")
        temp = Info[show_list[choice-1]]
        del Info[show_list[choice-1]]
        Info[backend] = temp
        Save_Info(Info)
        print("33[1;31;0m""修改backend成功""33[0m")
        main()
    else:
        print("------------------------------",show_list[choice-1],"------------------------------")
        for a,b in enumerate(Info[show_list[choice-1]]):
            print("{} server:{}     weight:{}   maxconn:{}".format(a+1,b["server"],b["weight"],b["maxconn"]))
        print("----------------------------------------------------------------------------")
        selection = input("修改哪个server？>")
        while JudeS(selection,a):
            selection = input("33[1;31;0m""请输入正确的server编号:""33[0m")
        selection = int(selection)
        sername = input("修改server为：")
        Info[show_list[choice-1]][a-1]["server"] = sername
        Save_Info(Info)
        print("33[1;31;0m""修改server成功""33[0m")
        main()