Python中如何将数据存储为json格式的文件(续)

将上一篇中的例子，修改一下，将两个程序合二为一，如果存储了用户喜欢的水果就显示它，否则提示用户输入他喜欢的水果并将其存储到文件中。

favorite.py

import json

filename = "favorite_fruit.json"

def greet_user():
    """问候用户，并指出他喜欢的水果"""
    fruit = get_saved_fruit()
    if fruit:
        print("I know your favorite fruit !  It's " + fruit)
    else:
        fruit = save_user_new_fruit()
        print("I know your favorite fruit !  It's " + fruit)


    

def save_user_new_fruit():
    """提示用户输入喜欢的水果"""
    fruit = input(" What is your favorite fruit ? ")
    with open(filename,'w') as  file_obj:
        json.dump(file_obj,fruit)
    return fruit

def get_saved_fruit():
    """如果存储了用户喜欢的水果，就获取它"""
    try :
        with open(filename) as  file_obj:
            fruit = json.load(file_obj)
    except FileNotFoundError:
        return None
    else:
        return fruit
        
greet_user()

解释：在优化代码过后，每个函数只需要执行单一任务，这更符合程序的设计

第一次运行程序后的控制台如下：

如果我们是第一次运行，如下图

会让用户先输入最喜欢的水果然后回应用户他最喜欢的水果。

而之后的运行，则如图：

因为程序在第一次运行之后已经将用户喜欢的水果存入了json文件。

所以不会提示用户输入。