前情提要:编写一个程序,用户登录时向其名字被存储的用户发出问候。用户首次运行程序时被提示输入自己的名字,向非第一次登录的用户发出问候。
10-13 验证用户:最后一个 remember_me.py 版本假设用户要么已输入其用户名,要么是首次运行该程序。我们应修改这个程序,以应对这样的情形:当前和最后一次运行该程序的用户并非同一个人。为此,在 greet_user()中打印欢迎用户回来的消息前,先询问他用户名是否是对的。如果不对,就调用 get_new_username()让用户输入正确的用户名。
模块json可以将简单的数据结构转储到文件中,并在程序再次运行时加载该文件中的数据。这里用到了json.dump()来将用户名存储到json文件中,用json.load()将json文件中的内容读取到内存中。
import json
def get_stored_username():
"""如果存储了用户名,就获取它"""
filename = 'username.json'
try:
with open(filename) as f_obj:
username = json.load(f_obj)
except FileNotFoundError:
return None
else:
return username
def get_new_username():
"""提示用户输入用户名"""
username = input("What's your name?")
filename = 'username.json'
with open(filename, 'w') as f_obj:
json.dump(username, f_obj)
return username
def greet_user():
"""问候用户并指出其名字"""
username = get_stored_username()
if username:
print("Plece confirm your user name is " + username) # 询问用户名是否是正确的
quire = input("If not please enter 'n' ")
if quire == 'n':
username = get_new_username()
print("We'll remember you name when you com back, " + username + "!")
else:
print("Welcome back, " + username)
else:
username = get_new_username()
print("We'll remember you name when you com back, " + username + "!")
greet_user()