使用字典写一个简单的通讯录 AND 用户登陆程序

# -*- coding: gbk -*-

print("""
|--- 欢迎进入通讯录程序 ---|
|---  1：查询联系人资料 ---|
|---  2：插入新的联系人 ---|
|---  3：删除已有联系人 ---|
|---  4：退出通讯录程序 ---|""")

contacts = {"木又盛":"021-5566321"}

while True:
button = int(input("请输入相关指令："))
if button == 1:
name = input('请输入联系人姓名：')
if name in contacts:
print("%s : %s "%(name,contacts[name]))
else:
print('您输入的姓名不再通讯录中！')
if button == 2:
name = input('请输入联系人姓名：')
if name in contacts:
print("您输入的姓名已在通讯录中存在 --> %s:%s"\
%(name,contacts[name]))
flag = input("是否修改用户资料（YES/NO）：")
if  flag== 'YES':
tel = input('请输入用户联系电话：')
contacts.update({name:tel}) #更新字典
print("联系人已更新!")
else:
continue
else:
contacts[name] =  input('请输入联系人电话:')
print("联系人已保存!")
if button == 3:
name = input('请输入要删除的联系人姓名：')
if name in contacts:
flag = input('确定要删除此联系人?(YES/NO)')
if flag in ['YES' ,"yes" ,"Y" ,"y" ] :
del(contacts[name])
print('联系人 %s 删除成功！'%name)
else:
continue
else:
print('%s 不在通讯录中！'%name)
if button == 4:
print("|--- 感谢使用通讯录程序 ---|")
break

# -*- coding: gbk -*-
def Load():
dict1 = {'木又盛':'041962@zss'}
while 1:
key = input('''
|--- 新建用户：N/n ---|
|--- 登录帐号：E/e ---|
|--- 退出程序：Q/q ---|
|--- 请输入指令代码：''')
if key in [ 'N' , 'n' ]:
temp_name = input('请输入用户名：')
while temp_name in dict1:
temp_name = input('此用户名已经被使用，请重新输入：')

temp_password = input('请输入密码：')
dict1[temp_name] = temp_password
print('注册成功，赶紧试试登录吧^_^')
continue

elif key in ['E' , 'e']:
temp_name = input('请输入用户名:')
while temp_name not in dict1:
temp_name = input('您输入的用户名不存在，请重新输入：')
temp_password = input('请输入密码：')
while temp_password != dict1[temp_name]:
temp_password = input('密码错误，请重新输入：')
print('欢迎进入系统，请点右上角的X结束程序！')
continue

elif key == 'Q' or key == 'q':
break

Load()

'''
def main():
Load()

if __name__ == "__main()__":
main()
'''