python(13)- 文件处理应用Ⅱ:增删改查

用户选择1,增加功能:
    用户输入www.oldboy2.org和server 11111 weight 2222 maxconn 3333后,
在www.oldboy2.org下增加一条server信息:server 11111 weight 2222 maxconn 3333
用户选择2,删除功能:
    用户输入www.oldboy1.org和10.10.10.1 10.10.10.1后,
server 10.10.10.1 10.10.10.1 weight 22 maxconn 2000将被删除

用户选择3,修改功能:
    用户输入www.oldboy2.org,更改前ip所在的行server 3.3.3.3 3.3.3.3 weight 20 maxconn 3000,
更改后ip所在的行server 2.2.2.4 2.2.2.4 weight 20 maxconn 3000

用户选择4,查询功能:
    用户输入www.oldboy1.org用户信息后,打印对应的server信息:
        server 10.10.0.10 weight 9999 maxconn 33333333333
        server 10.10.10.1 weight 22 maxconn 2000
        server 2.2.2.4 weight 20 maxconn 3000

用户选择5,退出程序

def menu():
    menu={"1":"增加","2":"删除","3":"修改","4":"查询","5":"退出"}
    for key in menu:
        print(key,menu[key])

def search():
    l=[]
    flag=True
    while flag:
        url=input("please input url you want to search:").strip()   #输入想要查询的域名
        with open("haproxy.conf", encoding="utf8") as s:    #打开文件
            for line in s:
                if line.startswith("backend") and url in line:   #第一步,读到想要查询的域名所在的行,改变标志位状态
                    flag=False
                    continue
                if line.startswith("backend") and flag == False:  #第三步,再次读取域名所在的行后,本次循环结束
                    break

                if flag==False:     #第二步,将域名下的ip所在行写入列表中
                    l.append(line)
        for i in l:                 #第四步,打印需要查询的域名下的ip地址所在的行
            print(i.strip())
        break

def delete():
    flag=True
    l=[]
    url = input("input url you want to delete:").strip()   #输入想删除的ip地址所对应的域名
    ip = input("input ip you want to delete:").strip()     #输入想要删除的域名
    while flag:
        with open("haproxy.conf") as d,\
        open("haproxy.conf_delete","w+") as d_2:   #打开原文档haproxy.conf ,创建删除后的文档haproxy.conf_delete
            for line in d:
                if line.startswith("backend") and url in line:    #第二步,当读取到想删除的ip地址对应的域名所对应的行
                    d_2.writelines(line)                            #将该行写入haproxy.conf_delete文档中
                    flag=False
                    continue
                if line.startswith("backend") and flag==False:
        # 第四步,第二次读到域名行,将列表里的内容写入haproxy.conf_delete文档中,要删除的ip行不写入文档中
                    flag=True
                    for line in l:
                        if url in line:
                            continue
                        d_2.writelines(line)
                    d_2.writelines(line)
                if flag==False:                        #第三步,将域名下所对应的ip行写入列表中
                    l.append(line)
                if flag:                               #第一步将读取到的行写入haproxy.conf_delete文档中
                    d_2.writelines(line)
        break

def add():
    url =input("please input url you want to add:").strip()          #输入想要增加ip所在的域名和增加的ip所在行
    ip = input("please input ip you want to add:")
    with open("haproxy.conf") as a,open("haproxy.conf_add","w") as a_2: #打开文档haproxy.conf,新建文档haproxy.conf_add
        for line in a:
            a_2.writelines(line)                                   #循环写入读取到的行到文档haproxy.conf_add中
            if line.startswith("backend") and url in line:
                a_2.writelines(ip)                #当读取到ip所在的域名,将要加入的ip所在的行写入文档haproxy.conf_add中
                a_2.writelines("\n")

def change():
    l=[]
    flag=True
    url = input("please input url you want to change:").strip()  # 输入想要修改ip所在的域名
    ip = input("please input ip you want to change:")    # 输入想要修改ip所在行
    ip_changed=input("please input ip you want to changed:")  #输入想要修改过ip所在行
    with open("haproxy.conf") as c,open("haproxy.conf_change", "w") as c_2:  ##打开文档haproxy.conf,新建文档haproxy.conf_change
        for line in c:
            if line.startswith("backend") and url in line:
                c_2.writelines(line)              #第二步,读到想要修改ip所在的域名的行将该行写入文档haproxy.conf_change
                flag=False                       #改变标志位状态
            if line.startswith("backend") and flag==False:
                for line in l:     #第四步,循环到列表中需要改变的ip地址,将修改过的ip地址写入文档haproxy.conf_change中
                    if line==ip:
                        c_2.writelines(ip_changed)
                        continue
                    c_2.writelines(line)    #将列表中其他内容写入文档haproxy.conf_change中
                flag=True
            if flag:                       #第一步,将读到的行写入文件haproxy.conf_change中
                c_2.writelines(line)
            if flag==False:               #第三步,将域名所在的行放入列表中
                l.append(line)

def main():
    while True:
        menu()
        choice=int(input("input your choice:"))
        if choice==1:
            add()
        elif choice==2:
            delete()
        elif choice==3:
            change()
        elif choice==4:
            search()
        elif choice==5:
            break

main()

  

 
 
    
 
 

posted @ 2017-04-09 18:40  许二哈哈哈  阅读(855)  评论(0编辑  收藏  举报