小甲鱼第25课课后练习

尝试利用字典的特性编写一个通讯录程序吧：

contacts = dict([("小甲鱼","12345"),("王小驴","123456")])
print("欢迎进入通讯录程序\n1:查询联系人资料\n2:插入新的联系人\n3:删除已有联系人\n4.退出通讯录程序")
while 1:
    a = int(input("请输入相关的指令代码："))
    if a == 1:
        name = input("请输入联系人姓名：")
        if name in contacts:
            print(name + ":" + contacts[name])
            #can only concatenate str (not "int") to str。由于通讯录字典里面的电话号码一定要改成字符串类型。不然print内部加号连接出现不一样的变量类型
        else:
            print("您输入的姓名不在通讯录中")

    if a == 2:
        name = input("请输入联系人姓名：")
        if name in contacts:
            print("您输入的姓名已经在通讯录中", end = "")
            print(name + ":" + contacts[name])
            print("您是否要修改用户信息(YES OR NO)？")
            if input("您是否要修改用户信息(YES OR NO)？" == "YES"):
                contacts[name] = input("请输入联系方式：")
        else:
            contacts[name] = input("请输入联系方式：")

    if a == 3:
        name = input("请输入联系人姓名：")
        if name in contacts:
            del(contacts[name])
        else:
            print("您输入的姓名不在通讯录中")

    if a == 4:
        break