很多程序都要求用户输入某种信息,如让用户存储游戏首选项或提供要可视化的数据。不管关注点是什么,程序都把用户提供的信息存储在列表和字典等数据结构中。用户关闭程序时,几乎总是要保存他们提供的信息。一种简单的方式是使用模块json来存储数据。
模块json让你能够将简单的Python数据结构转储到文件中,并在程序再次运行时加载该文件中的数据。你还可以使用json在Python程序之间分享数据。更重要的是,JSON数据格式并非Python专用的,这让你能够将以JSON格式存储的数据与使用其他编程语言的人分享。这是一种轻便而有用的格式,也易于学习。
使用json.dump()和json.load()
函数json.dump()接受两个实参:要存储的数据,以及可用于存储数据的文件对象。
import json
numbers = [2,3,5,7,11,13]
filename = 'numbers.json'
with open(filename,'w') as f:
json.dump(numbers,f)
结果:
[2, 3, 5, 7, 11, 13]
json.load()将列表读取到内存中
import json
file_name = 'numbers.json'
with open(file_name) as f:
numbers = json.load(f)
print(numbers)
保存和读取用户生成的数据
使用json保存用户生成的数据大有裨益,因为如果不以某种方式存储,用户的信息会在程序停止运行时丢失。
下面来看一个这样的例子:提示用户首次运行程序时输入自己的名字,并在再次运行程序时记住他。
import json
filename = 'username.json'
with open(filename) as f:
username = json.load(f)
print(f"Welcome back,{username}!")
需要将这两个程序合并到一个程序中
import json
#如果以前存储了用户名,就加载它
#否则,提示用户输入用户名并存储它
filename = 'username.json'
try:
with open(filename) as f:
username= json.load(f)
except FileNotFoundError:
username = input("What is your name? ")
with open(filename, 'w') as f:
json.dump(username, f)
print(f"We will remember you when you come back,{username}!")
else:
print(f"Welcome back,{username}!")
重构
你经常会遇到这样的情况:代码能够正确地运行,但通过将其划分为一系列完成具体工作的函数,还可以改进。这样的过程称为重构。重构让代码更清晰、更易于理解、更容易扩展。
import json
def get_stored_username():
"""如果存储了用户名,就获取它。"""
filename = 'username.json'
try:
with open(filename) as f:
username = json.load(f)
except FileNotFoundError:
return None
else:
return username
def get_new_username():
"""提示用户输入用户名"""
username = input("What is your name? ")
with open(filename, 'w') as f:
json.dump(username, f)
return username
def greet_user():
"""问候用户,并指出其名字"""
username = get_stored_username()
if username:
print(f"Welcome back,{username}")
else:
username = get_new_username()
print(f"We will remember you when you come back,{username}")
greet_user()