[python]字典應用之通訊錄

print('|----歡迎進入通訊錄系統----|')
print('|----1:查詢聯繫人的資料----|')
print('|----2:插入新的聯繫人----|')
print('|----3:刪除已有的聯繫人----|')
print('|----4:退出通訊錄程序----|')

contact = dict()
while 1:
      instr = int(input('請輸入操作碼'))
      if instr == 1:
            name = input('請輸入姓名：')
            if name in contact:
                  print(name + ' ：' + contact[name])
            else:
                  print('查無此人')
      if instr == 2:
            name = input('請輸入姓名：')
            if name in contact:
                  print('此人已經在通訊錄中', end = ' ')
                  print(name + ' ：' + contact[name])
                  if input('是否要修改用戶資料[YES/NO] ：') == 'YES' :
                        contact[name] = input('請輸入用戶聯繫電話：')
            else:
                  contact[name] = input('請輸入用戶聯繫電話：')
      if instr == 3:
            name = input('請輸入姓名：')
            if name in contact:
                  del(contact[name])
            else:
                  print('輸入的聯繫人不存在')
      if instr == 4:
            break
print('|----感謝使用通訊錄程序----|')