先编写一个程序，用户首次运行程序时被提示输入自己的名字，这样再次运行程序时就记住他了。
我们调用json.dump()将用户名存储到文件中，使用json.load() 将存储在username.json中的信息读取到变量username 中。恢复用户名后，我们就可以欢迎用户回来了。
# -*- coding: utf-8 -*-
import json
# 如果以前存储了用戶名，就加载它
# 否则，就提示用户输入用户名并存储它
filename = "username.json"
try:
with open(filename) as f_obj:
username = json.load(f_obj)
except IOError:
username = raw_input("What is your name: ")
filename = "username.json"
with open(filename,'w') as f_obj:
json.dump(username,f_obj)
print("We'll remember you when you come back, " + username + "!")
else:
print("Welcome back, " + username + "!")
首次运行的结果：
What is your name: leilei
We'll remember you when you come back, leilei!
Process finished with exit code 0
第二次运行的结果：
Welcome back, leilei!
Process finished with exit code 0
重构
为了让代码更清晰、更易于理解、更容易扩展，使用重构的方法改良代码，把问候用户的代码全部放到一个名为greet_user() 的函数中，再调用这个函数：
# -*- coding: utf-8 -*-
import json
def greet_user():
"""问候用户，并指出其名字"""
filename = "username.json"
try:
with open(filename) as f_obj:
username = json.load(f_obj)
except IOError:
username = raw_input("What is your name: ")
filename = "username.json"
with open(filename,'w') as f_obj:
json.dump(username,f_obj)
print("We'll remember you when you come back, " + username + "!")
else:
print("Welcome back, " + username + "!")
greet_user()
结果：
Welcome back, leilei!
Process finished with exit code 0
现在是一个函数执行多项任务，为了更简洁，将获取存储的用户名的代码移到另一个函数中：
# -*- coding: utf-8 -*-
import json
def get_stored_username():
"""如果存储了用户名，就获取它"""
filename = "username.json"
try:
with open(filename) as f_obj:
username = json.load(f_obj)
except IOError:
return None
else:
return username
def greet_user():
"""问候用户，并指出其名字"""
username = get_stored_username()
if username:
print("Welcome back, " + username + "!")
else:
username = raw_input("What is your name: ")
filename = "username.json"
with open(filename,'w') as f_obj:
json.dump(username,f_obj)
print("We'll remember you when you come back, " + username + "!")
greet_user()
新增的函数get_stored_username() 实现获取用户名。如果存储了用户名，这个函数就获取并返回它；如果文件username.json不存在，这个函数就返回None。
这是一种不错的做法：函数要么返回预期的值，要么返回None ；这让我们能够使用函数的返回值做简单测试。
如果成功地获取了用户名，就打印一条欢迎用户回来的消息，否则就提示用户输入用户名。
除此之外，我们还可以将提示用户输入用户名部分的代码单独作为一个函数使用：
# -*- coding: utf-8 -*-
import json
def get_stored_username():
"""如果存储了用户名，就获取它"""
filename = "username.json"
try:
with open(filename) as f_obj:
username = json.load(f_obj)
except IOError:
return None
else:
return username
def get_new_username():
"""提示用户输入用户名"""
username = raw_input("What is 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("Welcome back, " + username + "!")
else:
username = get_new_username()
print("We'll remember you when you come back, " + username + "!")
greet_user()
从上面的代码看出每个函数都执行单一而清晰的任务。我们调用greet_user() ，它打印一条合适的消息：要么欢迎老用户回来，要么问候新用户。
为此，它首先调用get_stored_username() ，这个函数只负责获取存储的用户名（如果存储了的话），再在必要时调用get_new_username() ，这个函数只负责获取并存储新用户的用户名。
网友评论