python第十天

今天,检查了所有的作业,提交完成了!!

总结了一下,只常码代码才能慢慢熟悉相关的语法!

HAproxy配置文件操作 作业 经过测试完成啦!

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

先上一下流程图:

详细代码如下:

#！usr/bin/env python
#-*-coding:utf-8-*-
# Author calmyan
import time
time_format='%y-%m-%d_%H-%M-%S'#格式化时间输出格式
times=time.strftime(time_format)#将时间转为字符串

def filebak():#自动备份配置文件函数
    haproxy_bak='HA_bak_'+times
    with open('HAproxy','r+',encoding='utf-8') as ha_file,
            open(haproxy_bak,'w',encoding='utf-8') as new_file:
        for line in ha_file:
            new_file.write(line)
    return haproxy_bak#返回备份文件名

def file_query():#查询server信息函数
    file_dict={}#创建一个空字典
    with open('HAproxy','r+',encoding='utf-8') as ha_file:
        for line in ha_file:
            line=line.strip()#去除空格与回车
            try:#如果配置文件原先被破坏,会提示
                if line.startswith('backend'):#判断以backend开头的行
                    key_1=line.split()[1]#以空格进行分割赋于变量
                    file_dict[key_1]={}#设置字典格式
                    file_dict_key1=file_dict[key_1]#赋于变量
                if line.startswith('server'):
                    b={'server':[line.split()[1],line.split()[2]],'weight':line.split()[4],'maxconn':line.split()[6]}#
                    file_dict_key1.update(b)#更新字典
            except Exception as e:#出错可返回操作
                print("33[31;1mHAproxy文件出错,请仔细检查！33[0m")
                exit()
        return file_dict#返回字典 所有backend的信息

def out():#返回退出函数
    qu=input("请按提示操作！按33[31;1mY/y33[0m退出修改器!其它键返回！")
    if qu=='Y'or qu=='y':
        print("您已经退出修改器！" )
        exit()#退出程序
    else :
        pass

def show():#提示示例
    arg_list="{'backend':'www.oldboy3.org','record':{'server':['110.110.1.1','110.110.1.1'],'weight':20,'maxconn':344}}"
    print(arg_list)

