作业2：实现修改harproxy配置文件

要求，通过文件读写实现增删改查

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 30
backend www.sina.com
        server 100.1.7.7 100.1.7.7 weight 20 maxconn 3000
backend www.baidu.com
        server 100.1.7.9 weight 20 maxconn 30

以上是配置文件内容。

import os
# 导入os模块，获得haproxy.txt的路径
DB_PATH=os.path.join(os.path.dirname(__file__),"haproxy.txt")
# 获得new.txt的路径，这是一个临时文件，用于临时存需要修改的数据，之后会改名成haproxy.txt，而之前的haproxy.txt文件将会被删除
DB_NEW_PATH=os.path.join(os.path.dirname(__file__),"new.txt")

# 查找函数（查询函数中调用的）
def select(bef_find):
    with open(DB_PATH, "r", encoding="utf-8")as f:
        # 每行每行的遍历。防止文件过大 爆内存
        for line in f:
            # 查询到一样的字符串 就读取它的下一行。然后返回
            if bef_find == line:
                out_find = f.readline()
                return out_find
        # 若读完全部的行后都没找到打印该信息
        else:
            print("没有查询到")

# 添加消息函数
def add_message(one_line):
    with open(DB_PATH,'a',encoding="utf-8")as f:
        f.write(one_line)

def update_message(old_data,new_data):
    # 方法都一样。
    with open(DB_PATH,"r",encoding="utf-8")as f,\
        open(DB_NEW_PATH,"w",encoding="utf-8")as f1:
        for line in f:
            if line == old_data:
                line=line.replace(line,new_data) # 因为字符串是不可变类型，所以要重新进行赋值才能实现修改操作
            f1.write(line)
    os.remove(DB_PATH)
    os.rename(DB_NEW_PATH,DB_PATH)

# 查询函数
def search():
    while True:
        # 接受查询的网址
        find=input("请输入查询的网址：")
        if find=="q":
            break
        # 设置开头，拼接成查询的数据的最终格式
        before="backend "
        bef_find=before+find+"\n"
        # 调用查找函数
        res=select(bef_find)
        # 若查找函数没有返回值就什么都不做
        if not res:
            pass
        # 若查找函数有返回值，打印它
        else:
            print(res)

# 添加函数
def add():
    while True:
        # 接收记录
        args=input("请输入添加的记录(以字典的形式)：").strip()
        # 接收到 q 退出
        if args=='q':
            break
        '''arg = {'backend': 'www.oldboy.org','record': {'server': '100.1.7.9','weight': 20,'maxconn': 30}}'''
        # 如果输入正确，是能够调用eval函数不报错的进入try的函数
        try:
            # 执行eval内置函数，将字符串转换成python类型
            args=eval(args)
            # 拼接字符串
            bef_find = f"backend " + args["backend"] + "\n"
            # 调用查询函数，接收返回值
            old_data = select(bef_find)
            # 有返回值，打印”已经存在了“
            if old_data:
                print("已经存在了，不能添加该网址的配置")
                break
            # 没有返回值就继续拼接字符串
            one_line = f"backend {args['backend']}\n\t\tserver {args['record']['server']} weight {args['record']['weight']} maxconn {args['record']['maxconn']}\n"
            # 调用添加消息函数。
            add_message(one_line)
        # 否则就抛出异常
        except (SyntaxError,TypeError,KeyError,NameError,UnboundLocalError):
            print("输入格式不符合规范")




def update():
    while True:
        # 还是接收记录
        args=input("请输入要更新的记录（以字典的形式）：").strip()
        if args=="q":
            break
        # 都跟删除的一样的套路
        try:
            args = eval(args)
            # 拼接字典的值
            bef_find=f"backend "+ args["backend"] +"\n"
            # 调用查找函数
            old_data=select(bef_find)
            '''arg = {'backend': 'www.oldboy.org','record': {'server': '100.1.7.9','weight': 20,'maxconn': 30}}'''
            # 继续拼接字符串
            new_data=f"\t\tserver {args['record']['server']} weight {args['record']['weight']} maxconn {args['record']['maxconn']}\n"
            # 调用更新记录函数，将老数据和新数据传进去
            update_message(old_data,new_data)
        except (SyntaxError,TypeError,KeyError,NameError,UnboundLocalError):
            print("输入格式不符合规范")


def delete():
    while True:
        # 接收网址
        website=input("请输入要删除的网址：").strip()
        # q就退出
        if website=="q":
            break
        # 创建一个旗帜，默认为False，通过旗帜来控制是否将haproxy.txt的内容是否写入到new.txt的文件中（最后通过改名变相的起到了删除的作用）
        delete_flag=False
        # 计数，删除多少条（貌似没啥子用）
        count=0
        # 用读方式打开，haproxy.txt， 用写的方式打开new.txt（w方式下没有就会新建一个空文件）
        with open(DB_PATH,"r",encoding="utf-8")as f,\
            open(DB_NEW_PATH,"w",encoding="utf-8")as f1:
            # 还是遍历每一行，免得爆内存
            for line in f:
                # 还是老办法，继续拼接字符串
                if f"backend {website}" == line.strip():
                    # 若相等，就将旗帜设置为True（之后判断是否写入到new文件中的时候，True就表示不写）
                    delete_flag=True
                    count+=1
                    continue
                # 遍历到下一行的时候，碰到backend开头的字符串，并且 后面的网址不存在的就把旗帜给调会False（不恢复的话后面的都不会写入new文件）
                if line.strip().startswith("backend") and line.strip()!=website.strip():
                    delete_flag=False
                # 旗帜为False 就写写写
                if not delete_flag:
                    f1.write(line)
        print(f"找到{count}条，已经删除！")
        # 移除之前的文档，把新文档改名成老文档，实现删除功能
        os.remove(DB_PATH)
        os.rename(DB_NEW_PATH,DB_PATH)

# 创建一个功能字典，用于存放【功能编号】 和 【函数地址】
run_dict={
    "1":search,
    "2":add,
    "3":update,
    "4":delete,
}
def run():
    while True:
        # 遍历功能字典，取出函数名称
        for k in run_dict:
            print(k,run_dict[k].__name__)
        # 获取功能编号
        choice=input("请输入功能：").strip()
        # 接收到q就退出
        if choice=="q":
            break
        # 查看是否是数字
        if not choice.isdigit():
            print("您输入的不是数字！")
            continue
        # 查看是否有这个功能编号
        if choice not in run_dict.keys():
            print("没有该功能编码！")
            continue
        # 通过功能编号调用对应的函数
        run_dict[choice]()

run()