编写通讯录2

# 利用字典的特性编写一个通讯录，记录姓名和联系(contact)电话
# 新增加将通讯录储存到文件里，下次启动可以直接调用，而不是每次都归零
# 先新建一个空白的addressBook.txt

welcome = """
欢迎进入通讯录程序
1：查找联系人资料
2：插入新的联系人
3：删除联系人
4：退出程序
"""
print(welcome)
addressBook = {}
with open('addressBook.txt', 'r', encoding='GBK') as f:
    for eachline in f.readlines():
        each1 = eachline.strip('
')
        # print(each1.split(':'))
        (name, phone) = each1.split(':')
        addressBook[name] = phone
while True:  # 循环用来多次操作直到退出程序
    try:
        n = int(input('请输入相应的选项：'))
    except Exception:
        print('请输入正确的数字选项！')
        continue
    if n == 1:
        name = input("请输入要查找的联系人：")
        if name in addressBook:
            print(name + ':' + addressBook[name])
        else:
            print('查无此人！')
    elif n == 2:
        name = input("请输入新的联系人姓名：")
        if name in addressBook:
            print(name + '已经在通讯录里面')
        else:
            phone = input("请输入新的联系人的联系电话：")
            addressBook[name] = phone
            with open('addressBook.txt', 'a+') as f:
                f.write(name + ':' + phone + '
')
            print(name + ':' + phone + '已添加')
    elif n == 3:
        name = input("请输入要删除的联系人：")
        if not(name in addressBook):
            print(name + '不在通讯录里面')
        else:
            del addressBook[name]
            print(name + ':' + '已删除')
    elif n == 4:
        print('感谢使用通讯录，再见！')
        break