def add():#添加backend信息函数
    while True:
        print('33[43;1m添加backend信息！输入格式示例:33[0m'.center(40,'-'))
        show()#显示提示
        print('
操作提示：按 33[31;1m B/b33[0m返回上层菜单<<<  33[31;1mQ/q33[0m直接退出！
')
        arg=input('>>>>>请输入你要添加的信息:')
        if arg=='b' or arg=='B':
            break
        elif arg=='q' or arg=='Q':
            out()
        else :
            try:#进行异常处理
                arg_dict=eval(arg)#输入信息转为字典
                if judge(arg_dict):#判断是否已经存在
                    print('33[41;1mbackend已经存在33[0m'.center(40,'*'))
                    continue
                else:
                    with open('HAproxy','a',encoding='utf-8') as add_file:
                        backend=arg_dict['backend']#key赋于变量 backend信息
                        record=arg_dict['record']##key赋于变量
                        print_info(backend,record)#输出相关信息
                        #要添加的字符串变量
                        backend_add='''
backend {_backend_}
        server {_server_1} {_server_2} weight {_weight_} maxconn {_maxconn_}'''
                        backend_add_for=backend_add.format(_backend_=backend,_server_1=record[server][0],_server_2=record[server][1],_weight_=record[weight],_maxconn_=record[maxconn])#进行格式化
                        q=input("33[43;1m警告！33[0m是否进行添加操作,按33[43;1mY/y！33[0m确认！否则返回")
                        if q=='y' or q=='Y':
                            add_file.write(backend_add_for)#写入文件
                            add_file.flush()#实时写入
                            print('33[32;1m恭喜！添加成功！33[0m

')
                            continue
                        else:
                            pass
            except Exception as e:#出错可返回操作
                print("33[31;1m输入的格式有误，请重新输入！33[0m")
                pass

def char_file():#修改backend信息
    while True:
        print('33[35;1m修改backend信息！输入格式示例：33[0m'.center(40,'='))
        show()#显示提示
        print('
操作提示：按 33[35;1m B/b33[0m返回上层菜单<<<  33[35;1mQ/q33[0m直接退出！
')
        arg=input('>>>>>请输入你要修改的信息:')
        if arg=='b' or arg=='B':
            break
        elif arg=='q' or arg=='Q':
            out()
        else:
            try:
                arg_dict=eval(arg)#转为字典
                if judge(arg_dict):#判断是否存在
                    backend=arg_dict['backend']#key赋于变量 backend信息
                    record=arg_dict['record']##key赋于变量
                    dict_file=file_query()#执行查询函数
                    print('33[37;1m旧的配置33[0m'.center(50,'-'))
                    print_info(backend,dict_file[backend] )#显示修改前的信息
                    print('33[32;1m新的配置33[0m'.center(50,'-'))
                    print_info(backend,record)#显示要修改后的信息
                    server_infor='server '+ dict_file[backend][server][0]+' '+dict_file[backend][server][1]#ip判断文本所在行
                    backend_add='''	server {_server_1} {_server_2} weight {_weight_} maxconn {_maxconn_}'''#新的配置信息
                    backend_add_for=backend_add.format(_server_1=record[server][0],_server_2=record[server][1],_weight_=record[weight],_maxconn_=record[maxconn])#进行格式化字符串
                    file_pop_list=[]
                    with open('HAproxy','r',encoding='utf-8') as chars_file:
                        for line in chars_file:
                            if server_infor in line:#判断 backend所在行
                                file_pop_list.append(backend_add_for)#写入列表
                            else:
                                file_pop_list.append(line)#写入列表
                    q=input("33[45;1m警告！33[0m是否进行修改操作,按33[45;1mY/y！33[0m确认！否则退出修改操作!")
                    if q=='y' or q=='Y':
                        write_file(file_pop_list)#写文件函数

                        print("33[31;1m修改完成!33[0m")
                        #file_pop_list=[]#清空列表
                        continue
                    else:
                        pass
                else:
                    jud=input('该信息不存在，是否进行添加？按 33[35;1m B/b33[0m返回不添加 <<<  33[35;1mY/y33[0m确定添加！
')
                    if jud=='b' or jud=='B':
                        break
                    elif jud=='Y' or jud=='y':
                        add()#调用添加函数
                    else:
                        pass
            except Exception as e:#出错可返回操作
                print("33[31;1m输入的格式有误，请重新输入！33[0m")
                pass

def write_file(file_pop_list):#写文件函数
    with open('HAproxy','w',encoding='utf-8') as pop_file:
        for i in range(len(file_pop_list)):
            pop_file.write(file_pop_list[i])#修改文件
        pop_file.flush()
    #file_pop_list=[]#清空列表

def judge(arg_dict):#判断是否存在的函数
    backend=arg_dict['backend']#key赋于变量 backend信息
    file_dict=file_query()#调用查询函数
    if backend in file_dict:#判断是否已经存在
        return True#返回真
    else:
        return False#返回假

def infor_pop():#删除backend信息
    while True:
        print('33[31;1m删除backend信息！输入格式示例:33[0m'.center(40,'-'))
        show()#显示提示
        print('
操作提示：按 33[31;1m B/b33[0m返回上层菜单<<<  33[31;1mQ/q33[0m直接退出！
')
        arg=input('>>>>>请输入你要删除的信息:')
        if arg=='b' or arg=='B':
            break
        elif arg=='q' or arg=='Q':
            out()
        else:
            try:#进行异常处理
                arg_dict=eval(arg)#转为字典
                if judge(arg_dict):#进行判断是否存在
                    backend=arg_dict['backend']#key赋于变量 backend信息
                    record=arg_dict['record']##key赋于变量
                    server_infor='server '+ record[server][0]+' '+record[server][1]#判断ip文本
                    file_pop_list=[]
                    with open('HAproxy','r+',encoding='utf-8') as pop_file:
                        for line in pop_file:
                            if line.startswith('backend') and backend in line:#判断 backend所在行
                                line=''#
                                file_pop_list.append(line)#写入列表
                            if server_infor in line:#判断server所在行
                                line=''
                                file_pop_list.append(line)#写入列表
                            file_pop_list.append(line)#写入列表 追加
                    print_info(backend,record)
                    q=input("33[41;1m警告！33[0m是否进行删除操作,按33[41;1mY/y！33[0m确认！否则退出返回!")
                    if q=='y' or q=='Y':
                        write_file(file_pop_list)#写文件函数
                        #file_pop_list=[]#清空列表
                        print("33[31;1m删除完成!33[0m")
                        #清空列表
                        continue
                    else:
                        pass
                else:
                    print("33[31;1m您所的信息不存,请重新输入!33[0m")
            except Exception as e:#出错可返回操作
                print("33[31;1m输入的格式有误，请重新输入！33[0m")
                pass

def print_info(backend,record):#输出相关信息
    infor_pr='''
        33[35;1m=======配置信息=========33[0m
            backend:{arg}
            server: {server} {server1}
            weight: {weight}
            maxconn: {maxconn}
        33[35;1m=======================33[0m
    '''#定义输出文本
    infor_pr=infor_pr.format(arg=backend,server=record[server][0],server1=record[server][1],weight=record[weight],maxconn=record[maxconn])#格式化文本
    print(infor_pr)

def print_sss():#界面提示文字
    infor_pr_1='''
        33[35;1m===========确认信息=============33[0m

    33[36;1m温馨提示:33[0m33[31;1m操作有风险,33[0m33[32;1m阁下需谨慎!33[0m33[33;1m请再次确认!'33[0m
            33[37;1m确认操作系统将为您自动备份!33[0m

        33[35;1m===============================33[0m
    '''#定义输出文本
    print(infor_pr_1)

#程序开始
print('33[35;1mHAproxy 配置文件修改器33[0m'.center(50,'='))
#file_pop_list=[]#修改,删除,时读写的文件列表
backend_key='backend'
server='server'#定义字符串变量
weight='weight'
maxconn='maxconn'
while True:
    print('''您可以进行以下相关操作:

            按33[32;1m 1 >>> 查询backend信息.33[0m
            按33[33;1m 2 >>> 添加修改相关backend信息.33[0m
            按33[35;1m 3 >>> 修改相关backend信息.33[0m
            按33[31;1m 4 >>> 删除相关backend信息.33[0m
            按33[30;1m 5 >>> 直接出退修改器.33[0m
    ''')
    choose=input('请输入您想要的操作 >>> ：')
    if choose=='1':
        while True:
            print('33[32;1m查询backend信息界面！33[0m'.center(50,'='))
            print('33[42;1m查询示例33[0m'.center(50,'-'))
            print('www.oldboy.org
')
            print('操作提示：按 33[31;1m B/b33[0m返回上层菜单<<<  33[31;1mQ/q33[0m直接退出！')
            arg=input('请输您要查询的信息:')
            if arg=='b' or arg=='B':
                break
            elif arg=='q' or arg=='Q':
                out()
            else :
                dict_file=file_query()#执行查询函数
                if arg in dict_file:#如果所输入的存在
                    dict_file_key=dict_file[arg]#所对应的信息赋于变量
                    print_info(arg,dict_file_key)#调用函数格式化输出
                    out()
                else:
                    print('您输入的不存在,请重新输入')
                    pass
    elif choose=='2':
        while True:
            print('33[33;1m添加backend信息界面！33[0m'.center(50,'='))
            print_sss()
            qch=input('按提示进行操作：33[33;1mB/b<<<33[0m返回上层 33[33;1mY/y>>>33[0m确定开始添加 ')
            if qch=='b' or qch=='B':
                break
            elif qch=='Y' or qch=='y':
                print('33[31;1m添加backend信息，有一定的风险！已经为您自动备份配置文件！33[0m')
                print('备份文件：33[33;1m%s33[0m'%filebak())#自动备份配置文件
                add()#增加信息函数
            else:
                pass
    elif choose=='3':
        while True:
            print('33[35;1m修改backend信息界面！33[0m'.center(50,'='))
            print_sss()
            qch=input('按提示进行操作：33[35;1mB/b<<<33[0m返回上层 33[35;1mY/y>>>33[0m确定开始添加 ')
            if qch=='b' or qch=='B':
                break
            elif qch=='Y' or qch=='y':
                print('33[31;1m修改backend信息，有一定的风险！已经为您自动备份配置文件！33[0m')
                print('备份文件：33[35;1m%s33[0m'%filebak())
                char_file()#修改文件函数
            else:
                pass
    elif choose=='4':
        while True:
            print('33[31;1m删除backend信息界面！33[0m'.center(40,'='))
            print_sss()
            qch=input('按提示进行操作：33[31;1mB/b<<<33[0m返回上层 33[31;1mY/y>>>33[0m确定开始删除 ')
            if qch=='b' or qch=='B':
                break
            elif qch=='Y' or qch=='y':
                print('33[31;1m删除backend信息，有一定的风险！已经为您自动备份配置文件！33[0m')
                print('备份文件：33[31;1m%s33[0m'%filebak())
                infor_pop()#删除信息函数
            else:
                pass
    elif choose=='5':
        print('33[30;1m您已经出退修改器.33[0m')
        exit()
    else:
        print('33[41;1m您的输入有误！请重新输入！33[0m')
        pass

功能全部实现!