[python入门㊷] - python存储数据

目录

❤  json.dump()存储数据

❤  json.laod()读取数据

❤  保存和读取用户生成的数据

❤  重构

---

JSON(JavaScript Object Notation)格式最初是为JavaScript开发的，但随后成了一种常见格式，被包括Python在内的众多语言采用

❤  json.dump()存储数据

函数json.dump() 接受两个实参：要存储的数据以及可用于存储数据的文件对象

语法：json.dump(存储数据,文件名)

import json
numbers = [1,2,3,4,5,6]
file_name = 'TestDump.json'
with open(file_name,'w') as file_obj:json.dump(numbers,file_obj)

❤  json.laod()读取数据

语法：json.load(文件名)

下面是我们读取数据文件TestDump.json内容的程序

import json
file_name = 'TestDump.json'
with open(file_name) as file_obj:numbers = json.load(file_obj)
print(numbers)

❤  保存和读取用户生成的数据

对于用户生成的数据，使用json保存他们大有裨益，因为如果不以某种方式进行存储，等程序停止运行时用户的数据将会丢失。

import json
username = input("what is your name?")filename = 'username.json'
with open(filename,'w') as file_obj:json.dump(username,file_obj)print("we will remeber when you come back," + username+"!")

输出:

what is your name?olover
we will remeber when you come back,olover!

我们提示输入用户名，并将其存储在一个变量中，接下来，我们又调用json.dump(),并将用户名和一个文件对象传递给他，从而将用户名存储在文件中。

import json
file_name = 'username.json'
with open(file_name) as f_obj:username = json.load(f_obj)print("Welcome back,"+username+"!")

 输出:

Welcome back,olover!

我们将两个程序合并到一个程序中。在这个程序运行时，我们将尝试从文件username.json中获取用户名，因此我们首先编写一个尝试恢复用户名的try代码块。当文件不存在的时候，我们就在except中提示输入用户名，并将其存储到username.json文件中，以便程序再次运行时获取它。

import json
#如果以前存储了用户名，就加载它，否则，就提示用户输入用户名
file_name = 'username.json'
try:with open(file_name) as f_obj:username = json.load(f_obj)
except FileNotFoundError:username = input("what is your name?")with open(file_name,'w') as f_obj:json.dump(username,f_obj)print("we'll remeber you when you come back,"+username+"!")
else:print("Welcome back,"+username+"!")

 输出:

Welcome back,olover!

 ❤  重构

你经常会遇到这样的情况：代码能够正确运行，但可做进一步的改进，将代码划分为一系列完成具体工作的函数，这个过程称为重构

import json
def greet_user():'''问候用户，并指出其名字'''file_name = 'username.json'try:with open(file_name) as f_obj:username = json.load(f_obj)except FileNotFoundError:username  = input("what is your name?")with open(file_name,'w') as f_obj:json.dump(username,f_obj)print("we'll remeber you when you come back,"+username+"!")else:print("Welcome back,"+username+"!")
#调用函数
greet_user()

或许有人觉得，上面的程序执行的任务太多了，我们不想它执行那么多任务，我们首先将获取存储的用户名的代码移到另外一个函数中：

提取存在文件时执行操作

import json
def  get_stored_username():'''如果存储了用户名，就获取它'''file_name = 'username.json'try:with open(file_name) as f_obj:username = json.load(f_obj)except FileNotFoundError:return Noneelse:return username
def greet_user():'''问候用户，并指出其名字'''username = get_stored_username()if username:print("Welcome back," + username + "!")else:username = input("what is your name?")file_name = 'username.json'with open(file_name, 'w') as f_obj:json.dump(username, f_obj)print("we'll remeber you when you come back," + username + "!")
#调用函数
greet_user()

新增的函数get_stored_username()目标很明确，如果文件存储了用户名，这个函数就获取并返回它，如果文件不存在，这个函数就返回None,这是一种不错的方法：函数要么返回预期的值，要么返回None.

提取没有存储用户名时提示用户输入的操作

上面的程序还可以继续优化，将greet_user()中的另外一块代码提取出来：将没有存储用户名时提示用户输入的代码放在一个独立的函数中

import json
def  get_stored_username():'''如果存储了用户名，就获取它'''file_name = 'username.json'try:with open(file_name) as f_obj:username = json.load(f_obj)except FileNotFoundError:return Noneelse:return username
def get_new_username():'''提示用户输入用户名'''username = input("what is your name?")file_name = 'username.json'with open(file_name,'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 remeber you when you come back," + username + "!")
#调用函数
greet_user()